Skip to content

Comments

feat: infer model name for agent nodes in graph schema#193

Merged
cristipufu merged 1 commit intomainfrom
feat/agent-framework-model-node-type
Feb 22, 2026
Merged

feat: infer model name for agent nodes in graph schema#193
cristipufu merged 1 commit intomainfrom
feat/agent-framework-model-node-type

Conversation

@cristipufu
Copy link
Member

Summary

  • Agent executor nodes that have a chat client with a model_id are now emitted as type="model" with metadata={"model_name": "..."} in the graph schema
  • Adds _get_model_name() helper that extracts model name via agent.client.model_id (works for both OpenAIChatClient and AnthropicClient)
  • Bumps version to 0.0.9

Test plan

  • 4 new graph tests covering: model node detection, fallback to regular node, plain executor, multi-agent with different models
  • All 120 tests pass
  • mypy clean
  • ruff format/check clean

🤖 Generated with Claude Code

AgentExecutor nodes with a chat client are now emitted as type "model"
with the model name in metadata, enabling downstream consumers to
identify which LLM each agent uses.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@cristipufu cristipufu merged commit ef6f42a into main Feb 22, 2026
46 checks passed
@cristipufu cristipufu deleted the feat/agent-framework-model-node-type branch February 22, 2026 06:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant