Upload OpenAi.py

llmservice/OpenAi.py

@@ -0,0 +1,27 @@
+import openai
+
+class OpenAi:
+    def __init__(self, apiKey, options={}, socket=None):
+        self.apiKey = apiKey
+        self.options = options
+        self.socket = socket
+        openai.api_key = self.apiKey
+
+    def ask(self, context, question, functions=[]):
+        try:
+            response = openai.chat.completions.create(
+                model="gpt-4o-mini-2024-07-18",
+                messages=context,
+                # functions=functions,
+                max_tokens=150,
+                n=1,
+                stop=None,
+                temperature=0.7
+            )
+            return {
+                'msg': response.choices[0].message.content,
+                'raw': response
+            }
+        except Exception as e:
+            print(f"Error in OpenAi.ask: {e}")
+            return None