Spaces:
Sleeping
Sleeping
File size: 1,950 Bytes
8039e4b 5c271a3 583f6dd 8039e4b 583f6dd ef9fa4b 583f6dd ef9fa4b 5c271a3 8039e4b 583f6dd 8039e4b 5c271a3 ef9fa4b 5c271a3 a6a0614 5c271a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import logging
from fastapi import APIRouter
from fastapi.responses import StreamingResponse
from api.stored_data import stored_data
from src.genai.brainstroming_agent.agent import BrainstormingAgent
from api.schemas.brainstorming import BrainstormRequest
from api.backup_prompts import brainstorm_backup_prompt
router=APIRouter()
agent = BrainstormingAgent()
brainstorm_graph = agent.brainstorming_graph()
app_logger = logging.getLogger("app_logger")
error_logger = logging.getLogger("error_logger")
@router.post("/brainstorm")
def brainstorming_endpoint(request: BrainstormRequest):
try:
idea = (
[stored_data['refined_ideation']]
if stored_data.get('refined_ideation')
else [str(stored_data['final_ideation'])]
if stored_data.get('final_ideation')
else [brainstorm_backup_prompt]
)
def event_stream():
for stream_mode, chunk in brainstorm_graph.stream(
{
"idea": idea,
"images": request.image_base64_list,
"latest_preferred_topics": request.preferred_topics,
"business_details": stored_data["business_details"],
},
config={"configurable": {"thread_id": request.thread_id},"run_name":"brainstorming"},
stream_mode=['messages','values']
):
stored_data['brainstorming_response'] = chunk if isinstance(chunk, dict) else stored_data.get('brainstorming_response')
# if isinstance(chunk, tuple): yield f"data: {json.dumps(chunk[0].content)}"
if isinstance(chunk, tuple): yield chunk[0].content
app_logger.info('Executed brainstorming agent.')
return StreamingResponse(event_stream(), media_type="text/event-stream")
except Exception as e:
error_logger.error('Unable to execute brainstorm agent.')
raise
|