Newchat context

aistarterkit/settings.py

@@ -187,6 +187,7 @@
 OPENAI_API_TYPE = os.getenv("OPENAI_API_TYPE", "openai")
 OPENAI_API_VERSION = os.getenv("OPENAI_API_VERSION", "2024-10-21")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+OPENAI_API_BASE = os.getenv("OPENAI_API_BASE")  # Add this line
 
 # Azure OpenAI Settings
 AZURE_OPENAI_API_KEY = os.getenv("AZURE_OPENAI_API_KEY")

chat/ai/agent.py

@@ -21,7 +21,10 @@
         azure_endpoint=settings.AZURE_OPENAI_ENDPOINT,
     )
 else:
-    client = OpenAI()
+    client = OpenAI(
+        api_key=settings.OPENAI_API_KEY,
+        base_url=settings.OPENAI_API_BASE
+    )
 
 
 class Agent:
@@ -119,3 +122,34 @@ def _update_history(self, role, content):
             Message.objects.create(
                 thread=self.thread, user=self.thread.user, content=content, role=role
             )
+
+    def generate_title(self, conversation):
+        """Generates a title for the conversation.
+
+        Args:
+            conversation: The first exchange between user and assistant.
+
+        Returns:
+            A string containing the generated title.
+        """
+        prompt = """Based on the conversation below, generate a concise title (10 words or less) that captures the main topic.
+        Keep it simple and descriptive. Just return the title, nothing else.
+
+        Conversation:
+        {conversation}
+        """
+
+        messages = [
+            {"role": "system", "content": prompt.format(conversation=conversation)},
+        ]
+
+        try:
+            completion = client.chat.completions.create(
+                model=self.thread.model if self.thread else "gpt-3.5-turbo",
+                messages=messages,
+                temperature=0,
+                max_tokens=60
+            )
+            return completion.choices[0].message.content.strip()
+        except Exception as e:
+            return "New Chat"

chat/views.py

@@ -225,12 +225,12 @@ def thread_detail(request, pk):
 
 @login_required
 def create_thread(request):
-    # Generate a default name for the thread, e.g., "Chat on <current date>"
-    default_name = f"Chat on {timezone.now().strftime('%Y-%m-%d %H:%M:%S')}"
-
-    # Create a new thread with the default name
-    new_thread = Thread.objects.create(name=default_name, user=request.user)
-
+    # Create a new thread with a temporary name
+    new_thread = Thread.objects.create(
+        name="New Chat",
+        user=request.user
+    )
+    
     # Redirect the user to the new thread's detail page
     return redirect("thread_detail", pk=new_thread.pk)
 
@@ -250,14 +250,19 @@ def new_message(request, pk):
     thread = get_object_or_404(Thread, pk=pk)
     if request.method == "POST":
         form = MessageForm(request.POST)
-        thread_form = ThreadForm(
-            request.POST, instance=thread
-        )  # Pass the current thread instance
+        thread_form = ThreadForm(request.POST, instance=thread)
         if form.is_valid() and thread_form.is_valid():
             message = form.save(commit=False)
-            thread = thread_form.save()  # Save the thread form to update the thread
+            thread = thread_form.save()
             agent = Agent(thread=thread, prompt=thread.prompt)
-            agent.chat(message.content)
+            ai_reply = agent.chat(message.content)
+
+            # Generate title if this is the first message in the thread
+            if thread.message_set.count() <= 2:  # User message + AI reply
+                conversation = f"User: {message.content}\nAssistant: {ai_reply}"
+                thread.name = agent.generate_title(conversation)
+                thread.save()
+
             return redirect("thread_detail", pk=thread.pk)
         else:
             print(form.errors)