From 2017fc6240766a51cefdf88491214a8c84c4082d Mon Sep 17 00:00:00 2001 From: zhonghao lu Date: Fri, 19 Sep 2025 10:32:25 +0800 Subject: [PATCH 1/2] fix: fix agent service lint error --- python/valuecell/server/services/agent_stream_service.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python/valuecell/server/services/agent_stream_service.py b/python/valuecell/server/services/agent_stream_service.py index db699c0e7..0cf33726c 100644 --- a/python/valuecell/server/services/agent_stream_service.py +++ b/python/valuecell/server/services/agent_stream_service.py @@ -4,7 +4,6 @@ from typing import AsyncGenerator, Optional from valuecell.core.coordinate.orchestrator import get_default_orchestrator -from valuecell.core.coordinate.tests.test_orchestrator import session_id from valuecell.core.types import UserInput, UserInputMetadata import logging From 533cb33defce035a0f737876163ed6d6a33eb133 Mon Sep 17 00:00:00 2001 From: zhonghao lu Date: Fri, 19 Sep 2025 10:33:45 +0800 Subject: [PATCH 2/2] fix format --- python/valuecell/agents/sec_agent.py | 3 --- python/valuecell/core/coordinate/models.py | 4 +++- python/valuecell/server/api/schemas/agent_stream.py | 6 ++++-- 3 files changed, 7 insertions(+), 6 deletions(-) diff --git a/python/valuecell/agents/sec_agent.py b/python/valuecell/agents/sec_agent.py index c11600773..4ae4d2b95 100644 --- a/python/valuecell/agents/sec_agent.py +++ b/python/valuecell/agents/sec_agent.py @@ -256,9 +256,6 @@ async def _process_fund_holdings_query( o = filings[2].obj() previous_filing = o.infotable.to_json() - - - logger.info("Successfully parsed current and historical holdings data") # Generate 13F analysis report diff --git a/python/valuecell/core/coordinate/models.py b/python/valuecell/core/coordinate/models.py index c4cacb88f..c35fcfbd6 100644 --- a/python/valuecell/core/coordinate/models.py +++ b/python/valuecell/core/coordinate/models.py @@ -8,7 +8,9 @@ class ExecutionPlan(BaseModel): """Execution plan containing multiple tasks""" plan_id: str = Field(..., description="Unique plan identifier") - session_id: Optional[str] = Field(..., description="Session ID this plan belongs to") + session_id: Optional[str] = Field( + ..., description="Session ID this plan belongs to" + ) user_id: str = Field(..., description="User ID who requested this plan") query: str = Field(..., description="Original user input") tasks: List[Task] = Field(default_factory=list, description="Tasks to execute") diff --git a/python/valuecell/server/api/schemas/agent_stream.py b/python/valuecell/server/api/schemas/agent_stream.py index 7470d51cb..59f5f16fb 100644 --- a/python/valuecell/server/api/schemas/agent_stream.py +++ b/python/valuecell/server/api/schemas/agent_stream.py @@ -9,7 +9,9 @@ class AgentStreamRequest(BaseModel): """Request model for agent streaming queries.""" query: str = Field(..., description="User query to send to the agent") - agent_name: str = Field(None, description="Specific agent name to use for the query") + agent_name: str = Field( + None, description="Specific agent name to use for the query" + ) class Config: json_schema_extra = { @@ -29,4 +31,4 @@ class StreamChunk(BaseModel): class Config: json_schema_extra = { "example": {"content": "The current market shows...", "is_final": False} - } \ No newline at end of file + }