Mohammad Haghir commited on
Commit
900248e
·
1 Parent(s): 59c44f9

simple solution

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- return cleaned_text
 
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
  """