Function test_openai_tool_calling

tests/test_llm_clients.py:168–224

Test OpenAI client with tool calling.

()

@pytest.mark.asyncio
async def test_openai_tool_calling():
    """Test OpenAI client with tool calling."""
    print("\n=== Testing OpenAI Tool Calling ===")

    config = load_config()

    # Create OpenAI client
    client = OpenAIClient(
        api_key=config["api_key"],
        api_base="https://api.minimaxi.com/v1",
        model=config.get("model", "MiniMax-M2.5"),
    )

    # Define tool using dict format (will be converted internally for OpenAI)
    tools = [
        {
            "name": "get_weather",
            "description": "Get weather of a location",
            "input_schema": {
                "type": "object",
                "properties": {
                    "location": {
                        "type": "string",
                        "description": "The city and state, e.g. San Francisco, US",
                    }
                },
                "required": ["location"],
            },
        }
    ]

    # Messages requesting tool use
    messages = [
        Message(role="user", content="What's the weather in New York?"),
    ]

    try:
        response = await client.generate(messages=messages, tools=tools)

        print(f"Response: {response.content}")
        print(f"Thinking: {response.thinking}")
        print(f"Tool calls: {response.tool_calls}")

        if response.tool_calls:
            assert len(response.tool_calls) > 0
            assert response.tool_calls[0].function.name == "get_weather"
            print("✅ OpenAI tool calling test passed")
        else:
            print("⚠️  Warning: LLM didn't use tools, but request succeeded")

        return True
    except Exception as e:
        print(f"❌ OpenAI tool calling test failed: {e}")
        import traceback

        traceback.print_exc()
        return False