cai/examples/model_providers
Víctor Mayoral Vilches 21048444c6 Minor changes for alignment
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-26 18:55:21 +01:00
..
README.md Fix error in example path 2025-03-24 19:47:13 +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 Minor changes for alignment 2025-03-26 18:55:21 +01:00
litellm_config.yaml Update docs to integrate CAI, minor refinements 2025-03-26 12:41:45 +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/model_providers/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