-
-
Notifications
You must be signed in to change notification settings - Fork 740
fix: resolve OpenRouter XML tool call routing issue #1080
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.
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
Open
MervinPraison
wants to merge
2
commits into
main
Choose a base branch
from
claude/openrouter-xml-tools-fix-20250805
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Changes from all commits
Commits
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or 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
This file contains hidden or 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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,87 @@ | ||
#!/usr/bin/env python3 | ||
"""Test script for OpenRouter XML tool call fix""" | ||
|
||
import os | ||
import sys | ||
|
||
# Add the source directory to the path | ||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src', 'praisonai-agents')) | ||
|
||
from praisonaiagents import Agent | ||
|
||
def get_weather(city: str) -> str: | ||
"""Get weather information for a city""" | ||
return f"The weather in {city} is sunny with 22°C" | ||
|
||
def main(): | ||
print("Testing OpenRouter XML tool call fix...") | ||
|
||
# Test with auto-detection (should detect Qwen as XML format) | ||
agent = Agent( | ||
instructions="You are a helpful assistant", | ||
llm="openrouter/qwen/qwen-2.5-7b-instruct", | ||
tools=[get_weather], | ||
verbose=True | ||
) | ||
|
||
print("Created agent with Qwen model...") | ||
|
||
# Get the LLM instance directly from the agent | ||
llm_instance = agent.llm_instance # This should be the LLM object | ||
print(f"XML tool format supported: {llm_instance._supports_xml_tool_format()}") | ||
|
||
# Test the tool call without actually making API request | ||
# We'll just verify the parameters are built correctly | ||
test_tools = [ | ||
{ | ||
"type": "function", | ||
"function": { | ||
"name": "get_weather", | ||
"description": "Get weather information for a city", | ||
"parameters": { | ||
"type": "object", | ||
"properties": { | ||
"city": { | ||
"type": "string", | ||
"description": "The city name" | ||
} | ||
}, | ||
"required": ["city"] | ||
} | ||
} | ||
} | ||
] | ||
|
||
# Test _build_completion_params | ||
params = llm_instance._build_completion_params( | ||
messages=[{"role": "user", "content": "What's the weather in Tokyo?"}], | ||
tools=test_tools, | ||
temperature=0.2 | ||
) | ||
|
||
print("\n=== Completion Parameters ===") | ||
print(f"Model: {params.get('model')}") | ||
print(f"Tools included: {'tools' in params}") | ||
print(f"Tool choice included: {'tool_choice' in params}") | ||
|
||
# Test _build_messages | ||
messages, original = llm_instance._build_messages( | ||
prompt="What's the weather in Tokyo?", | ||
system_prompt="You are a helpful assistant", | ||
tools=test_tools | ||
) | ||
|
||
print("\n=== System Message ===") | ||
for msg in messages: | ||
if msg['role'] == 'system': | ||
print(msg['content']) | ||
break | ||
|
||
print("\n✅ Test completed successfully!") | ||
print("Key improvements:") | ||
print("- Tools parameter is removed for XML format models") | ||
print("- Tool descriptions are added to system prompt") | ||
print("- XML tool call format instructions are included") | ||
|
||
if __name__ == "__main__": | ||
main() | ||
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.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The current test script relies on manual inspection of the output. Convert it into an automated test with assertions to prevent future regressions and facilitate integration with testing frameworks like
pytest
.Here's a suggested structure for an automated test: