cai/examples/model_providers
Víctor Mayoral Vilches 0f8c357231 Restore structure and place litellm integraton under model_providers
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-24 18:39:17 +01:00
..
README.md Restore structure and place litellm integraton under model_providers 2025-03-24 18:39:17 +01:00
custom_example_agent.py Update custom models to use tools 2025-03-13 13:10:26 -04:00
custom_example_global.py Update custom models to use tools 2025-03-13 13:10:26 -04:00
custom_example_provider.py Update custom models to use tools 2025-03-13 13:10:26 -04:00
litellm.py Restore structure and place litellm integraton under model_providers 2025-03-24 18:39:17 +01:00
litellm_config.yaml Restore structure and place litellm integraton under model_providers 2025-03-24 18:39:17 +01:00

README.md

Custom LLM providers

The examples in this directory demonstrate how you might use a non-OpenAI LLM provider. To run them, first set a base URL, API key and model.

export EXAMPLE_BASE_URL="..."
export EXAMPLE_API_KEY="..."
export EXAMPLE_MODEL_NAME"..."

Then run the examples, e.g.:

python examples/model_providers/custom_example_provider.py

Loops within themselves,
Function calls its own being,
Depth without ending.

LiteLLM Proxy Server integration

LiteLLM integration helps out switch between models easily and rapidly. This is easy to integrate via AsyncOpenAI:

# launch server proxy with your configuration
litellm --config examples/model_providers/litellm_config.yaml

# then use the proxy via the SDK
python3 examples/agent_patterns/litellm.py

Testing the proxy server

Testing some basic models against proxy to verify it's operational:

# qwen2.5:14b
curl -s http://localhost:4000/v1/chat/completions -H "Content-Type: application/json" -d '{"model": "qwen2.5:14b", "messages": [{"role": "user", "content": "Say hi"}], "max_tokens": 10}' | jq

# claude-3-7
curl -s http://localhost:4000/v1/chat/completions -H "Content-Type: application/json" -d '{"model": "claude-3-7", "messages": [{"role": "user", "content": "Say hi"}], "max_tokens": 10}' | jq

# gpt-4o
curl -s http://localhost:4000/v1/chat/completions -H "Content-Type: application/json" -d '{"model": "gpt-4o", "messages": [{"role": "user", "content": "Say hi"}], "max_tokens": 10}' | jq