ryan/simbarag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

i only ship bugs

Browse Source
This commit is contained in:
Ryan Chen
2025-10-08 21:13:15 -04:00
parent 64e286e623
commit 166ffb4c09
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
llm.py
View File

@@ -29,13 +29,13 @@ class LLMClient:
system_prompt: str, system_prompt: str,
): ):
if self.PROVIDER == "ollama": if self.PROVIDER == "ollama":
response = ollama_client.chat( response = self.ollama_client.chat(
model="gemma3:4b", model="gemma3:4b",
prompt=prompt, prompt=prompt,
) )
output = response["response"] output = response["response"]
elif self.PROVIDER == "openai": elif self.PROVIDER == "openai":
response = openai_client.chat.completions.create( response = self.openai_client.chat.completions.create(
model="gpt-4o-mini", model="gpt-4o-mini",
messages=[ messages=[
{ {