Mohammad Haghir commited on
Commit ·
900248e
1
Parent(s): 59c44f9
simple solution
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import requests
|
|
| 4 |
import inspect
|
| 5 |
import pandas as pd
|
| 6 |
import re
|
|
|
|
| 7 |
|
| 8 |
from langchain_groq import ChatGroq
|
| 9 |
from langchain_core.messages import HumanMessage
|
|
@@ -51,7 +52,8 @@ class BasicAgent:
|
|
| 51 |
messages = [HumanMessage(content=prompt)]
|
| 52 |
response = llm.invoke(messages)
|
| 53 |
cleaned_text = re.sub(r"<think>.*?</think>", "", response.content, flags=re.DOTALL)
|
| 54 |
-
|
|
|
|
| 55 |
|
| 56 |
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
| 57 |
"""
|
|
|
|
| 4 |
import inspect
|
| 5 |
import pandas as pd
|
| 6 |
import re
|
| 7 |
+
import json
|
| 8 |
|
| 9 |
from langchain_groq import ChatGroq
|
| 10 |
from langchain_core.messages import HumanMessage
|
|
|
|
| 52 |
messages = [HumanMessage(content=prompt)]
|
| 53 |
response = llm.invoke(messages)
|
| 54 |
cleaned_text = re.sub(r"<think>.*?</think>", "", response.content, flags=re.DOTALL)
|
| 55 |
+
res = json.loads(cleaned_text)
|
| 56 |
+
return res
|
| 57 |
|
| 58 |
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
| 59 |
"""
|