gerglitzen commited on
Commit
3583222
·
1 Parent(s): ec4f003
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -57,6 +57,7 @@ class PineconeSearch:
57
 
58
 
59
  def query_tool(category, pinecone_topk, query):
 
60
  data = {
61
  "1_D3_receptor": "demo-richter-target-400-30-1",
62
  "2_dopamine": "demo-richter-target-400-30-2",
@@ -93,7 +94,7 @@ def stream(input_text, history, user_prompt, topic, topk) -> Generator:
93
 
94
  # Create a funciton to call - this will run in a thread
95
  def task():
96
- tool_resp = query_tool(topic, topk, input_text)
97
 
98
  response = call_openai(
99
  messages=[{"role": "system", "content": agent_prompt},
 
57
 
58
 
59
  def query_tool(category, pinecone_topk, query):
60
+ print(query)
61
  data = {
62
  "1_D3_receptor": "demo-richter-target-400-30-1",
63
  "2_dopamine": "demo-richter-target-400-30-2",
 
94
 
95
  # Create a funciton to call - this will run in a thread
96
  def task():
97
+ tool_resp = query_tool(topic, topk, str(input_text))
98
 
99
  response = call_openai(
100
  messages=[{"role": "system", "content": agent_prompt},