Spaces:
Sleeping
Sleeping
Commit
·
d3a918c
1
Parent(s):
17a4846
15th changes
Browse files
Email_Reader/email_response.py
CHANGED
|
@@ -48,11 +48,12 @@ class EmailResponder:
|
|
| 48 |
|
| 49 |
|
| 50 |
self.DB_PATH = DB_PATH
|
| 51 |
-
self.vectorstore = Chroma(persist_directory=self.DB_PATH, embedding_function=self.embed_model)
|
| 52 |
self.retriever = self.vectorstore.as_retriever()
|
| 53 |
self.prompt = ChatPromptTemplate.from_template(self.template)
|
| 54 |
self.ollama_llm = OLLAMA_MODEL
|
| 55 |
self.model_local = ChatOllama(model=self.ollama_llm)
|
|
|
|
| 56 |
self.chain = (
|
| 57 |
{"context": self.retriever, "question": RunnablePassthrough()}
|
| 58 |
| self.prompt
|
|
@@ -60,6 +61,7 @@ class EmailResponder:
|
|
| 60 |
| self.model_local
|
| 61 |
| StrOutputParser()
|
| 62 |
)
|
|
|
|
| 63 |
with open('emailCategories.json') as user_file:
|
| 64 |
self.jasonFile = json.load(user_file)
|
| 65 |
|
|
@@ -87,15 +89,15 @@ class EmailResponder:
|
|
| 87 |
orderNum = 0xFFFF
|
| 88 |
self.getShopifyInfo(orderNum)
|
| 89 |
query = f"Todays date -{today}\n sentiment - {sentiment_label}\n Subject -{subject}\n Body-{body} "
|
| 90 |
-
reply_body = self.chain.invoke(query)
|
| 91 |
-
|
| 92 |
reply_body = 0
|
| 93 |
orderNum = self.getOrderNumber(body, subject)
|
| 94 |
if orderNum != 0xFFFF:
|
| 95 |
self.getShopifyInfo(orderNum)
|
| 96 |
reply_body = self.get_GPTcompletion(orderNum, body)
|
| 97 |
return sentiment_label, sentiment_score, reply_body
|
| 98 |
-
|
| 99 |
except Exception as e:
|
| 100 |
logging.error(f"Error generating response: {e}")
|
| 101 |
raise
|
|
|
|
| 48 |
|
| 49 |
|
| 50 |
self.DB_PATH = DB_PATH
|
| 51 |
+
#self.vectorstore = Chroma(persist_directory=self.DB_PATH, embedding_function=self.embed_model)
|
| 52 |
self.retriever = self.vectorstore.as_retriever()
|
| 53 |
self.prompt = ChatPromptTemplate.from_template(self.template)
|
| 54 |
self.ollama_llm = OLLAMA_MODEL
|
| 55 |
self.model_local = ChatOllama(model=self.ollama_llm)
|
| 56 |
+
'''
|
| 57 |
self.chain = (
|
| 58 |
{"context": self.retriever, "question": RunnablePassthrough()}
|
| 59 |
| self.prompt
|
|
|
|
| 61 |
| self.model_local
|
| 62 |
| StrOutputParser()
|
| 63 |
)
|
| 64 |
+
'''
|
| 65 |
with open('emailCategories.json') as user_file:
|
| 66 |
self.jasonFile = json.load(user_file)
|
| 67 |
|
|
|
|
| 89 |
orderNum = 0xFFFF
|
| 90 |
self.getShopifyInfo(orderNum)
|
| 91 |
query = f"Todays date -{today}\n sentiment - {sentiment_label}\n Subject -{subject}\n Body-{body} "
|
| 92 |
+
#reply_body = self.chain.invoke(query)
|
| 93 |
+
|
| 94 |
reply_body = 0
|
| 95 |
orderNum = self.getOrderNumber(body, subject)
|
| 96 |
if orderNum != 0xFFFF:
|
| 97 |
self.getShopifyInfo(orderNum)
|
| 98 |
reply_body = self.get_GPTcompletion(orderNum, body)
|
| 99 |
return sentiment_label, sentiment_score, reply_body
|
| 100 |
+
|
| 101 |
except Exception as e:
|
| 102 |
logging.error(f"Error generating response: {e}")
|
| 103 |
raise
|