add paraGenerator

This commit is contained in:
podiukov.iv
2026-04-23 15:35:34 +05:00
parent c84bb64551
commit 3e0a0ee403
5 changed files with 90 additions and 8 deletions

33
paraGenerator.py Normal file
View File

@@ -0,0 +1,33 @@
# 2 модуль - Генератор парафраза на основе llama.cpp
from openai import OpenAI
class ParaphraseGenerator:
"""
Класс для использования генеративной модели на базе llama.cpp
"""
def __init__(self):
self.client = OpenAI(base_url="http://127.0.0.1:8080", api_key="")
def generate(self, srcText, entities):
"""
Генерация парафраза по промпту.
Использование: generate(<исходный текст>, <сущности>)
"""
entityList = ', '.join(entity['text'] for entity in entities) # генерация списка выделенных сущностей
prompt = 'Перефразируй следующий текст, сохранив все именованные сущности "' + entityList + '" в точности такими же, как в оригинале. ' \
'Не изменяй эти фрагменты текста, не заменяй их синонимами, не переставляй слова внутри них. ' \
'Можешь изменять грамматическую структуру предложения, порядок слов, использовать синонимы для' \
'остальных частей текста, но именованные сущности должны остаться неизменными. Исходный текст: "' + srcText + '"'
try:
response = self.client.chat.completions.create(
model = "",
messages = [{"role": "user", "content": prompt}],
temperature = 0.8,
top_p = 0.95,
max_tokens = 200
)
return response.choices[0].message.content
except Exception as e:
print(f"Ошибка генерации: {e}")
return None