Code update

.chainlit/config.toml

@@ -33,7 +33,7 @@ edit_message = true
 
 # Authorize users to spontaneously upload files with messages
 [features.spontaneous_file_upload]
-    enabled = true
+    enabled = false
     # Define accepted file types using MIME types
     # Examples:
     # 1. For specific file types:
@@ -53,9 +53,9 @@ edit_message = true
 
 [UI]
 # Name of the assistant.
-name = "Assistant"
+name = "Docuswift Legal Search"
 
-# default_theme = "dark"
+default_theme = "light"
 
 # layout = "wide"
 

.gitignore

@@ -5,7 +5,8 @@ __pycache__/
 
 # C extensions
 *.so
-*.json
+translations/
+# *.json
 
 # Distribution / packaging
 .Python

aisearch_v1.py

@@ -29,6 +29,12 @@ az_client = AsyncAzureOpenAI(
         azure_endpoint=os.environ.get('AZURE_OPENAI_ENDPOINT')
     )
 
+az_client_in = AsyncAzureOpenAI(
+        api_key=os.environ.get('AZURE_OPENAI_API_KEY_IN'),
+        api_version="2024-12-01-preview",  # Use the appropriate API version
+        azure_endpoint=os.environ.get('AZURE_OPENAI_ENDPOINT_IN')
+    )
+
 class ReportResponse(BaseModel):
     title : str
     justification : str
@@ -123,11 +129,11 @@ class aiSearch:
         </output format>"""
 
     async def gpt4omini(self,query):
-        chat_completion = await az_client.chat.completions.create(
+        chat_completion = await az_client_in.chat.completions.create(
             messages=[
                 {
                     "role" : "system",
-                    "content" : self.research_prompt
+                    "content" : self.sp_final_report
                 },
                 {
                     "role": "user",
@@ -135,13 +141,13 @@ class aiSearch:
                 }
             ],
             stream=True,
-            model="gpt4o-mini",
-            # temperature=0,
+            model="gpt-4o-mini",
+            temperature=0,
             # reasoning_effort='low',
             # response_format=ReportResponse
         )
 
-        return chat_completion.choices[0].message.content
+        return chat_completion
 
     async def o3_mini(self,query):
         chat_completion = await az_client.beta.chat.completions.parse(
@@ -307,5 +313,5 @@ class aiSearch:
         reports = [resp for resp in batch_responses]
         print(reports)
 
-        stream_responses = await self.gemini(str(reports))
+        stream_responses = await self.gpt4omini(str(reports))
         return stream_responses

app.py

@@ -34,9 +34,13 @@ async def main(message: cl.Message):
     # await cl.Message(content=result).send()
 
     # for stream in streams:
-    async for part in streams:
-        if token := part.text or "":
-            await msg.stream_token(token)
+    # async for part in streams:
+    #     if token := part.text or "":
+    #         await msg.stream_token(token)
+
+    async for update in streams:
+        if update.choices:
+            await msg.stream_token(update.choices[0].delta.content or "")
     
     # # After streaming is complete, add citations if available
     # if sources and len(sources) > 0:

public/theme.json

@@ -0,0 +1,68 @@
+{
+    "custom_fonts": [],
+    "variables": {
+        "light": {
+            "--font-sans": "'Inter', sans-serif",
+            "--font-mono": "source-code-pro, Menlo, Monaco, Consolas, 'Courier New', monospace",
+            "--background": "0 0% 100%",
+            "--foreground": "0 0% 5%",
+            "--card": "0 0% 100%",
+            "--card-foreground": "0 0% 5%",
+            "--popover": "0 0% 100%",
+            "--popover-foreground": "0 0% 5%",
+            "--primary": "230, 100%, 50%",
+            "--primary-foreground": "0 0% 100%",
+            "--secondary": "210 40% 96.1%",
+            "--secondary-foreground": "222.2 47.4% 11.2%",
+            "--muted": "0 0% 90%",
+            "--muted-foreground": "0 0% 36%",
+            "--accent": "0 0% 95%",
+            "--accent-foreground": "222.2 47.4% 11.2%",
+            "--destructive": "0 84.2% 60.2%",
+            "--destructive-foreground": "210 40% 98%",
+            "--border": "0 0% 90%",
+            "--input": "0 0% 90%",
+            "--ring": "340 92% 52%",
+            "--radius": "0.75rem",
+            "--sidebar-background": "0 0% 98%",
+            "--sidebar-foreground": "240 5.3% 26.1%",
+            "--sidebar-primary": "240 5.9% 10%",
+            "--sidebar-primary-foreground": "0 0% 98%",
+            "--sidebar-accent": "240 4.8% 95.9%",
+            "--sidebar-accent-foreground": "240 5.9% 10%",
+            "--sidebar-border": "220 13% 91%",
+            "--sidebar-ring": "217.2 91.2% 59.8%"
+        },
+        "dark": {
+            "--font-sans": "'Inter', sans-serif",
+            "--font-mono": "source-code-pro, Menlo, Monaco, Consolas, 'Courier New', monospace",
+            "--background": "0 0% 13%",
+            "--foreground": "0 0% 93%",
+            "--card": "0 0% 18%",
+            "--card-foreground": "210 40% 98%",
+            "--popover": "0 0% 18%",
+            "--popover-foreground": "210 40% 98%",
+            "--primary": "340 92% 52%",
+            "--primary-foreground": "0 0% 100%",
+            "--secondary": "0 0% 19%",
+            "--secondary-foreground": "210 40% 98%",
+            "--muted": "0 1% 26%",
+            "--muted-foreground": "0 0% 71%",
+            "--accent": "0 0% 26%",
+            "--accent-foreground": "210 40% 98%",
+            "--destructive": "0 62.8% 30.6%",
+            "--destructive-foreground": "210 40% 98%",
+            "--border": "0 1% 26%",
+            "--input": "0 1% 26%",
+            "--ring": "340 92% 52%",
+            "--sidebar-background": "0 0% 9%",
+            "--sidebar-foreground": "240 4.8% 95.9%",
+            "--sidebar-primary": "224.3 76.3% 48%",
+            "--sidebar-primary-foreground": "0 0% 100%",
+            "--sidebar-accent": "0 0% 13%",
+            "--sidebar-accent-foreground": "240 4.8% 95.9%",
+            "--sidebar-border": "240 3.7% 15.9%",
+            "--sidebar-ring": "217.2 91.2% 59.8%"
+        }
+    }
+}