Sweep: create a chat function like completion in upsonic_on_prem/utils/ai/ai.py (✓ Sandbox Passed)

upsonic_on_prem/utils/ai/ai.py

@@ -395,4 +395,11 @@ def generate_readme(self, top_library, summary_list):
 
         return result
 
+    def chat(self, input_text, model="gpt-3.5-turbo"):
+        try:
+            return self.gpt(input_text, model=model)
+        except Exception as e:
+            failed(f"Failed to generate chat response: {e}")
+            return None
+
 AI = AI_()