Update TextGen/router.py
Browse files- 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
|
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,
|