litellm-mirror/snowflake_testing.py
2025-03-03 01:34:44 -05:00

9 lines
No EOL
276 B
Python

import os
from litellm import completion
os.environ["SNOWFLAKE_ACCOUNT_ID"] = "YOUR ACCOUNT"
os.environ["SNOWFLAKE_JWT"] = "YOUR JWT"
messages = [{"role": "user", "content": "Write me a poem about the blue sky"}]
completion(model="snowflake/mistral-7b", messages=messages)