from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel

def test_openai_compatible():
    # Initialize the agent with DuckDuckGo search and OpenAI-compatible model
    agent = CodeAgent(
        tools=[DuckDuckGoSearchTool()],
        model=OpenAIServerModel(
            model_id="qwen2.5-coder-32b-instruct",  # or whatever model ID your server uses
            api_base="http://localhost:1234/v1",  # replace with your server URL
            api_key="lm-studio",  # replace with your API key
            temperature=0.7
        )
    )

    # Test the agent with a query
    query = "How many seconds would it take for a leopard at full speed to run through Pont des Arts?"
    result = agent.run(query)

    # Print the result
    print(f"Query: {query}")
    print(f"Response: {result}")

if __name__ == "__main__":
    test_openai_compatible()