Fix type of excalidraw image response

This commit is contained in:
sabaimran 2024-11-20 19:01:13 -08:00
parent 467de76fc1
commit 3519dd76f0

View file

@ -826,7 +826,7 @@ async def generate_excalidraw_diagram_from_description(
user: KhojUser = None, user: KhojUser = None,
agent: Agent = None, agent: Agent = None,
tracer: dict = {}, tracer: dict = {},
) -> str: ) -> Dict[str, Any]:
personality_context = ( personality_context = (
prompts.personality_context.format(personality=agent.personality) if agent and agent.personality else "" prompts.personality_context.format(personality=agent.personality) if agent and agent.personality else ""
) )