VibecoderMcSwaggins commited on
Commit
6ccc865
·
2 Parent(s): 2c12244 586a3f1

Merge branch 'dev'

Browse files
Files changed (1) hide show
  1. src/agents/graph/workflow.py +4 -2
src/agents/graph/workflow.py CHANGED
@@ -1,5 +1,7 @@
1
  """DeepBoner research workflow definition using LangGraph."""
2
 
 
 
3
  from functools import partial
4
  from typing import Any
5
 
@@ -21,9 +23,9 @@ from src.services.embeddings import EmbeddingService
21
 
22
  def create_research_graph(
23
  llm: BaseChatModel | None = None,
24
- checkpointer: "BaseCheckpointSaver[Any]" | None = None, # Generic type from langgraph
25
  embedding_service: EmbeddingService | None = None,
26
- ) -> "CompiledStateGraph[Any]": # type: ignore[type-arg]
27
  """Build the research state graph.
28
 
29
  Args:
 
1
  """DeepBoner research workflow definition using LangGraph."""
2
 
3
+ from __future__ import annotations
4
+
5
  from functools import partial
6
  from typing import Any
7
 
 
23
 
24
  def create_research_graph(
25
  llm: BaseChatModel | None = None,
26
+ checkpointer: BaseCheckpointSaver[Any] | None = None,
27
  embedding_service: EmbeddingService | None = None,
28
+ ) -> CompiledStateGraph[Any]:
29
  """Build the research state graph.
30
 
31
  Args: