camparchimedes commited on
Commit
5e7e4bb
ยท
verified ยท
1 Parent(s): 9a9d8fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -8,7 +8,7 @@ import time
8
  import json
9
  import torch
10
 
11
- from api_docs_mck import daysoff_api_docs, daysoff_api_docs_str
12
 
13
  import chainlit as cl
14
 
@@ -75,7 +75,7 @@ def setup_multiple_chains():
75
 
76
  api_chain = APIChain.from_llm_and_api_docs(
77
  llm=llm,
78
- api_docs=daysoff_api_docs,
79
  api_url_prompt=api_url_prompt,
80
  api_response_prompt=api_response_prompt,
81
  verbose=True,
@@ -124,7 +124,4 @@ async def handle_message(message: cl.Message):
124
 
125
  response_key = "output" if "output" in response else "text"
126
  await cl.Message(response.get(response_key, "")).send()
127
- return message.content
128
-
129
-
130
-
 
8
  import json
9
  import torch
10
 
11
+ from api_docs_mck import daysoff_api_docs
12
 
13
  import chainlit as cl
14
 
 
75
 
76
  api_chain = APIChain.from_llm_and_api_docs(
77
  llm=llm,
78
+ api_docs=daysoff_api_docs_str,
79
  api_url_prompt=api_url_prompt,
80
  api_response_prompt=api_response_prompt,
81
  verbose=True,
 
124
 
125
  response_key = "output" if "output" in response else "text"
126
  await cl.Message(response.get(response_key, "")).send()
127
+ return message.content