Shago commited on
Commit
f183d0d
·
verified ·
1 Parent(s): 5f273ad

Upload interface.py

Browse files
Files changed (1) hide show
  1. interfaces/interface.py +2 -3
interfaces/interface.py CHANGED
@@ -22,8 +22,7 @@ def create_interface():
22
  try:
23
  inputs = {"messages": [HumanMessage(content=query)]}
24
  result = app.invoke(inputs)
25
- return result
26
- # return messages_to_dict(result['messages'])[2]['data']['content']
27
  except Exception as e:
28
  return {"error": f"Execution error: {str(e)}"}
29
 
@@ -46,4 +45,4 @@ def create_interface():
46
  inputs=input_text,
47
  outputs=output_json
48
  )
49
- return interface
 
22
  try:
23
  inputs = {"messages": [HumanMessage(content=query)]}
24
  result = app.invoke(inputs)
25
+ return messages_to_dict(result['messages'])[2]['data']['content']
 
26
  except Exception as e:
27
  return {"error": f"Execution error: {str(e)}"}
28
 
 
45
  inputs=input_text,
46
  outputs=output_json
47
  )
48
+ return interface