Spaces:
Running
Running
Commit
·
cb8a8b6
1
Parent(s):
dcb385e
Switch primary model in write_message method to ChatGroq with fallback to ChatGoogleGenerativeAI
Browse files
app.py
CHANGED
|
@@ -143,11 +143,12 @@ def write_message(data: dict, header: str = Header()):
|
|
| 143 |
message = data['message']
|
| 144 |
|
| 145 |
try:
|
| 146 |
-
model =
|
| 147 |
transaction_bot = Agent(model, system=prompt)
|
| 148 |
transaction_bot.graph.invoke({"messages": [message]})
|
| 149 |
except Exception as e: #fallback model
|
| 150 |
-
|
|
|
|
| 151 |
transaction_bot = Agent(model, system=prompt)
|
| 152 |
transaction_bot.graph.invoke({"messages": [message]})
|
| 153 |
|
|
|
|
| 143 |
message = data['message']
|
| 144 |
|
| 145 |
try:
|
| 146 |
+
model = ChatGroq(model=GROQ_MODEL, temperature=1, callbacks = [OpikTracer()])
|
| 147 |
transaction_bot = Agent(model, system=prompt)
|
| 148 |
transaction_bot.graph.invoke({"messages": [message]})
|
| 149 |
except Exception as e: #fallback model
|
| 150 |
+
|
| 151 |
+
model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, callbacks = [OpikTracer()])
|
| 152 |
transaction_bot = Agent(model, system=prompt)
|
| 153 |
transaction_bot.graph.invoke({"messages": [message]})
|
| 154 |
|