30 lines
851 B
Python
Executable File
30 lines
851 B
Python
Executable File
#!/home/hollorol/miniconda3/envs/llm/bin/python
|
|
|
|
import subprocess as sp
|
|
from openai import OpenAI
|
|
import sys
|
|
|
|
API_KEY = sp.getoutput("pass show openai_apikey")
|
|
client = OpenAI(api_key=API_KEY)
|
|
temperature = 0.4
|
|
|
|
def generate(text_message):
|
|
prompt = f"""
|
|
You are an AI that generates text based on the prompt. You have to use the user provided text as a base, all text should be continiuos and coherent.
|
|
"""
|
|
|
|
response = client.chat.completions.create(
|
|
model="gpt-4o-mini",
|
|
messages=[{"role": "system", "content": prompt},
|
|
{"role": "user", "content": text_message}],
|
|
temperature=temperature # Adjust for creativity
|
|
)
|
|
|
|
response = response.choices[0].message.content
|
|
print(response)
|
|
|
|
if __name__ == "__main__":
|
|
text_message = sys.stdin.read().strip()
|
|
generate(text_message)
|
|
|