Update services/base_ai_client.py

services/base_ai_client.py

@@ -0,0 +1,22 @@
+import os
+
+import google.generativeai as genai
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+class BaseAIClient:
+    def __init__(self,system,model_response):
+        genai.configure(api_key="AIzaSyDiIEOjW6ofy64JlYsHtRNXXXIx4C9Kje4")
+        self.system = system
+        self.model = model_response
+
+    def create(self, conversation):
+        model = genai.GenerativeModel('gemini-pro')
+        new_conversation = [
+            {"role": 'user', "parts": [self.system]},
+            {"role": 'model', "parts": [self.model]},
+        ]
+        new_conversation.extend(conversation)
+        return model.start_chat(history=new_conversation)