call_openai

utils.py

@@ -8,6 +8,7 @@ from langchain.embeddings import OpenAIEmbeddings
 from langchain.prompts.prompt import PromptTemplate
 from langchain.embeddings.base import Embeddings
 from langchain.llms.base import LLM
+import requests
 
 import gradio as gr
 import datetime
@@ -41,7 +42,23 @@ def set_openai_api_key(api_key):
         return llm, base_embeddings
     return None, None
     
-
+def call_openai(system_prompt,ques):
+    headers = {
+    'Content-Type': 'application/json',
+    'Authorization': 'Bearer ' + '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'}
+
+    json_data = {
+        'model': 'gpt-3.5-turbo',
+        'messages': [
+            {
+                'role': 'system',
+                'content': system_prompt,},
+            {"role": "user", "content": ques}
+        ],
+        'temperature': 0.3}
+    response = requests.post('https://api.openai.com/v1/chat/completions', headers=headers, json=json_data)
+
+    return response.json()['choices'][0]['message']['content']
     
     
 def add_text(history, text,llm):