|
13 | 13 | import json |
14 | 14 | import logging |
15 | 15 | import time |
| 16 | +from collections.abc import AsyncIterator |
16 | 17 | from os import getenv |
17 | 18 | from pathlib import Path |
18 | | -from collections.abc import AsyncIterator |
19 | 19 | from typing import Any |
20 | 20 | from uuid import uuid4 |
21 | 21 |
|
22 | 22 | import uvicorn |
23 | | -from dotenv import load_dotenv |
24 | | -from langchain.agents import create_agent |
25 | | -from langchain_core.messages import AIMessage, HumanMessage |
26 | | -from langchain_core.tools import tool |
27 | | -from langchain_openai import ChatOpenAI |
28 | | -from starlette.applications import Starlette |
29 | | -from starlette.exceptions import HTTPException |
30 | | -from starlette.requests import Request |
31 | | -from starlette.responses import FileResponse, JSONResponse, StreamingResponse |
32 | | -from starlette.routing import Route |
33 | | - |
34 | 23 | from a2a.server.agent_execution import AgentExecutor, RequestContext |
35 | 24 | from a2a.server.apps import A2AStarletteApplication |
36 | 25 | from a2a.server.events import EventQueue |
|
44 | 33 | SendMessageRequest, |
45 | 34 | ) |
46 | 35 | from a2a.utils import new_agent_text_message |
| 36 | +from dotenv import load_dotenv |
| 37 | +from langchain.agents import create_agent |
| 38 | +from langchain_core.messages import AIMessage, HumanMessage |
| 39 | +from langchain_core.tools import tool |
| 40 | +from langchain_openai import ChatOpenAI |
| 41 | +from starlette.applications import Starlette |
| 42 | +from starlette.exceptions import HTTPException |
| 43 | +from starlette.requests import Request |
| 44 | +from starlette.responses import FileResponse, JSONResponse, StreamingResponse |
| 45 | +from starlette.routing import Route |
47 | 46 |
|
48 | 47 | from .a2a_reply import send_a2a_text_message |
49 | 48 |
|
@@ -278,7 +277,8 @@ async def _stream_orchestrator_sse( |
278 | 277 | tool_calls = getattr(message, "tool_calls", None) or [] |
279 | 278 | if tool_calls: |
280 | 279 | tool_calls_delta = [ |
281 | | - _tool_call_to_delta(tc, i) for i, tc in enumerate(tool_calls) |
| 280 | + _tool_call_to_delta(tc, i) |
| 281 | + for i, tc in enumerate(tool_calls) |
282 | 282 | ] |
283 | 283 | data = { |
284 | 284 | "id": completion_id, |
|
0 commit comments