ajalisatgi commited on
Commit
22db157
·
verified ·
1 Parent(s): e509278

Update rag_gradio_app.py

Browse files
Files changed (1) hide show
  1. rag_gradio_app.py +7 -6
rag_gradio_app.py CHANGED
@@ -22,12 +22,13 @@ def rag_pipeline(question):
22
 
23
  # Generate AI response
24
  full_prompt = f"Context: {context}\\n\\nQuestion: {question}"
25
- response = openai.ChatCompletion.create(
26
- model="gpt-4",
27
- messages=[{"role": "user", "content": full_prompt}],
28
- max_tokens=300,
29
- temperature=0.7
30
- )
 
31
 
32
  return response['choices'][0]['message']['content'].strip(), retrieved_docs
33
 
 
22
 
23
  # Generate AI response
24
  full_prompt = f"Context: {context}\\n\\nQuestion: {question}"
25
+ import openai
26
+
27
+ response = openai.ChatCompletion.create(
28
+ model="gpt-4",
29
+ messages=[{"role": "system", "content": "You are a helpful assistant."},
30
+ {"role": "
31
+
32
 
33
  return response['choices'][0]['message']['content'].strip(), retrieved_docs
34