32 lines
883 B
Python
32 lines
883 B
Python
import openai
|
|
|
|
class Llm:
|
|
def __init__(self, apiKey:str):
|
|
self.client = openai.OpenAI(
|
|
api_key=apiKey,
|
|
base_url='https://api.intelligence.io.solutions/api/v1/'
|
|
)
|
|
|
|
def generate(self, model:str, systemPrompt:str, userPrompt:str, temp:float):
|
|
'''
|
|
prompt[user_promtp, system_ptompt]
|
|
Function generate text by prompt
|
|
'''
|
|
|
|
# Получаем ответ от нейросети
|
|
response = self.client.chat.completions.create(
|
|
model=model,
|
|
messages=[
|
|
{'role': 'system', 'content': systemPrompt},
|
|
{'role': 'user', 'content': userPrompt},
|
|
],
|
|
temperature=temp,
|
|
stream=False
|
|
)
|
|
|
|
# Достаем текст
|
|
text = str(response.choices[0].message.content)
|
|
|
|
return text, text
|
|
|