khalidsabri commited on
Commit
d3a918c
·
1 Parent(s): 17a4846

15th changes

Browse files
Files changed (1) hide show
  1. Email_Reader/email_response.py +6 -4
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