working version. something wrong with icons
This commit is contained in:
@@ -0,0 +1,3 @@
|
||||
{
|
||||
"prompt": "Используй ТОЛЬКО текст из транскрибации. Любая информация, которой нет в транскрибации, запрещена. Если какой-то факт или определение раскрыт не до конца, или лектор сказал что-то неполно, или тему нужно дополнить для лучшего понимания — вставляй дополнение сразу в нужное место, но только под отдельной подписью: AI overview: текст дополнения (confidence: low|medium|high) Ответ должен быть строго в формате Markdown (.md). Никаких 'привет', 'вот результат' и т. п. — только готовый текст. Все цитаты из транскрибации — буквально, максимум 25 слов подряд."
|
||||
}
|
||||
@@ -0,0 +1,17 @@
|
||||
import openai
|
||||
|
||||
|
||||
class LLMrequest:
|
||||
def __init__(self, api_key: str, model_name: str, base_url: str = None):
|
||||
if base_url:
|
||||
self.client = openai.OpenAI(api_key=api_key, base_url=base_url)
|
||||
else:
|
||||
self.client = openai.OpenAI(api_key=api_key)
|
||||
self.model = model_name
|
||||
|
||||
def get_response(self, prompt: str) -> str:
|
||||
response = self.client.chat.completions.create(
|
||||
model=self.model,
|
||||
messages=[{"role": "user", "content": prompt}],
|
||||
)
|
||||
return response.choices[0].message.content
|
||||
@@ -0,0 +1,9 @@
|
||||
def save_to_file(
|
||||
thing: str | dict,
|
||||
path: str,
|
||||
):
|
||||
if isinstance(thing, dict):
|
||||
pass
|
||||
elif isinstance(thing, str):
|
||||
with open(path, "w") as outputfile:
|
||||
outputfile.write(thing)
|
||||
Reference in New Issue
Block a user