Spaces:
Sleeping
Sleeping
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
|
| 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
|
| 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 =
|
| 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]}
|