refactoring

This commit is contained in:
2025-02-11 11:19:52 -05:00
parent 5ee99664f3
commit 650c12d073
11 changed files with 287 additions and 234 deletions

View File

@ -12,7 +12,7 @@ class GPT:
api_key=self.api_key,
timeout=10.0,
)
self.default_system_prompt = "You are a helpful assistant who will provide tidbits of info on songs the user may listen to."
self.default_system_prompt = "You are a helpful assistant who will provide only totally accurate tidbits of info on the specific songs the user may listen to."
async def get_completion(self, prompt: str, system_prompt: Optional[str] = None) -> None:
if not system_prompt:
@ -29,5 +29,6 @@ class GPT:
}
],
model="gpt-4o-mini",
temperature=0.35,
)
return chat_completion.choices[0].message.content