robinroy03 commited on
Commit
3ebd2b4
1 Parent(s): 051904a

migrated to new db, improved references

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -118,9 +118,9 @@ async def on_message(message):
118
  data = ast.literal_eval(data)
119
  references += ("<https://github.com/fury-gl/fury/tree/master/" + data['path'] + ">")
120
  if data.get("function_name"):
121
- references += f"\tFunction Name: {data.get("function_name")}"
122
  else:
123
- references += f"\tClass Name: {data.get("class_name")}"
124
  references += "\n"
125
 
126
  llm_answer: str = await llm_output(question, db_context) # for the highest knn result (for the test only right now) TODO: make this better
 
118
  data = ast.literal_eval(data)
119
  references += ("<https://github.com/fury-gl/fury/tree/master/" + data['path'] + ">")
120
  if data.get("function_name"):
121
+ references += f"\tFunction Name: {data.get('function_name')}"
122
  else:
123
+ references += f"\tClass Name: {data.get('class_name')}"
124
  references += "\n"
125
 
126
  llm_answer: str = await llm_output(question, db_context) # for the highest knn result (for the test only right now) TODO: make this better