my code is here:
import os
from cerebras.cloud.sdk import Cerebras
class chattoAI:
def __init__(self, prompt: str):
self.prompt = prompt
def chatToCerebras(self):
client = Cerebras(
api_key=os.environ.get("My API")
)
completion = client.chat.completions.create(
messages=[{'role':'user', 'content':self.prompt}],
model='zai-glm-4.7',
max_completion_tokens=1024,
temperature=0.2
top_p=1,
stream=False
)
return completion.choices[0].message.content
Top comments (0)