Skip to main content

Sarvam.ai

LiteLLM supports all the text models from Sarvam ai

Usage​

import os
from litellm import completion

# Set your Sarvam API key
os.environ["SARVAM_API_KEY"] = ""

messages = [{"role": "user", "content": "Hello"}]

response = completion(
    model="sarvam/sarvam-m",
    messages=messages,
)
print(response)

Usage with LiteLLM Proxy Server​

Here's how to call a Sarvam.ai model with the LiteLLM Proxy Server

  1. Modify the config.yaml:

    model_list:
      - model_name: my-model
        litellm_params:
          model: sarvam/<your-model-name>  # add sarvam/ prefix to route as Sarvam provider
          api_key: api-key                 # api key to send your model

  2. Start the proxy:

    $ litellm --config /path/to/config.yaml

  3. Send a request to LiteLLM Proxy Server:

    import openai

    client = openai.OpenAI(
        api_key="sk-1234",             # pass litellm proxy key, if you're using virtual keys
        base_url="http://0.0.0.0:4000" # litellm-proxy-base url
    )

    response = client.chat.completions.create(
        model="my-model",
        messages=[
            {
                "role": "user",
                "content": "what llm are you"
            }
        ],
    )

    print(response)