Jofthomas HF staff commited on
Commit
48a69b6
1 Parent(s): 5553350

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -70,7 +70,7 @@ def generate_text(messages: List[str], npc:str):
70
  )
71
  llm_response = llm.invoke(new_messages)
72
  print(llm_response)
73
- return Generate(text=llm_response["content"])
74
 
75
  app.add_middleware(
76
  CORSMiddleware,
 
70
  )
71
  llm_response = llm.invoke(new_messages)
72
  print(llm_response)
73
+ return Generate(text=llm_response.content)
74
 
75
  app.add_middleware(
76
  CORSMiddleware,