adityamanwatkar commited on
Commit
4df72e6
·
verified ·
1 Parent(s): dd2e19c

Update src/app/services/llm_service.py

Browse files
Files changed (1) hide show
  1. src/app/services/llm_service.py +15 -0
src/app/services/llm_service.py CHANGED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from openai import OpenAI
2
+ from app.config import OPENAI_API_KEY, MODEL
3
+
4
+ client = OpenAI(api_key=OPENAI_API_KEY)
5
+
6
+ def call_gpt(system_role, prompt, temperature=0.3):
7
+ response = client.chat.completions.create(
8
+ model=MODEL,
9
+ messages=[
10
+ {"role": "system", "content": system_role},
11
+ {"role": "user", "content": prompt}
12
+ ],
13
+ temperature=temperature
14
+ )
15
+ return response.choices[0].message.content