跳至主要內容

PublicAI

總覽

屬性詳細資訊
說明PublicAI 提供大型語言模型,包括 swiss-ai apertus 模型等關鍵模型。
LiteLLM 提供者路由publicai/
提供者文件連結PublicAI ↗
Base URLhttps://platform.publicai.co/
支援的操作/chat/completions


https://platform.publicai.co/

我們支援所有 PublicAI 模型,只要在送出 completion 請求時將 publicai/ 設為前綴即可

必要變數

Environment Variables
os.environ["PUBLICAI_API_KEY"] = ""  # your PublicAI API key

您可以使用以下方式覆寫 base url:

os.environ["PUBLICAI_API_BASE"] = "https://platform.publicai.co/v1"

使用方式 - LiteLLM Python SDK

非串流

PublicAI Non-streaming Completion
import os
import litellm
from litellm import completion

os.environ["PUBLICAI_API_KEY"] = "" # your PublicAI API key

messages = [{"content": "Hello, how are you?", "role": "user"}]

# PublicAI call
response = completion(
model="publicai/swiss-ai/apertus-8b-instruct",
messages=messages
)

print(response)

串流

PublicAI Streaming Completion
import os
import litellm
from litellm import completion

os.environ["PUBLICAI_API_KEY"] = "" # your PublicAI API key

messages = [{"content": "Hello, how are you?", "role": "user"}]

# PublicAI call with streaming
response = completion(
model="publicai/swiss-ai/apertus-8b-instruct",
messages=messages,
stream=True
)

for chunk in response:
print(chunk)

使用方式 - LiteLLM Proxy

將以下內容加入您的 LiteLLM Proxy 設定檔:

config.yaml
model_list:
- model_name: swiss-ai-apertus-8b
litellm_params:
model: publicai/swiss-ai/apertus-8b-instruct
api_key: os.environ/PUBLICAI_API_KEY

- model_name: swiss-ai-apertus-70b
litellm_params:
model: publicai/swiss-ai/apertus-70b-instruct
api_key: os.environ/PUBLICAI_API_KEY

啟動您的 LiteLLM Proxy 伺服器:

Start LiteLLM Proxy
litellm --config config.yaml

# RUNNING on http://0.0.0.0:4000
PublicAI via Proxy - Non-streaming
from openai import OpenAI

# Initialize client with your proxy URL
client = OpenAI(
base_url="http://localhost:4000", # Your proxy URL
api_key="your-proxy-api-key" # Your proxy API key
)

# Non-streaming response
response = client.chat.completions.create(
model="swiss-ai-apertus-8b",
messages=[{"role": "user", "content": "hello from litellm"}]
)

print(response.choices[0].message.content)
PublicAI via Proxy - Streaming
from openai import OpenAI

# Initialize client with your proxy URL
client = OpenAI(
base_url="http://localhost:4000", # Your proxy URL
api_key="your-proxy-api-key" # Your proxy API key
)

# Streaming response
response = client.chat.completions.create(
model="swiss-ai-apertus-8b",
messages=[{"role": "user", "content": "hello from litellm"}],
stream=True
)

for chunk in response:
if chunk.choices[0].delta.content is not None:
print(chunk.choices[0].delta.content, end="")

如需有關使用 LiteLLM Proxy 的更詳細資訊,請參閱 LiteLLM Proxy 文件

🚅
LiteLLM Enterprise
為正式環境打造的 SSO/SAML、稽核記錄、支出追蹤、多團隊管理與防護欄。
深入瞭解 →