Minor clean up in preparation of graph agent #779
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
dmontagu
force-pushed
the
dmontagu/graph-agent-prep
branch
from
a064ead to
fb95046
Compare
Deploying pydantic-ai with
|
| Latest commit: |
fbff76c
|
| Status: | ✅ Deploy successful! |
| Preview URL: | https://2f5633e9.pydantic-ai.pages.dev |
| Branch Preview URL: | https://dmontagu-graph-agent-prep.pydantic-ai.pages.dev |
sydney-runkle
approved these changes
Jan 27, 2025
Comment on lines
+162
to
164
| # TODO: Move this state off the Tool class, which is otherwise stateless. | ||
| # This should be tracked inside a specific agent run, not the tool. | ||
| current_retry: int = field(default=0, init=False) |
There was a problem hiding this comment.
I've created #784 to track this
|
Gemini is having live test issues. |
dmontagu
force-pushed
the
dmontagu/graph-agent-prep
branch
from
31bee5d to
fbff76c
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
RunResultDatatypevar toRunResultDataTfor consistency with other typevars from Improve variance of classes #726get_captured_run_messagesfunction to access the private context var's data in another module (we will need access to this for that contextvar to work with GraphAgent. This change can be undone later; the function is not part of__all__.)Tool.runfrom_messages.ModelRequestPartto the more specific_messages.ToolReturnPart | _messages.RetryPromptPart(no code changes necessary, just a more specific signature)node_spanto theGraphRunContext