subashpoudel commited on
Commit
2c2c90a
·
1 Parent(s): cdba388

Edited the filename of initial interaction

Browse files
my_agent/utils/initial_interaction.py CHANGED
@@ -6,7 +6,7 @@ from langchain_core.messages import SystemMessage
6
  from pydantic import BaseModel, ConfigDict, Field
7
  from typing import Optional, List
8
  from .models_loader import llm
9
- from .prompts import business_interaction_prompt , details_extract_prompt
10
 
11
  # Pydantic model for extracted business info
12
  class DetailsFormatter(BaseModel):
@@ -25,7 +25,7 @@ class State(BaseModel):
25
  # Global business state (shared)
26
  business_state = State()
27
 
28
- class BusinessInteractionChatbot:
29
  def __init__(self):
30
  self.memory = MemorySaver()
31
  # self.llm = ChatGroq(model_name="Gemma2-9b-It")
@@ -42,7 +42,7 @@ class BusinessInteractionChatbot:
42
  return workflow
43
 
44
  def _call_model(self, state):
45
- template = business_interaction_prompt
46
  messages = [SystemMessage(content=template)] + state["messages"]
47
  response = self.llm.invoke(messages)
48
  return {"messages": [response]}
 
6
  from pydantic import BaseModel, ConfigDict, Field
7
  from typing import Optional, List
8
  from .models_loader import llm
9
+ from .prompts import introduction_prompt , details_extract_prompt
10
 
11
  # Pydantic model for extracted business info
12
  class DetailsFormatter(BaseModel):
 
25
  # Global business state (shared)
26
  business_state = State()
27
 
28
+ class IntroductionChatbot:
29
  def __init__(self):
30
  self.memory = MemorySaver()
31
  # self.llm = ChatGroq(model_name="Gemma2-9b-It")
 
42
  return workflow
43
 
44
  def _call_model(self, state):
45
+ template = introduction_prompt
46
  messages = [SystemMessage(content=template)] + state["messages"]
47
  response = self.llm.invoke(messages)
48
  return {"messages": [response]}