koenieeeee
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
|
|
import os
|
|
|
|
import openai
|
|
openai.api_key = "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"
|
|
|
|
from prompt_techniques import LLM_prompt_technique
|
|
from logger import ResearchLogger
|
|
|
|
|
|
#https://github.com/zauberzeug/nicegui/blob/main/examples/progress/main.py
|
|
|
|
class LLM_Communicator():
|
|
|
|
def __init__(self) -> None:
|
|
self.client = None
|
|
self.question = None
|
|
self.answer = None
|
|
|
|
def write_log_output(self, log: ResearchLogger):
|
|
if log != None:
|
|
|
|
log_contents = f"""
|
|
|
|
## Input prompt, technique: {self.question.name()}
|
|
{str(self.question)}
|
|
|
|
## Rimay Output
|
|
```
|
|
{self.answer}
|
|
```
|
|
"""
|
|
|
|
log.append_result(log_contents)
|
|
|
|
self.question = None
|
|
self.answer = None
|
|
else:
|
|
print("LLM_Communicator: Error no logger found")
|
|
|
|
|
|
def ask_llm_to_convert(self, question: LLM_prompt_technique, temp):
|
|
self.question = None
|
|
self.answer = None
|
|
|
|
self.question = question
|
|
print("Temperature? " + temp)
|
|
myTemp = temp
|
|
|
|
response = openai.ChatCompletion.create(
|
|
model="gpt-3.5-turbo", # engine = "deployment_name".
|
|
temperature=eval(myTemp),
|
|
messages=[
|
|
|
|
{"role": "user", "content": str(question)},
|
|
|
|
]
|
|
)
|
|
anwer_content = response['choices'][0]['message']['content']
|
|
|
|
self.answer = anwer_content
|
|
|
|
return anwer_content
|