Skip to content

Fix/celery error #183

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Mile-Away merged 5 commits into from
Jan 19, 2026
Merged

Fix/celery error #183

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
2fdb6f2
feat: add celery-types
xinquiry Jan 19, 2026
4615d2a
fix: improve streaming support for LangGraph agents
xinquiry Jan 19, 2026
ab02a5b
Merge branch 'main' of github.com:ScienceOL/Xyzen into feature/better…
xinquiry Jan 19, 2026
3231061
fix: use get of state
xinquiry Jan 19, 2026
c3db10a
Merge branch 'main' of github.com:ScienceOL/Xyzen into fix/celery-error
xinquiry Jan 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions service/app/agents/graph_builder.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -331,15 +331,15 @@ async def _build_llm_node(self, config: GraphNodeConfig) -> NodeFunction:
else:
configured_llm = base_llm

async def llm_node(state: StateDict) -> StateDict:
async def llm_node(state: StateDict | BaseModel) -> StateDict:
logger.info(f"[LLM Node: {config.id}] Starting execution")

# Access messages directly from state
# LangGraph passes state as dict with 'messages' key containing BaseMessage objects
messages: list[BaseMessage] = list(state.get("messages", []))
# Convert state to dict (handles both dict and Pydantic BaseModel)
state_dict = self._state_to_dict(state)
messages: list[BaseMessage] = list(state_dict.get("messages", []))

# Render prompt template
prompt = self._render_template(llm_config.prompt_template, state)
prompt = self._render_template(llm_config.prompt_template, state_dict)
Copy link

Copilot AI Jan 19, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Passing state_dict to _render_template is redundant because _render_template already calls self._state_to_dict(state) internally at line 255. This results in _state_to_dict being called twice on the already-converted dict. Instead, pass state directly to _render_template to avoid the redundant conversion: prompt = self._render_template(llm_config.prompt_template, state)

Suggested change
prompt = self._render_template(llm_config.prompt_template, state_dict)
prompt = self._render_template(llm_config.prompt_template, state)

Copilot uses AI. Check for mistakes.

# Build messages for LLM
llm_messages = messages + [HumanMessage(content=prompt)]
Expand Down
Loading