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,
|