apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-01 11:26:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

docker fixes for agents

Browse source
This commit is contained in:
Ramnique Singh 2025-03-26 17:01:49 +05:30
parent 695a961333
commit 277a553ee1
6 changed files with 46 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

9
apps/rowboat_agents/src/app/main.py
View file

@ -120,8 +120,6 @@ async def chat_stream_init():
data = await request.get_json()
redis_client.setex(f"stream_request_{stream_id}", 600, json.dumps(data))
print('* stream init'*200)
return jsonify({"streamId": stream_id})
def format_sse(data: dict, event: str = None) -> str:
@ -157,11 +155,8 @@ async def chat_stream(stream_id):
elif not msg.get("role"):
msg["role"] = "user"
print('*'*200)
print("Request:")
print('*'*200)
pprint(request_data)
print('='*200)
async def generate():
try:
@ -176,14 +171,10 @@ async def chat_stream(stream_id):
complete_request=request_data
):
if event_type == 'message':
print('*'*200)
print("Yielding message:")
print('*'*200)
yield format_sse(event_data, "message")
elif event_type == 'done':
print('*'*200)
print("Yielding done:")
print('*'*200)
yield format_sse(event_data, "done")
except Exception as e: