hellokawei commited on
Commit
2c757d1
·
verified ·
1 Parent(s): b8c1378

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -2
app.py CHANGED
@@ -3,5 +3,18 @@ import os
3
  from openai import OpenAI
4
 
5
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
6
- openai_client = OpenAI(api_key=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)
7
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  from openai import OpenAI
4
 
5
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
6
+ openai_client = OpenAI(api_key="OPENAI_API_KEY")
7
+ def generate_response(prompt, model_provider, temperature, top_p, max_tokens, repetition_penalty):
8
+ try:
9
+ response = openai_client.chat.completions.create(
10
+ model="gpt-3.5-turbo", # or another model of your choice
11
+ messages=[{"role": "user", "content": prompt}],
12
+ temperature=temperature,
13
+ top_p=top_p,
14
+ max_tokens=max_tokens,
15
+ presence_penalty=repetition_penalty,
16
+ stream=False
17
+ )
18
+ return response.choices[0].message.content.strip()
19
+ except Exception as e:
20
+ return f"OpenAI API Error: {str(e)}"