├── Knowmore ├── __init__.py ├── handlers │ ├── __init__.py │ ├── message_processor.py │ └── stream_handler.py ├── services │ ├── __init__.py │ ├── ai_provider.py │ ├── openai_service.py │ ├── claude_service.py │ ├── web_search_firecrawl.py │ └── search_orchestrator.py ├── asgi.py ├── wsgi.py ├── urls.py ├── utils.py ├── sse.py ├── views.py └── settings.py ├── ui ├── src │ ├── vite-env.d.ts │ ├── lib │ │ └── utils.ts │ ├── chat │ │ ├── chat-header.tsx │ │ ├── chat-input.tsx │ │ ├── model-selector.tsx │ │ ├── chat-source.tsx │ │ └── chat-message.tsx │ ├── main.tsx │ ├── components │ │ ├── ui │ │ │ ├── textarea.tsx │ │ │ ├── scroll-button.tsx │ │ │ ├── avatar.tsx │ │ │ ├── chat-container.tsx │ │ │ ├── badge.tsx │ │ │ ├── scroll-area.tsx │ │ │ ├── tooltip.tsx │ │ │ ├── button.tsx │ │ │ ├── card.tsx │ │ │ ├── code-block.tsx │ │ │ ├── message.tsx │ │ │ ├── markdown.tsx │ │ │ ├── prompt-input.tsx │ │ │ ├── dropdown-menu.tsx │ │ │ └── loader.tsx │ │ └── chat-source-placeholder.tsx │ ├── App.tsx │ └── index.css ├── logo.png ├── dark-logo.png ├── tsconfig.json ├── .gitignore ├── components.json ├── index.html ├── eslint.config.js ├── tsconfig.node.json ├── vite.config.ts ├── tsconfig.app.json ├── public │ └── vite.svg ├── package.json └── README.md ├── demo-knowmore.png ├── static ├── favicon.ico ├── favicon-16x16.png ├── favicon-32x32.png ├── apple-touch-icon.png ├── android-chrome-192x192.png ├── android-chrome-512x512.png └── site.webmanifest ├── .gitignore ├── requirements.txt ├── Makefile ├── Dockerfile ├── run_asgi.py ├── manage.py ├── templates └── react_app.html ├── LICENSE └── README.md /Knowmore/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Knowmore/handlers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Knowmore/services/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ui/src/vite-env.d.ts: -------------------------------------------------------------------------------- 1 | /// 2 | -------------------------------------------------------------------------------- /ui/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/ui/logo.png -------------------------------------------------------------------------------- /ui/dark-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/ui/dark-logo.png -------------------------------------------------------------------------------- /demo-knowmore.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/demo-knowmore.png -------------------------------------------------------------------------------- /static/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/static/favicon.ico -------------------------------------------------------------------------------- /static/favicon-16x16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/static/favicon-16x16.png -------------------------------------------------------------------------------- /static/favicon-32x32.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/static/favicon-32x32.png -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | node_modules/ 2 | static/CACHE/ 3 | static/dist/ 4 | .venv/ 5 | bun.lockb 6 | db.sqlite3 7 | .env 8 | -------------------------------------------------------------------------------- /static/apple-touch-icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/static/apple-touch-icon.png -------------------------------------------------------------------------------- /static/android-chrome-192x192.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/static/android-chrome-192x192.png -------------------------------------------------------------------------------- /static/android-chrome-512x512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadrosid/Knowmore/HEAD/static/android-chrome-512x512.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | anthropic 2 | openai 3 | Django==5.2.3 4 | daphne==4.2.0 5 | django-compressor==4.5.1 6 | django-environ 7 | requests 8 | -------------------------------------------------------------------------------- /ui/src/lib/utils.ts: -------------------------------------------------------------------------------- 1 | import { clsx, type ClassValue } from "clsx" 2 | import { twMerge } from "tailwind-merge" 3 | 4 | export function cn(...inputs: ClassValue[]) { 5 | return twMerge(clsx(inputs)) 6 | } 7 | -------------------------------------------------------------------------------- /static/site.webmanifest: -------------------------------------------------------------------------------- 1 | {"name":"","short_name":"","icons":[{"src":"/android-chrome-192x192.png","sizes":"192x192","type":"image/png"},{"src":"/android-chrome-512x512.png","sizes":"512x512","type":"image/png"}],"theme_color":"#ffffff","background_color":"#ffffff","display":"standalone"} -------------------------------------------------------------------------------- /ui/src/chat/chat-header.tsx: -------------------------------------------------------------------------------- 1 | export function ChatHeader() { 2 | return ( 3 |

4 |

5 | Knowmore 6 |

7 |

8 | ) 9 | } -------------------------------------------------------------------------------- /ui/tsconfig.json: -------------------------------------------------------------------------------- 1 | { 2 | "files": [], 3 | "references": [ 4 | { "path": "./tsconfig.app.json" }, 5 | { "path": "./tsconfig.node.json" } 6 | ], 7 | "compilerOptions": { 8 | "baseUrl": ".", 9 | "paths": { 10 | "@/*": ["./src/*"] 11 | } 12 | } 13 | } 14 | -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- 1 | .PHONY: build 2 | 3 | build: ui backend 4 | 5 | dev: ui backend run-backend 6 | 7 | ui: 8 | ./build_ui.sh 9 | 10 | serve-ui: 11 | cd ui && bun run dev 12 | 13 | backend: 14 | docker build . -t knowmore 15 | 16 | run-backend: 17 | docker run --env-file .env -p 7000:8000 knowmore -------------------------------------------------------------------------------- /ui/src/main.tsx: -------------------------------------------------------------------------------- 1 | import { StrictMode } from 'react' 2 | import { createRoot } from 'react-dom/client' 3 | import './index.css' 4 | import '../dark-logo.png' 5 | import '../logo.png' 6 | import App from './App.tsx' 7 | 8 | createRoot(document.getElementById('root')!).render( 9 | 10 | 11 | , 12 | ) 13 | -------------------------------------------------------------------------------- /ui/.gitignore: -------------------------------------------------------------------------------- 1 | # Logs 2 | logs 3 | *.log 4 | npm-debug.log* 5 | yarn-debug.log* 6 | yarn-error.log* 7 | pnpm-debug.log* 8 | lerna-debug.log* 9 | 10 | node_modules 11 | dist 12 | dist-ssr 13 | *.local 14 | 15 | # Editor directories and files 16 | .vscode/* 17 | !.vscode/extensions.json 18 | .idea 19 | .DS_Store 20 | *.suo 21 | *.ntvs* 22 | *.njsproj 23 | *.sln 24 | *.sw? 25 | -------------------------------------------------------------------------------- /Knowmore/asgi.py: -------------------------------------------------------------------------------- 1 | """ 2 | ASGI config for Knowmore project. 3 | 4 | It exposes the ASGI callable as a module-level variable named ``application``. 5 | 6 | For more information on this file, see 7 | https://docs.djangoproject.com/en/4.2/howto/deployment/asgi/ 8 | """ 9 | 10 | import os 11 | 12 | from django.core.asgi import get_asgi_application 13 | 14 | os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Knowmore.settings') 15 | 16 | application = get_asgi_application() 17 | -------------------------------------------------------------------------------- /Knowmore/wsgi.py: -------------------------------------------------------------------------------- 1 | """ 2 | WSGI config for Knowmore project. 3 | 4 | It exposes the WSGI callable as a module-level variable named ``application``. 5 | 6 | For more information on this file, see 7 | https://docs.djangoproject.com/en/4.2/howto/deployment/wsgi/ 8 | """ 9 | 10 | import os 11 | 12 | from django.core.wsgi import get_wsgi_application 13 | 14 | os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Knowmore.settings') 15 | 16 | application = get_wsgi_application() 17 | -------------------------------------------------------------------------------- /ui/components.json: -------------------------------------------------------------------------------- 1 | { 2 | "$schema": "https://ui.shadcn.com/schema.json", 3 | "style": "new-york", 4 | "rsc": false, 5 | "tsx": true, 6 | "tailwind": { 7 | "config": "", 8 | "css": "src/index.css", 9 | "baseColor": "neutral", 10 | "cssVariables": true, 11 | "prefix": "" 12 | }, 13 | "aliases": { 14 | "components": "@/components", 15 | "utils": "@/lib/utils", 16 | "ui": "@/components/ui", 17 | "lib": "@/lib", 18 | "hooks": "@/hooks" 19 | }, 20 | "iconLibrary": "lucide" 21 | } -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- 1 | FROM python:3.10-slim 2 | 3 | # Set environment variables 4 | ENV PYTHONDONTWRITEBYTECODE 1 5 | ENV PYTHONUNBUFFERED 1 6 | 7 | # Set work directory 8 | WORKDIR /app 9 | 10 | # Install dependencies 11 | COPY requirements.txt . 12 | RUN pip install --upgrade pip 13 | RUN pip install -r requirements.txt 14 | 15 | # Copy project 16 | COPY . . 17 | 18 | # Expose port 19 | EXPOSE 8000 20 | 21 | # Run the application with Daphne for proper streaming support 22 | CMD ["daphne", "-b", "0.0.0.0", "-p", "8000", "Knowmore.asgi:application"] -------------------------------------------------------------------------------- /ui/index.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | Knowmore 10 | 11 | 12 |

13 | 14 | 15 | 16 | -------------------------------------------------------------------------------- /run_asgi.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | """ 3 | Run the Django application with Daphne ASGI server for proper SSE streaming support. 4 | """ 5 | 6 | import os 7 | import sys 8 | 9 | if __name__ == "__main__": 10 | os.environ.setdefault("DJANGO_SETTINGS_MODULE", "Knowmore.settings") 11 | 12 | # Run with Daphne 13 | from daphne.cli import CommandLineInterface 14 | 15 | # Run on port 8000 16 | sys.argv = ["daphne", "-b", "127.0.0.1", "-p", "8000", "Knowmore.asgi:application"] 17 | 18 | cli = CommandLineInterface() 19 | cli.run(sys.argv[1:]) -------------------------------------------------------------------------------- /Knowmore/urls.py: -------------------------------------------------------------------------------- 1 | from django.contrib import admin 2 | from django.urls import path 3 | from django.conf import settings 4 | from django.conf.urls.static import static 5 | from .views import index, sse_stream, get_manifest, get_models 6 | 7 | urlpatterns = [ 8 | path('admin/', admin.site.urls), 9 | path('', index, name='react_app'), 10 | path('api/stream', sse_stream, name='sse_stream'), 11 | path('api/models', get_models, name='get_models'), 12 | path('manifest/', get_manifest, name='get_manifest'), 13 | ] 14 | 15 | if settings.DEBUG: 16 | urlpatterns += static(settings.STATIC_URL, document_root=settings.STATICFILES_DIRS[0]) 17 | -------------------------------------------------------------------------------- /ui/eslint.config.js: -------------------------------------------------------------------------------- 1 | import js from '@eslint/js' 2 | import globals from 'globals' 3 | import reactHooks from 'eslint-plugin-react-hooks' 4 | import reactRefresh from 'eslint-plugin-react-refresh' 5 | import tseslint from 'typescript-eslint' 6 | import { globalIgnores } from 'eslint/config' 7 | 8 | export default tseslint.config([ 9 | globalIgnores(['dist']), 10 | { 11 | files: ['**/*.{ts,tsx}'], 12 | extends: [ 13 | js.configs.recommended, 14 | tseslint.configs.recommended, 15 | reactHooks.configs['recommended-latest'], 16 | reactRefresh.configs.vite, 17 | ], 18 | languageOptions: { 19 | ecmaVersion: 2020, 20 | globals: globals.browser, 21 | }, 22 | }, 23 | ]) 24 | -------------------------------------------------------------------------------- /ui/tsconfig.node.json: -------------------------------------------------------------------------------- 1 | { 2 | "compilerOptions": { 3 | "tsBuildInfoFile": "./node_modules/.tmp/tsconfig.node.tsbuildinfo", 4 | "target": "ES2023", 5 | "lib": ["ES2023"], 6 | "module": "ESNext", 7 | "skipLibCheck": true, 8 | 9 | /* Bundler mode */ 10 | "moduleResolution": "bundler", 11 | "allowImportingTsExtensions": true, 12 | "verbatimModuleSyntax": true, 13 | "moduleDetection": "force", 14 | "noEmit": true, 15 | 16 | /* Linting */ 17 | "strict": true, 18 | "noUnusedLocals": true, 19 | "noUnusedParameters": true, 20 | "erasableSyntaxOnly": true, 21 | "noFallthroughCasesInSwitch": true, 22 | "noUncheckedSideEffectImports": true 23 | }, 24 | "include": ["vite.config.ts"] 25 | } 26 | -------------------------------------------------------------------------------- /manage.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | """Django's command-line utility for administrative tasks.""" 3 | import os 4 | import sys 5 | 6 | 7 | def main(): 8 | """Run administrative tasks.""" 9 | os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Knowmore.settings') 10 | try: 11 | from django.core.management import execute_from_command_line 12 | except ImportError as exc: 13 | raise ImportError( 14 | "Couldn't import Django. Are you sure it's installed and " 15 | "available on your PYTHONPATH environment variable? Did you " 16 | "forget to activate a virtual environment?" 17 | ) from exc 18 | execute_from_command_line(sys.argv) 19 | 20 | 21 | if __name__ == '__main__': 22 | main() 23 | -------------------------------------------------------------------------------- /ui/src/components/ui/textarea.tsx: -------------------------------------------------------------------------------- 1 | import * as React from "react" 2 | 3 | import { cn } from "@/lib/utils" 4 | 5 | function Textarea({ className, ...props }: React.ComponentProps<"textarea">) { 6 | return ( 7 |

15 |   )
16 | }
17 | 
18 | export { Textarea }
19 |

--------------------------------------------------------------------------------
/ui/vite.config.ts:
--------------------------------------------------------------------------------
 1 | import { defineConfig } from 'vite'
 2 | import react from '@vitejs/plugin-react'
 3 | import tailwindcss from '@tailwindcss/vite'
 4 | import path from "path"
 5 | 
 6 | export default defineConfig({
 7 |   plugins: [react(), tailwindcss()],
 8 |   resolve: {
 9 |     alias: {
10 |       "@": path.resolve(__dirname, "./src"),
11 |     },
12 |   },
13 |   build: {
14 |     outDir: '../static/dist',
15 |     assetsDir: 'assets',
16 |     emptyOutDir: true,
17 |     rollupOptions: {
18 |       output: {
19 |         entryFileNames: 'assets/[name]-[hash].js',
20 |         chunkFileNames: 'assets/[name]-[hash].js',
21 |         assetFileNames: 'assets/[name].[ext]'
22 |       }
23 |     }
24 |   },
25 |   server: {
26 |     port: 3000,
27 |     proxy: {
28 |       "/api": {
29 |         target: "http://127.0.0.1:7000",
30 |         changeOrigin: true
31 |       },
32 |     },
33 |   },
34 |   base: '/static/dist/'
35 | })
36 |

--------------------------------------------------------------------------------
/ui/tsconfig.app.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "compilerOptions": {
 3 |     "tsBuildInfoFile": "./node_modules/.tmp/tsconfig.app.tsbuildinfo",
 4 |     "target": "ES2022",
 5 |     "useDefineForClassFields": true,
 6 |     "lib": ["ES2022", "DOM", "DOM.Iterable"],
 7 |     "module": "ESNext",
 8 |     "skipLibCheck": true,
 9 | 
10 |     /* Bundler mode */
11 |     "moduleResolution": "bundler",
12 |     "allowImportingTsExtensions": true,
13 |     "verbatimModuleSyntax": true,
14 |     "moduleDetection": "force",
15 |     "noEmit": true,
16 |     "jsx": "react-jsx",
17 | 
18 |     /* Linting */
19 |     "strict": true,
20 |     "noUnusedLocals": true,
21 |     "noUnusedParameters": true,
22 |     "erasableSyntaxOnly": true,
23 |     "noFallthroughCasesInSwitch": true,
24 |     "noUncheckedSideEffectImports": true,
25 |     "baseUrl": ".",
26 |     "paths": {
27 |       "@/*": [
28 |         "./src/*"
29 |       ]
30 |     }
31 |   },
32 |   "include": ["src"]
33 | }
34 |

--------------------------------------------------------------------------------
/Knowmore/handlers/message_processor.py:
--------------------------------------------------------------------------------
 1 | def format_messages(messages):
 2 |     """Format messages to only include role and content (ignore parts)"""
 3 |     formatted_messages = []
 4 |     for msg in messages:
 5 |         if 'parts' in msg:
 6 |             # Extract text content from parts
 7 |             content = ""
 8 |             for part in msg['parts']:
 9 |                 if isinstance(part, dict) and 'text' in part:
10 |                     content += part['text']
11 |                 elif isinstance(part, str):
12 |                     content += part
13 |             formatted_messages.append({
14 |                 "role": msg['role'],
15 |                 "content": content
16 |             })
17 |         else:
18 |             # Message already in correct format
19 |             formatted_messages.append({
20 |                 "role": msg['role'],
21 |                 "content": msg.get('content', '')
22 |             })
23 |     return formatted_messages

--------------------------------------------------------------------------------
/Knowmore/utils.py:
--------------------------------------------------------------------------------
 1 | import os
 2 | import re
 3 | from django.conf import settings
 4 | 
 5 | def get_vite_assets():
 6 |     """
 7 |     Parse the Vite-generated index.html to extract asset paths
 8 |     """
 9 |     try:
10 |         vite_html_path = os.path.join(settings.BASE_DIR, 'static', 'dist', 'index.html')
11 |         
12 |         with open(vite_html_path, 'r') as f:
13 |             content = f.read()
14 |         
15 |         # Extract CSS file
16 |         css_match = re.search(r'href="([^"]*\.css)"', content)
17 |         css_file = css_match.group(1) if css_match else None
18 |         
19 |         # Extract JS file  
20 |         js_match = re.search(r'src="([^"]*\.js)"', content)
21 |         js_file = js_match.group(1) if js_match else None
22 |         
23 |         return {
24 |             'css': css_file.replace('/static/', '') if css_file else None,
25 |             'js': js_file.replace('/static/', '') if js_file else None
26 |         }
27 |     except FileNotFoundError:
28 |         return {'css': None, 'js': None}

--------------------------------------------------------------------------------
/templates/react_app.html:
--------------------------------------------------------------------------------
 1 | <!DOCTYPE html>
 2 | <html lang="en">
 3 |   <head>
 4 |     <meta charset="UTF-8" />
 5 |     {% load static %}
 6 |     <link rel="icon" type="image/png" href="{% static 'dist/assets/dark-logo.png' %}" media="(prefers-color-scheme: dark)">
 7 |     <link rel="icon" type="image/png" href="{% static 'dist/assets/logo.png' %}" media="(prefers-color-scheme: light)">
 8 |     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
 9 |     <title>Knowmore</title>
10 |     
11 |     
12 |     {% if assets.css %}
13 |     <link rel="stylesheet" crossorigin href="{% static assets.css %}" />
14 |     {% endif %}
15 |   </head>
16 |   <body>
17 |     <div id="root"></div>
18 |     {% if assets.js %}
19 |     <script type="module" crossorigin src="{% static assets.js %}"></script>
20 |     {% else %}
21 |     <div style="padding: 20px; text-align: center; color: #666;">
22 |       <h2>Frontend not built</h2>
23 |       <p>Please run: <code>cd frontend && npm run build</code></p>
24 |     </div>
25 |     {% endif %}
26 |   </body>
27 | </html>

--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
 1 | MIT License
 2 | 
 3 | Copyright (c) 2025 Ahmad Rosid
 4 | 
 5 | Permission is hereby granted, free of charge, to any person obtaining a copy
 6 | of this software and associated documentation files (the "Software"), to deal
 7 | in the Software without restriction, including without limitation the rights
 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 | 
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 | 
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.

--------------------------------------------------------------------------------
/ui/src/components/ui/scroll-button.tsx:
--------------------------------------------------------------------------------
 1 | import { Button, buttonVariants } from "@/components/ui/button"
 2 | import { cn } from "@/lib/utils"
 3 | import { type VariantProps } from "class-variance-authority"
 4 | import { ChevronDown } from "lucide-react"
 5 | import { useStickToBottomContext } from "use-stick-to-bottom"
 6 | 
 7 | export type ScrollButtonProps = {
 8 |   className?: string
 9 |   variant?: VariantProps<typeof buttonVariants>["variant"]
10 |   size?: VariantProps<typeof buttonVariants>["size"]
11 | } & React.ButtonHTMLAttributes<HTMLButtonElement>
12 | 
13 | function ScrollButton({
14 |   className,
15 |   variant = "outline",
16 |   size = "sm",
17 |   ...props
18 | }: ScrollButtonProps) {
19 |   const { isAtBottom, scrollToBottom } = useStickToBottomContext()
20 | 
21 |   return (
22 |     <Button
23 |       variant={variant}
24 |       size={size}
25 |       className={cn(
26 |         "h-10 w-10 rounded-full transition-all duration-150 ease-out",
27 |         !isAtBottom
28 |           ? "translate-y-0 scale-100 opacity-100"
29 |           : "pointer-events-none translate-y-4 scale-95 opacity-0",
30 |         className
31 |       )}
32 |       onClick={() => scrollToBottom()}
33 |       {...props}
34 |     >
35 |       <ChevronDown className="h-5 w-5" />
36 |     </Button>
37 |   )
38 | }
39 | 
40 | export { ScrollButton }
41 |

--------------------------------------------------------------------------------
/ui/src/components/ui/avatar.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as AvatarPrimitive from "@radix-ui/react-avatar"
 5 | 
 6 | import { cn } from "@/lib/utils"
 7 | 
 8 | function Avatar({
 9 |   className,
10 |   ...props
11 | }: React.ComponentProps<typeof AvatarPrimitive.Root>) {
12 |   return (
13 |     <AvatarPrimitive.Root
14 |       data-slot="avatar"
15 |       className={cn(
16 |         "relative flex size-8 shrink-0 overflow-hidden rounded-full",
17 |         className
18 |       )}
19 |       {...props}
20 |     />
21 |   )
22 | }
23 | 
24 | function AvatarImage({
25 |   className,
26 |   ...props
27 | }: React.ComponentProps<typeof AvatarPrimitive.Image>) {
28 |   return (
29 |     <AvatarPrimitive.Image
30 |       data-slot="avatar-image"
31 |       className={cn("aspect-square size-full", className)}
32 |       {...props}
33 |     />
34 |   )
35 | }
36 | 
37 | function AvatarFallback({
38 |   className,
39 |   ...props
40 | }: React.ComponentProps<typeof AvatarPrimitive.Fallback>) {
41 |   return (
42 |     <AvatarPrimitive.Fallback
43 |       data-slot="avatar-fallback"
44 |       className={cn(
45 |         "bg-muted flex size-full items-center justify-center rounded-full",
46 |         className
47 |       )}
48 |       {...props}
49 |     />
50 |   )
51 | }
52 | 
53 | export { Avatar, AvatarImage, AvatarFallback }
54 |

--------------------------------------------------------------------------------
/Knowmore/services/ai_provider.py:
--------------------------------------------------------------------------------
 1 | from .claude_service import ClaudeService
 2 | from .openai_service import OpenAIService
 3 | 
 4 | 
 5 | class AIProviderFactory:
 6 |     @staticmethod
 7 |     def get_provider(model_name):
 8 |         if model_name.startswith('claude'):
 9 |             return ClaudeService()
10 |         elif model_name.startswith('gpt') or model_name.startswith('o4'):
11 |             return OpenAIService()
12 |         else:
13 |             # Default to Claude
14 |             return ClaudeService()
15 | 
16 |     @staticmethod
17 |     def get_supported_models():
18 |         return {
19 |             'claude': [
20 |                 {'id': 'claude-opus-4-20250514', 'name': 'Claude Opus 4'},
21 |                 {'id': 'claude-sonnet-4-20250514', 'name': 'Claude Sonnet 4'},
22 |                 {'id': 'claude-3-7-sonnet-20250219', 'name': 'Claude Sonnet 3.7'},
23 |                 {'id': 'claude-3-5-sonnet-latest', 'name': 'Claude Sonnet 3.5'},
24 |                 {'id': 'claude-3-5-haiku-latest', 'name': 'Claude Haiku 3.5'},
25 |                 {'id': 'claude-3-5-sonnet-20240620', 'name': 'Claude Sonnet 3.5'}
26 |             ],
27 |             'openai': [
28 |                 {'id': 'gpt-4o-2024-08-06', 'name': 'GPT-4o'},
29 |                 {'id': 'o4-mini-2025-04-16', 'name': 'O4 Mini'},
30 |                 {'id': 'gpt-4.1-2025-04-14', 'name': 'GPT-4.1'}
31 |             ]
32 |         }

--------------------------------------------------------------------------------
/ui/public/vite.svg:
--------------------------------------------------------------------------------
1 | <svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="31.88" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 257"><defs><linearGradient id="IconifyId1813088fe1fbc01fb466" x1="-.828%" x2="57.636%" y1="7.652%" y2="78.411%"><stop offset="0%" stop-color="#41D1FF"></stop><stop offset="100%" stop-color="#BD34FE"></stop></linearGradient><linearGradient id="IconifyId1813088fe1fbc01fb467" x1="43.376%" x2="50.316%" y1="2.242%" y2="89.03%"><stop offset="0%" stop-color="#FFEA83"></stop><stop offset="8.333%" stop-color="#FFDD35"></stop><stop offset="100%" stop-color="#FFA800"></stop></linearGradient></defs><path fill="url(#IconifyId1813088fe1fbc01fb466)" d="M255.153 37.938L134.897 252.976c-2.483 4.44-8.862 4.466-11.382.048L.875 37.958c-2.746-4.814 1.371-10.646 6.827-9.67l120.385 21.517a6.537 6.537 0 0 0 2.322-.004l117.867-21.483c5.438-.991 9.574 4.796 6.877 9.62Z"></path><path fill="url(#IconifyId1813088fe1fbc01fb467)" d="M185.432.063L96.44 17.501a3.268 3.268 0 0 0-2.634 3.014l-5.474 92.456a3.268 3.268 0 0 0 3.997 3.378l24.777-5.718c2.318-.535 4.413 1.507 3.936 3.838l-7.361 36.047c-.495 2.426 1.782 4.5 4.151 3.78l15.304-4.649c2.372-.72 4.652 1.36 4.15 3.788l-11.698 56.621c-.732 3.542 3.979 5.473 5.943 2.437l1.313-2.028l72.516-144.72c1.215-2.423-.88-5.186-3.54-4.672l-25.505 4.922c-2.396.462-4.435-1.77-3.759-4.114l16.646-57.705c.677-2.35-1.37-4.583-3.769-4.113Z"></path></svg>

--------------------------------------------------------------------------------
/Knowmore/services/openai_service.py:
--------------------------------------------------------------------------------
 1 | import json
 2 | import environ
 3 | from openai import AsyncOpenAI
 4 | 
 5 | env = environ.Env(
 6 |     OPENAI_API_KEY=str,
 7 | )
 8 | 
 9 | class OpenAIService:
10 |     def __init__(self):
11 |         self.client = AsyncOpenAI(
12 |             api_key=env("OPENAI_API_KEY"),
13 |         )
14 | 
15 |     async def stream_response(self, messages, model="gpt-3.5-turbo", enable_web_search=False):
16 |         """Stream chat completion response without tool support"""
17 |         try:
18 |             stream_params = {
19 |                 "model": model,
20 |                 "messages": messages,
21 |                 "stream": True,
22 |                 "max_tokens": 1024,
23 |             }
24 |             
25 |             stream = await self.client.chat.completions.create(**stream_params)
26 |             
27 |             async for chunk in stream:
28 |                 choice = chunk.choices[0]
29 |                 delta = choice.delta
30 |                 
31 |                 # Handle regular text content
32 |                 if delta.content is not None:
33 |                     yield f'0:{json.dumps(delta.content)}\n'
34 |                 
35 |                 # Handle completion
36 |                 if choice.finish_reason is not None:
37 |                     yield f'd:{json.dumps({"finishReason": choice.finish_reason})}\n'
38 |                     
39 |         except Exception as e:
40 |             # Send error using 3: identifier
41 |             yield f'3:{json.dumps({"error": str(e)})}\n'

--------------------------------------------------------------------------------
/ui/package.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "name": "frontend",
 3 |   "private": true,
 4 |   "version": "0.0.0",
 5 |   "type": "module",
 6 |   "scripts": {
 7 |     "dev": "vite",
 8 |     "build": "tsc -b && vite build",
 9 |     "lint": "eslint .",
10 |     "preview": "vite preview"
11 |   },
12 |   "dependencies": {
13 |     "@ai-sdk/react": "^1.2.12",
14 |     "@radix-ui/react-avatar": "^1.1.10",
15 |     "@radix-ui/react-dropdown-menu": "^2.1.15",
16 |     "@radix-ui/react-scroll-area": "^1.2.9",
17 |     "@radix-ui/react-slot": "^1.2.3",
18 |     "@radix-ui/react-tooltip": "^1.2.7",
19 |     "@tailwindcss/typography": "^0.5.16",
20 |     "@tailwindcss/vite": "^4.1.11",
21 |     "class-variance-authority": "^0.7.1",
22 |     "clsx": "^2.1.1",
23 |     "lucide-react": "^0.523.0",
24 |     "marked": "^16.0.0",
25 |     "react": "^19.1.0",
26 |     "react-dom": "^19.1.0",
27 |     "react-markdown": "^10.1.0",
28 |     "remark-breaks": "^4.0.0",
29 |     "remark-gfm": "^4.0.1",
30 |     "shiki": "^3.7.0",
31 |     "swr": "^2.3.3",
32 |     "tailwind-merge": "^3.3.1",
33 |     "tailwindcss": "^4.1.11",
34 |     "tailwindcss-animate": "^1.0.7",
35 |     "use-stick-to-bottom": "^1.1.1"
36 |   },
37 |   "devDependencies": {
38 |     "@eslint/js": "^9.29.0",
39 |     "@types/node": "^24.0.4",
40 |     "@types/react": "^19.1.8",
41 |     "@types/react-dom": "^19.1.6",
42 |     "@vitejs/plugin-react": "^4.5.2",
43 |     "eslint": "^9.29.0",
44 |     "eslint-plugin-react-hooks": "^5.2.0",
45 |     "eslint-plugin-react-refresh": "^0.4.20",
46 |     "globals": "^16.2.0",
47 |     "tw-animate-css": "^1.3.4",
48 |     "typescript": "~5.8.3",
49 |     "typescript-eslint": "^8.34.1",
50 |     "vite": "^7.0.0"
51 |   }
52 | }
53 |

--------------------------------------------------------------------------------
/ui/src/components/ui/chat-container.tsx:
--------------------------------------------------------------------------------
 1 | import { cn } from "@/lib/utils"
 2 | import { StickToBottom } from "use-stick-to-bottom"
 3 | 
 4 | export type ChatContainerRootProps = {
 5 |   children: React.ReactNode
 6 |   className?: string
 7 | } & React.HTMLAttributes<HTMLDivElement>
 8 | 
 9 | export type ChatContainerContentProps = {
10 |   children: React.ReactNode
11 |   className?: string
12 | } & React.HTMLAttributes<HTMLDivElement>
13 | 
14 | export type ChatContainerScrollAnchorProps = {
15 |   className?: string
16 |   ref?: React.RefObject<HTMLDivElement>
17 | } & React.HTMLAttributes<HTMLDivElement>
18 | 
19 | function ChatContainerRoot({
20 |   children,
21 |   className,
22 |   ...props
23 | }: ChatContainerRootProps) {
24 |   return (
25 |     <StickToBottom
26 |       className={cn("flex overflow-y-auto scrollbar-hidden", className)}
27 |       resize="smooth"
28 |       initial="smooth"
29 |       role="log"
30 |       {...props}
31 |     >
32 |       {children}
33 |     </StickToBottom>
34 |   )
35 | }
36 | 
37 | function ChatContainerContent({
38 |   children,
39 |   className,
40 |   ...props
41 | }: ChatContainerContentProps) {
42 |   return (
43 |     <StickToBottom.Content
44 |       className={cn("flex w-full flex-col", className)}
45 |       {...props}
46 |     >
47 |       {children}
48 |     </StickToBottom.Content>
49 |   )
50 | }
51 | 
52 | function ChatContainerScrollAnchor({
53 |   className,
54 |   ...props
55 | }: ChatContainerScrollAnchorProps) {
56 |   return (
57 |     <div
58 |       className={cn("h-px w-full shrink-0 scroll-mt-4", className)}
59 |       aria-hidden="true"
60 |       {...props}
61 |     />
62 |   )
63 | }
64 | 
65 | export { ChatContainerRoot, ChatContainerContent, ChatContainerScrollAnchor }
66 |

--------------------------------------------------------------------------------
/ui/src/components/ui/badge.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | import { Slot } from "@radix-ui/react-slot"
 3 | import { cva, type VariantProps } from "class-variance-authority"
 4 | 
 5 | import { cn } from "@/lib/utils"
 6 | 
 7 | const badgeVariants = cva(
 8 |   "inline-flex items-center justify-center rounded-md border px-2 py-0.5 text-xs font-medium w-fit whitespace-nowrap shrink-0 [&>svg]:size-3 gap-1 [&>svg]:pointer-events-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive transition-[color,box-shadow] overflow-hidden",
 9 |   {
10 |     variants: {
11 |       variant: {
12 |         default:
13 |           "border-transparent bg-primary text-primary-foreground [a&]:hover:bg-primary/90",
14 |         secondary:
15 |           "border-transparent bg-secondary text-secondary-foreground [a&]:hover:bg-secondary/90",
16 |         destructive:
17 |           "border-transparent bg-destructive text-white [a&]:hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60",
18 |         outline:
19 |           "text-foreground [a&]:hover:bg-accent [a&]:hover:text-accent-foreground",
20 |       },
21 |     },
22 |     defaultVariants: {
23 |       variant: "default",
24 |     },
25 |   }
26 | )
27 | 
28 | function Badge({
29 |   className,
30 |   variant,
31 |   asChild = false,
32 |   ...props
33 | }: React.ComponentProps<"span"> &
34 |   VariantProps<typeof badgeVariants> & { asChild?: boolean }) {
35 |   const Comp = asChild ? Slot : "span"
36 | 
37 |   return (
38 |     <Comp
39 |       data-slot="badge"
40 |       className={cn(badgeVariants({ variant }), className)}
41 |       {...props}
42 |     />
43 |   )
44 | }
45 | 
46 | export { Badge, badgeVariants }
47 |

--------------------------------------------------------------------------------
/ui/src/components/ui/scroll-area.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | import * as ScrollAreaPrimitive from "@radix-ui/react-scroll-area"
 3 | 
 4 | import { cn } from "@/lib/utils"
 5 | 
 6 | function ScrollArea({
 7 |   className,
 8 |   children,
 9 |   ...props
10 | }: React.ComponentProps<typeof ScrollAreaPrimitive.Root>) {
11 |   return (
12 |     <ScrollAreaPrimitive.Root
13 |       data-slot="scroll-area"
14 |       className={cn("relative", className)}
15 |       {...props}
16 |     >
17 |       <ScrollAreaPrimitive.Viewport
18 |         data-slot="scroll-area-viewport"
19 |         className="focus-visible:ring-ring/50 size-full rounded-[inherit] transition-[color,box-shadow] outline-none focus-visible:ring-[3px] focus-visible:outline-1"
20 |       >
21 |         {children}
22 |       </ScrollAreaPrimitive.Viewport>
23 |       <ScrollBar />
24 |       <ScrollAreaPrimitive.Corner />
25 |     </ScrollAreaPrimitive.Root>
26 |   )
27 | }
28 | 
29 | function ScrollBar({
30 |   className,
31 |   orientation = "vertical",
32 |   ...props
33 | }: React.ComponentProps<typeof ScrollAreaPrimitive.ScrollAreaScrollbar>) {
34 |   return (
35 |     <ScrollAreaPrimitive.ScrollAreaScrollbar
36 |       data-slot="scroll-area-scrollbar"
37 |       orientation={orientation}
38 |       className={cn(
39 |         "flex touch-none p-px transition-colors select-none",
40 |         orientation === "vertical" &&
41 |           "h-full w-2.5 border-l border-l-transparent",
42 |         orientation === "horizontal" &&
43 |           "h-2.5 flex-col border-t border-t-transparent",
44 |         className
45 |       )}
46 |       {...props}
47 |     >
48 |       <ScrollAreaPrimitive.ScrollAreaThumb
49 |         data-slot="scroll-area-thumb"
50 |         className="bg-border relative flex-1 rounded-full"
51 |       />
52 |     </ScrollAreaPrimitive.ScrollAreaScrollbar>
53 |   )
54 | }
55 | 
56 | export { ScrollArea, ScrollBar }
57 |

--------------------------------------------------------------------------------
/ui/src/components/ui/tooltip.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as TooltipPrimitive from "@radix-ui/react-tooltip"
 5 | 
 6 | import { cn } from "@/lib/utils"
 7 | 
 8 | function TooltipProvider({
 9 |   delayDuration = 0,
10 |   ...props
11 | }: React.ComponentProps<typeof TooltipPrimitive.Provider>) {
12 |   return (
13 |     <TooltipPrimitive.Provider
14 |       data-slot="tooltip-provider"
15 |       delayDuration={delayDuration}
16 |       {...props}
17 |     />
18 |   )
19 | }
20 | 
21 | function Tooltip({
22 |   ...props
23 | }: React.ComponentProps<typeof TooltipPrimitive.Root>) {
24 |   return (
25 |     <TooltipProvider>
26 |       <TooltipPrimitive.Root data-slot="tooltip" {...props} />
27 |     </TooltipProvider>
28 |   )
29 | }
30 | 
31 | function TooltipTrigger({
32 |   ...props
33 | }: React.ComponentProps<typeof TooltipPrimitive.Trigger>) {
34 |   return <TooltipPrimitive.Trigger data-slot="tooltip-trigger" {...props} />
35 | }
36 | 
37 | function TooltipContent({
38 |   className,
39 |   sideOffset = 0,
40 |   children,
41 |   ...props
42 | }: React.ComponentProps<typeof TooltipPrimitive.Content>) {
43 |   return (
44 |     <TooltipPrimitive.Portal>
45 |       <TooltipPrimitive.Content
46 |         data-slot="tooltip-content"
47 |         sideOffset={sideOffset}
48 |         className={cn(
49 |           "bg-primary text-primary-foreground animate-in fade-in-0 zoom-in-95 data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=closed]:zoom-out-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 z-50 w-fit origin-(--radix-tooltip-content-transform-origin) rounded-md px-3 py-1.5 text-xs text-balance",
50 |           className
51 |         )}
52 |         {...props}
53 |       >
54 |         {children}
55 |         <TooltipPrimitive.Arrow className="bg-primary fill-primary z-50 size-2.5 translate-y-[calc(-50%_-_2px)] rotate-45 rounded-[2px]" />
56 |       </TooltipPrimitive.Content>
57 |     </TooltipPrimitive.Portal>
58 |   )
59 | }
60 | 
61 | export { Tooltip, TooltipTrigger, TooltipContent, TooltipProvider }
62 |

--------------------------------------------------------------------------------
/ui/src/chat/chat-input.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import {
 4 |   PromptInput,
 5 |   PromptInputAction,
 6 |   PromptInputActions,
 7 |   PromptInputTextarea,
 8 | } from "@/components/ui/prompt-input"
 9 | import { Button } from "@/components/ui/button"
10 | import { ArrowUp, Square } from "lucide-react"
11 | import { type ChatRequestOptions } from "@ai-sdk/ui-utils";
12 | import { cn } from "@/lib/utils";
13 | import { ModelSelector } from "./model-selector";
14 | 
15 | interface ChatInputProps {
16 |     input: string;
17 |     status: "submitted" | "streaming" | "ready" | "error";
18 |     setInput: (value: string) => void;
19 |     handleSubmit: (event?: {
20 |         preventDefault?: () => void;
21 |     }, chatRequestOptions?: ChatRequestOptions) => void;
22 |     selectedModel: string;
23 |     onModelChange: (modelId: string) => void;
24 | }
25 | 
26 | export function ChatInput({ input, setInput, handleSubmit, status, selectedModel, onModelChange }: ChatInputProps) {
27 | 
28 |   const isLoading = status === "streaming" || status === "submitted";
29 | 
30 |   return (
31 |     <PromptInput
32 |       value={input}
33 |       onValueChange={(value) => setInput(value)}
34 |       isLoading={isLoading}
35 |       onSubmit={handleSubmit}
36 |       className="w-full max-w-(--breakpoint-md)"
37 |     >
38 |       <PromptInputTextarea placeholder="Ask me anything..." />
39 |       <PromptInputActions className="justify-end pt-2 gap-2">
40 |         <ModelSelector selectedModel={selectedModel} onModelChange={onModelChange} />
41 |         <PromptInputAction
42 |           tooltip={isLoading ? "Stop generation" : "Send message"}
43 |         >
44 |           <Button
45 |             variant="default"
46 |             size="icon"
47 |             className={cn(input.length === 0 && "opacity-65" ,"h-8 w-8 rounded-full transition-opacity duration-300")}
48 |             onClick={handleSubmit}
49 |           >
50 |             {isLoading ? (
51 |               <Square className="size-3 fill-current" />
52 |             ) : (
53 |               <ArrowUp className="size-4" />
54 |             )}
55 |           </Button>
56 |         </PromptInputAction>
57 |       </PromptInputActions>
58 |     </PromptInput>
59 |   )
60 | }
61 |

--------------------------------------------------------------------------------
/Knowmore/services/claude_service.py:
--------------------------------------------------------------------------------
 1 | import json
 2 | import environ
 3 | from anthropic import AsyncAnthropic
 4 | 
 5 | env = environ.Env(
 6 |     ANTHROPIC_API_KEY=str,
 7 | )
 8 | 
 9 | class ClaudeService:
10 |     def __init__(self):
11 |         self.client = AsyncAnthropic(
12 |             api_key=env("ANTHROPIC_API_KEY"),
13 |         )
14 | 
15 |     async def stream_response(self, messages, model="claude-3-5-sonnet-20240620", enable_web_search=False):
16 |         stream_params = {
17 |             "max_tokens": 1024,
18 |             "messages": messages,
19 |             "model": model,
20 |         }
21 |         
22 |         try:
23 |             async with self.client.messages.stream(**stream_params) as stream:
24 |                 async for event in stream:
25 |                     # Handle different event types from Claude's streaming
26 |                     if hasattr(event, 'type'):
27 |                         if event.type == 'content_block_delta':
28 |                             if hasattr(event.delta, 'text'):
29 |                                 # Stream text using Vercel protocol
30 |                                 yield f'0:{json.dumps(event.delta.text)}\n'
31 |                         
32 |                         elif event.type == 'content_block_start':
33 |                             # Handle initial text content
34 |                             if hasattr(event, 'content_block') and hasattr(event.content_block, 'type'):
35 |                                 if event.content_block.type == 'text' and hasattr(event.content_block, 'text'):
36 |                                     if event.content_block.text:
37 |                                         yield f'0:{json.dumps(event.content_block.text)}\n'
38 |                         
39 |                         elif event.type == 'message_stop':
40 |                             # Send message finish with d: identifier
41 |                             yield f'd:{json.dumps({"finishReason": "stop"})}\n'
42 |                     
43 |                     else:
44 |                         # Fallback for text content
45 |                         if hasattr(event, 'text'):
46 |                             yield f'0:{json.dumps(event.text)}\n'
47 |         
48 |         except Exception as e:
49 |             # Send error using 3: identifier
50 |             yield f'3:{json.dumps({"error": str(e)})}\n'

--------------------------------------------------------------------------------
/ui/src/components/ui/button.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | import { Slot } from "@radix-ui/react-slot"
 3 | import { cva, type VariantProps } from "class-variance-authority"
 4 | 
 5 | import { cn } from "@/lib/utils"
 6 | 
 7 | const buttonVariants = cva(
 8 |   "cursor-pointer inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-all disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4 shrink-0 [&_svg]:shrink-0 outline-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
 9 |   {
10 |     variants: {
11 |       variant: {
12 |         default:
13 |           "bg-primary text-primary-foreground shadow-xs hover:bg-primary/90",
14 |         destructive:
15 |           "bg-destructive text-white shadow-xs hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60",
16 |         outline:
17 |           "border bg-background shadow-xs hover:bg-accent hover:text-accent-foreground dark:bg-input/30 dark:border-input dark:hover:bg-input/50",
18 |         secondary:
19 |           "bg-secondary text-secondary-foreground shadow-xs hover:bg-secondary/80",
20 |         ghost:
21 |           "hover:bg-accent hover:text-accent-foreground dark:hover:bg-accent/50",
22 |         link: "text-primary underline-offset-4 hover:underline",
23 |       },
24 |       size: {
25 |         default: "h-9 px-4 py-2 has-[>svg]:px-3",
26 |         sm: "h-8 rounded-md gap-1.5 px-3 has-[>svg]:px-2.5",
27 |         lg: "h-10 rounded-md px-6 has-[>svg]:px-4",
28 |         icon: "size-9",
29 |       },
30 |     },
31 |     defaultVariants: {
32 |       variant: "default",
33 |       size: "default",
34 |     },
35 |   }
36 | )
37 | 
38 | function Button({
39 |   className,
40 |   variant,
41 |   size,
42 |   asChild = false,
43 |   ...props
44 | }: React.ComponentProps<"button"> &
45 |   VariantProps<typeof buttonVariants> & {
46 |     asChild?: boolean
47 |   }) {
48 |   const Comp = asChild ? Slot : "button"
49 | 
50 |   return (
51 |     <Comp
52 |       data-slot="button"
53 |       className={cn(buttonVariants({ variant, size, className }))}
54 |       {...props}
55 |     />
56 |   )
57 | }
58 | 
59 | export { Button, buttonVariants }
60 |

--------------------------------------------------------------------------------
/ui/src/components/ui/card.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | 
 3 | import { cn } from "@/lib/utils"
 4 | 
 5 | function Card({ className, ...props }: React.ComponentProps<"div">) {
 6 |   return (
 7 |     <div
 8 |       data-slot="card"
 9 |       className={cn(
10 |         "bg-card text-card-foreground flex flex-col gap-6 rounded-xl border py-6 shadow-sm",
11 |         className
12 |       )}
13 |       {...props}
14 |     />
15 |   )
16 | }
17 | 
18 | function CardHeader({ className, ...props }: React.ComponentProps<"div">) {
19 |   return (
20 |     <div
21 |       data-slot="card-header"
22 |       className={cn(
23 |         "@container/card-header grid auto-rows-min grid-rows-[auto_auto] items-start gap-1.5 px-6 has-data-[slot=card-action]:grid-cols-[1fr_auto] [.border-b]:pb-6",
24 |         className
25 |       )}
26 |       {...props}
27 |     />
28 |   )
29 | }
30 | 
31 | function CardTitle({ className, ...props }: React.ComponentProps<"div">) {
32 |   return (
33 |     <div
34 |       data-slot="card-title"
35 |       className={cn("leading-none font-semibold", className)}
36 |       {...props}
37 |     />
38 |   )
39 | }
40 | 
41 | function CardDescription({ className, ...props }: React.ComponentProps<"div">) {
42 |   return (
43 |     <div
44 |       data-slot="card-description"
45 |       className={cn("text-muted-foreground text-sm", className)}
46 |       {...props}
47 |     />
48 |   )
49 | }
50 | 
51 | function CardAction({ className, ...props }: React.ComponentProps<"div">) {
52 |   return (
53 |     <div
54 |       data-slot="card-action"
55 |       className={cn(
56 |         "col-start-2 row-span-2 row-start-1 self-start justify-self-end",
57 |         className
58 |       )}
59 |       {...props}
60 |     />
61 |   )
62 | }
63 | 
64 | function CardContent({ className, ...props }: React.ComponentProps<"div">) {
65 |   return (
66 |     <div
67 |       data-slot="card-content"
68 |       className={cn("px-6", className)}
69 |       {...props}
70 |     />
71 |   )
72 | }
73 | 
74 | function CardFooter({ className, ...props }: React.ComponentProps<"div">) {
75 |   return (
76 |     <div
77 |       data-slot="card-footer"
78 |       className={cn("flex items-center px-6 [.border-t]:pt-6", className)}
79 |       {...props}
80 |     />
81 |   )
82 | }
83 | 
84 | export {
85 |   Card,
86 |   CardHeader,
87 |   CardFooter,
88 |   CardTitle,
89 |   CardAction,
90 |   CardDescription,
91 |   CardContent,
92 | }
93 |

--------------------------------------------------------------------------------
/ui/src/components/ui/code-block.tsx:
--------------------------------------------------------------------------------
 1 | import { cn } from "@/lib/utils"
 2 | import React, { useEffect, useState } from "react"
 3 | import { codeToHtml } from "shiki"
 4 | 
 5 | export type CodeBlockProps = {
 6 |   children?: React.ReactNode
 7 |   className?: string
 8 | } & React.HTMLProps<HTMLDivElement>
 9 | 
10 | function CodeBlock({ children, className, ...props }: CodeBlockProps) {
11 |   return (
12 |     <div
13 |       className={cn(
14 |         "not-prose flex w-full flex-col overflow-clip border",
15 |         "border-border bg-card text-card-foreground rounded-xl",
16 |         className
17 |       )}
18 |       {...props}
19 |     >
20 |       {children}
21 |     </div>
22 |   )
23 | }
24 | 
25 | export type CodeBlockCodeProps = {
26 |   code: string
27 |   language?: string
28 |   theme?: string
29 |   className?: string
30 | } & React.HTMLProps<HTMLDivElement>
31 | 
32 | function CodeBlockCode({
33 |   code,
34 |   language = "tsx",
35 |   theme = "github-light",
36 |   className,
37 |   ...props
38 | }: CodeBlockCodeProps) {
39 |   const [highlightedHtml, setHighlightedHtml] = useState<string | null>(null)
40 | 
41 |   useEffect(() => {
42 |     async function highlight() {
43 |       if (!code) {
44 |         setHighlightedHtml("<pre><code></code></pre>")
45 |         return
46 |       }
47 | 
48 |       const html = await codeToHtml(code, { lang: language, theme })
49 |       setHighlightedHtml(html)
50 |     }
51 |     highlight()
52 |   }, [code, language, theme])
53 | 
54 |   const classNames = cn(
55 |     "w-full overflow-x-auto text-[13px] [&>pre]:px-4 [&>pre]:py-4",
56 |     className
57 |   )
58 | 
59 |   // SSR fallback: render plain code if not hydrated yet
60 |   return highlightedHtml ? (
61 |     <div
62 |       className={classNames}
63 |       dangerouslySetInnerHTML={{ __html: highlightedHtml }}
64 |       {...props}
65 |     />
66 |   ) : (
67 |     <div className={classNames} {...props}>
68 |       <pre>
69 |         <code>{code}</code>
70 |       </pre>
71 |     </div>
72 |   )
73 | }
74 | 
75 | export type CodeBlockGroupProps = React.HTMLAttributes<HTMLDivElement>
76 | 
77 | function CodeBlockGroup({
78 |   children,
79 |   className,
80 |   ...props
81 | }: CodeBlockGroupProps) {
82 |   return (
83 |     <div
84 |       className={cn("flex items-center justify-between", className)}
85 |       {...props}
86 |     >
87 |       {children}
88 |     </div>
89 |   )
90 | }
91 | 
92 | export { CodeBlockGroup, CodeBlockCode, CodeBlock }
93 |

--------------------------------------------------------------------------------
/ui/README.md:
--------------------------------------------------------------------------------
 1 | # React + TypeScript + Vite
 2 | 
 3 | This template provides a minimal setup to get React working in Vite with HMR and some ESLint rules.
 4 | 
 5 | Currently, two official plugins are available:
 6 | 
 7 | - [@vitejs/plugin-react](https://github.com/vitejs/vite-plugin-react/blob/main/packages/plugin-react) uses [Babel](https://babeljs.io/) for Fast Refresh
 8 | - [@vitejs/plugin-react-swc](https://github.com/vitejs/vite-plugin-react/blob/main/packages/plugin-react-swc) uses [SWC](https://swc.rs/) for Fast Refresh
 9 | 
10 | ## Expanding the ESLint configuration
11 | 
12 | If you are developing a production application, we recommend updating the configuration to enable type-aware lint rules:
13 | 
14 | ```js
15 | export default tseslint.config([
16 |   globalIgnores(['dist']),
17 |   {
18 |     files: ['**/*.{ts,tsx}'],
19 |     extends: [
20 |       // Other configs...
21 | 
22 |       // Remove tseslint.configs.recommended and replace with this
23 |       ...tseslint.configs.recommendedTypeChecked,
24 |       // Alternatively, use this for stricter rules
25 |       ...tseslint.configs.strictTypeChecked,
26 |       // Optionally, add this for stylistic rules
27 |       ...tseslint.configs.stylisticTypeChecked,
28 | 
29 |       // Other configs...
30 |     ],
31 |     languageOptions: {
32 |       parserOptions: {
33 |         project: ['./tsconfig.node.json', './tsconfig.app.json'],
34 |         tsconfigRootDir: import.meta.dirname,
35 |       },
36 |       // other options...
37 |     },
38 |   },
39 | ])
40 | ```
41 | 
42 | You can also install [eslint-plugin-react-x](https://github.com/Rel1cx/eslint-react/tree/main/packages/plugins/eslint-plugin-react-x) and [eslint-plugin-react-dom](https://github.com/Rel1cx/eslint-react/tree/main/packages/plugins/eslint-plugin-react-dom) for React-specific lint rules:
43 | 
44 | ```js
45 | // eslint.config.js
46 | import reactX from 'eslint-plugin-react-x'
47 | import reactDom from 'eslint-plugin-react-dom'
48 | 
49 | export default tseslint.config([
50 |   globalIgnores(['dist']),
51 |   {
52 |     files: ['**/*.{ts,tsx}'],
53 |     extends: [
54 |       // Other configs...
55 |       // Enable lint rules for React
56 |       reactX.configs['recommended-typescript'],
57 |       // Enable lint rules for React DOM
58 |       reactDom.configs.recommended,
59 |     ],
60 |     languageOptions: {
61 |       parserOptions: {
62 |         project: ['./tsconfig.node.json', './tsconfig.app.json'],
63 |         tsconfigRootDir: import.meta.dirname,
64 |       },
65 |       // other options...
66 |     },
67 |   },
68 | ])
69 | ```
70 |

--------------------------------------------------------------------------------
/Knowmore/views.py:
--------------------------------------------------------------------------------
 1 | import os
 2 | import json
 3 | 
 4 | from django.http import StreamingHttpResponse, JsonResponse
 5 | from django.shortcuts import render
 6 | 
 7 | from .utils import get_vite_assets
 8 | from .handlers.stream_handler import event_stream, error_stream
 9 | from .handlers.message_processor import format_messages
10 | from .services.ai_provider import AIProviderFactory
11 | 
12 | def index(request):
13 |     assets = get_vite_assets()
14 |     return render(request, "react_app.html", {"assets": assets})
15 | 
16 | 
17 | async def sse_stream(request):
18 |     if request.method != 'POST':
19 |         return StreamingHttpResponse(error_stream('Method not allowed'), content_type='text/event-stream', status=405)
20 | 
21 |     try:
22 |         body = json.loads(request.body)
23 |         messages = body.get('messages', [])
24 |         model = body.get('model', 'claude-3-5-sonnet-20240620')
25 |         enable_web_search = body.get('enable_web_search', False)
26 | 
27 |         if not messages:
28 |             return StreamingHttpResponse(error_stream('No messages found'), content_type='text/event-stream', status=400)
29 | 
30 |         input_messages = format_messages(messages)
31 |         response = StreamingHttpResponse(
32 |             event_stream(input_messages, model, enable_web_search=enable_web_search), 
33 |             content_type='text/event-stream'
34 |         )
35 |         response['x-vercel-ai-data-stream'] = 'v1'
36 |         return response
37 |     except json.JSONDecodeError:
38 |         return StreamingHttpResponse(error_stream('Invalid JSON'), content_type='text/event-stream', status=400)
39 | 
40 | def get_models(request):
41 |     """Get available models from all providers"""
42 |     models_dict = AIProviderFactory.get_supported_models()
43 |     
44 |     # Transform into a flat list with provider info
45 |     models = []
46 |     for provider, model_list in models_dict.items():
47 |         for model in model_list:
48 |             models.append({
49 |                 'id': model['id'],
50 |                 'name': model['name'],
51 |                 'provider': provider
52 |             })
53 |     
54 |     return JsonResponse({'models': models})
55 | 
56 | 
57 | def get_manifest(request):
58 |     """Read Vite manifest to get the correct asset paths"""
59 |     try:
60 |         manifest_path = os.path.join(os.path.dirname(__file__), '..', 'static', 'dist', '.vite', 'manifest.json')
61 |         with open(manifest_path, 'r') as f:
62 |             manifest = json.load(f)
63 |         return JsonResponse(manifest)
64 |     except FileNotFoundError:
65 |         return JsonResponse({'error': 'Manifest not found. Please build the frontend first.'}, status=404)
66 |

--------------------------------------------------------------------------------
/ui/src/App.tsx:
--------------------------------------------------------------------------------
 1 | import { cn } from '@/lib/utils';
 2 | import { ChatInput } from '@/chat/chat-input'
 3 | import { useChat } from '@ai-sdk/react';
 4 | import { ChatMessage } from '@/chat/chat-message';
 5 | import { ChatHeader } from '@/chat/chat-header';
 6 | import {
 7 |   ChatContainerContent,
 8 |   ChatContainerRoot,
 9 |   ChatContainerScrollAnchor,
10 | } from "@/components/ui/chat-container"
11 | import { ScrollButton } from "@/components/ui/scroll-button"
12 | import { useState } from 'react';
13 | 
14 | export default function App() {
15 |   const [selectedModel, setSelectedModel] = useState(() => {
16 |     const stored = localStorage.getItem('selectedModel');
17 |     return stored || "claude-3-5-sonnet-latest";
18 |   });
19 |   
20 |   const handleModelChange = (modelId: string) => {
21 |     setSelectedModel(modelId);
22 |     localStorage.setItem('selectedModel', modelId);
23 |   };
24 |   
25 |   const { messages, input, setInput, status, handleSubmit } = useChat({
26 |     api: "/api/stream",
27 |     body: {
28 |       model: selectedModel,
29 |       enable_web_search: true,
30 |     },
31 |     // maxSteps removed - search is handled server-side in stream handler
32 |   });
33 | 
34 |   return (
35 |     <main className='flex h-screen w-full flex-col justify-center bg-accent'>
36 |       <div
37 |         className={cn(
38 |           messages.length > 0 && "h-[95vh]",
39 |           "max-w-3xl w-full mx-auto px-12 flex flex-col relative transition-all duration-500 ease-in-out"
40 |         )}
41 |       >
42 |         {messages.length > 0 && <ChatHeader />}
43 |         <ChatContainerRoot className={cn(
44 |           (messages.length > 0 || status === "submitted") ? "flex-1 w-full" : "hidden",
45 |           "scrollbar-hidden transition-opacity duration-500 ease-in-out"
46 |         )}>
47 |           <ChatContainerContent className={messages.length > 0 ? "mb-32" : ""}>
48 |             <ChatMessage messages={messages} isLoading={status === "submitted"} />
49 |             <ChatContainerScrollAnchor />
50 |           </ChatContainerContent>
51 |           <div className="absolute bottom-4 left-0 right-0 pb-20 pointer-events-none">
52 |             <div className='flex justify-center py-4'>
53 |               <ScrollButton className="shadow-none pointer-events-auto" />
54 |             </div>
55 |           </div>
56 |         </ChatContainerRoot>
57 |         <div
58 |           className={cn(
59 |             "px-12",
60 |             messages.length > 0
61 |               ? "absolute bottom-0 left-0 right-0 transform translate-y-0 transition-all duration-500 delay-300 ease-in-out"
62 |               : "transform -translate-y-20"
63 |           )}
64 |         >
65 |           {messages.length === 0 && <ChatHeader />}
66 |           <ChatInput 
67 |             input={input} 
68 |             setInput={setInput} 
69 |             status={status} 
70 |             handleSubmit={handleSubmit}
71 |             selectedModel={selectedModel}
72 |             onModelChange={handleModelChange}
73 |           />
74 |         </div>
75 |       </div>
76 |     </main >
77 |   )
78 | }
79 |

--------------------------------------------------------------------------------
/ui/src/chat/model-selector.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import { Brain, Check } from "lucide-react"
 4 | import { Button } from "@/components/ui/button"
 5 | import {
 6 |     DropdownMenu,
 7 |     DropdownMenuContent,
 8 |     DropdownMenuLabel,
 9 |     DropdownMenuItem,
10 |     DropdownMenuSeparator,
11 |     DropdownMenuTrigger,
12 | } from "@/components/ui/dropdown-menu"
13 | import useSWR from "swr"
14 | 
15 | export type ModelProvider = "claude" | "openai"
16 | 
17 | export interface Model {
18 |     id: string
19 |     name: string
20 |     provider: ModelProvider
21 | }
22 | 
23 | const fetcher = (url: string) => fetch(url).then(res => res.json())
24 | 
25 | interface ModelSelectorProps {
26 |     selectedModel: string
27 |     onModelChange: (modelId: string) => void
28 | }
29 | 
30 | export function ModelSelector({ selectedModel, onModelChange }: ModelSelectorProps) {
31 |     const { data, error, isLoading } = useSWR<{ models: Model[] }>('/api/models', fetcher, {
32 |         revalidateOnFocus: false,
33 |         revalidateOnReconnect: false,
34 |     })
35 | 
36 |     const models = data?.models || []
37 |     const currentModel = models.find(m => m.id === selectedModel) || models[0]
38 | 
39 |     return (
40 |         <DropdownMenu>
41 |             <DropdownMenuTrigger asChild>
42 |                 <Button variant="outline" size="sm" className="size-8 rounded-full" disabled={isLoading || !!error}>
43 |                     <Brain className="size-4 opacity-50" />
44 |                 </Button>
45 |             </DropdownMenuTrigger>
46 |             <DropdownMenuContent align="end" className="w-[200px]">
47 |                 <DropdownMenuLabel className="text-xs">
48 |                     {error ? "Error loading models" : "Select Model"}
49 |                 </DropdownMenuLabel>
50 |                 <DropdownMenuSeparator />
51 |                 {error ? (
52 |                     <DropdownMenuItem disabled className="text-sm text-muted-foreground">
53 |                         Failed to load models
54 |                     </DropdownMenuItem>
55 |                 ) : isLoading ? (
56 |                     <DropdownMenuItem disabled className="text-sm text-muted-foreground">
57 |                         Loading models...
58 |                     </DropdownMenuItem>
59 |                 ) : models.length === 0 ? (
60 |                     <DropdownMenuItem disabled className="text-sm text-muted-foreground">
61 |                         No models available
62 |                     </DropdownMenuItem>
63 |                 ) : (
64 |                     models.map(model => (
65 |                         <DropdownMenuItem
66 |                             key={model.id}
67 |                             onSelect={() => onModelChange(model.id)}
68 |                             className="text-sm justify-between cursor-pointer"
69 |                         >
70 |                             {model.name}
71 |                             {currentModel?.id === model.id && <Check className="size-4" />}
72 |                         </DropdownMenuItem>
73 |                     ))
74 |                 )}
75 |             </DropdownMenuContent>
76 |         </DropdownMenu>
77 |     )
78 | }

--------------------------------------------------------------------------------
/Knowmore/handlers/stream_handler.py:
--------------------------------------------------------------------------------
 1 | import json
 2 | import uuid
 3 | from ..services.ai_provider import AIProviderFactory
 4 | from ..services.search_orchestrator import SearchOrchestrator
 5 | 
 6 | 
 7 | async def event_stream(messages, model, enable_web_search=True):
 8 |     """Simplified stream handler with SearchOrchestrator"""
 9 |     try:
10 |         # Early return for non-search requests
11 |         if not enable_web_search:
12 |             provider = AIProviderFactory.get_provider(model)
13 |             async for chunk in provider.stream_response(messages, model, enable_web_search=False):
14 |                 yield chunk
15 |             return
16 |         
17 |         # Initialize search orchestrator
18 |         search_orchestrator = SearchOrchestrator()
19 |         
20 |         # Generate multiple search queries
21 |         search_queries = await search_orchestrator.generate_search_queries(messages)
22 |         
23 |         if search_queries:
24 |             # Generate tool call IDs for each query upfront
25 |             tool_call_ids = [f"search_{uuid.uuid4().hex[:8]}" for _ in search_queries]
26 |             
27 |             # Stream search indicators for each query
28 |             for i, (query, tool_call_id) in enumerate(zip(search_queries, tool_call_ids)):
29 |                 # Stream search start indicator
30 |                 yield f'b:{json.dumps({"toolCallId": tool_call_id, "toolName": "web_search"})}\n'
31 |                 
32 |                 # Stream search query and execution indicators
33 |                 yield f'c:{json.dumps({"toolCallId": tool_call_id, "argsTextDelta": query})}\n'
34 |                 yield f'9:{json.dumps({"toolCallId": tool_call_id, "toolName": "web_search", "args": {"query": query}})}\n'
35 |             
36 |             # Execute all searches concurrently
37 |             search_results_list = await search_orchestrator.execute_multiple_searches(search_queries)
38 |             
39 |             # Stream search results for each search with matching tool call IDs
40 |             for i, (search_results, tool_call_id) in enumerate(zip(search_results_list, tool_call_ids)):
41 |                 yield f'a:{json.dumps({"toolCallId": tool_call_id, "result": search_results})}\n'
42 |             
43 |             # Enhance messages with all search contexts
44 |             enhanced_messages = search_orchestrator.enhance_messages_with_multiple_searches(
45 |                 messages, search_results_list
46 |             )
47 |         else:
48 |             enhanced_messages = messages
49 |         
50 |         # Stream AI response
51 |         provider = AIProviderFactory.get_provider(model)
52 |         async for chunk in provider.stream_response(enhanced_messages, model, enable_web_search=False):
53 |             yield chunk
54 |             
55 |     except Exception as e:
56 |         # Log the error and stream it
57 |         print(f"Error in event_stream: {str(e)}")
58 |         import traceback
59 |         traceback.print_exc()
60 |         yield f'3:{json.dumps({"error": f"Stream error: {str(e)}"})}\n'
61 | 
62 | 
63 | 
64 | 
65 | async def error_stream(message):
66 |     """Generate error stream response using Vercel protocol"""
67 |     yield f'3:{json.dumps({"error": message})}\n'

--------------------------------------------------------------------------------
/ui/src/components/ui/message.tsx:
--------------------------------------------------------------------------------
  1 | import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar"
  2 | import {
  3 |   Tooltip,
  4 |   TooltipContent,
  5 |   TooltipProvider,
  6 |   TooltipTrigger,
  7 | } from "@/components/ui/tooltip"
  8 | import { cn } from "@/lib/utils"
  9 | import { Markdown } from "./markdown"
 10 | 
 11 | export type MessageProps = {
 12 |   children: React.ReactNode
 13 |   className?: string
 14 | } & React.HTMLProps<HTMLDivElement>
 15 | 
 16 | const Message = ({ children, className, ...props }: MessageProps) => (
 17 |   <div className={cn("flex gap-3", className)} {...props}>
 18 |     {children}
 19 |   </div>
 20 | )
 21 | 
 22 | export type MessageAvatarProps = {
 23 |   src: string
 24 |   alt: string
 25 |   fallback?: string
 26 |   delayMs?: number
 27 |   className?: string
 28 | }
 29 | 
 30 | const MessageAvatar = ({
 31 |   src,
 32 |   alt,
 33 |   fallback,
 34 |   delayMs,
 35 |   className,
 36 | }: MessageAvatarProps) => {
 37 |   return (
 38 |     <Avatar className={cn("h-8 w-8 shrink-0", className)}>
 39 |       <AvatarImage src={src} alt={alt} />
 40 |       {fallback && (
 41 |         <AvatarFallback delayMs={delayMs}>{fallback}</AvatarFallback>
 42 |       )}
 43 |     </Avatar>
 44 |   )
 45 | }
 46 | 
 47 | export type MessageContentProps = {
 48 |   children: React.ReactNode
 49 |   markdown?: boolean
 50 |   className?: string
 51 | } & React.ComponentProps<typeof Markdown> &
 52 |   React.HTMLProps<HTMLDivElement>
 53 | 
 54 | const MessageContent = ({
 55 |   children,
 56 |   markdown = false,
 57 |   className,
 58 |   ...props
 59 | }: MessageContentProps) => {
 60 |   const classNames = cn(
 61 |     "rounded-lg p-2 text-foreground bg-secondary prose break-words whitespace-normal",
 62 |     className
 63 |   )
 64 | 
 65 |   return markdown ? (
 66 |     <Markdown className={classNames} {...props}>
 67 |       {children as string}
 68 |     </Markdown>
 69 |   ) : (
 70 |     <div className={classNames} {...props}>
 71 |       {children}
 72 |     </div>
 73 |   )
 74 | }
 75 | 
 76 | export type MessageActionsProps = {
 77 |   children: React.ReactNode
 78 |   className?: string
 79 | } & React.HTMLProps<HTMLDivElement>
 80 | 
 81 | const MessageActions = ({
 82 |   children,
 83 |   className,
 84 |   ...props
 85 | }: MessageActionsProps) => (
 86 |   <div
 87 |     className={cn("text-muted-foreground flex items-center gap-2", className)}
 88 |     {...props}
 89 |   >
 90 |     {children}
 91 |   </div>
 92 | )
 93 | 
 94 | export type MessageActionProps = {
 95 |   className?: string
 96 |   tooltip: React.ReactNode
 97 |   children: React.ReactNode
 98 |   side?: "top" | "bottom" | "left" | "right"
 99 | } & React.ComponentProps<typeof Tooltip>
100 | 
101 | const MessageAction = ({
102 |   tooltip,
103 |   children,
104 |   className,
105 |   side = "top",
106 |   ...props
107 | }: MessageActionProps) => {
108 |   return (
109 |     <TooltipProvider>
110 |       <Tooltip {...props}>
111 |         <TooltipTrigger asChild>{children}</TooltipTrigger>
112 |         <TooltipContent side={side} className={className}>
113 |           {tooltip}
114 |         </TooltipContent>
115 |       </Tooltip>
116 |     </TooltipProvider>
117 |   )
118 | }
119 | 
120 | export { Message, MessageAvatar, MessageContent, MessageActions, MessageAction }
121 |

--------------------------------------------------------------------------------
/ui/src/components/ui/markdown.tsx:
--------------------------------------------------------------------------------
  1 | import { cn } from "@/lib/utils"
  2 | import { marked } from "marked"
  3 | import { memo, useId, useMemo } from "react"
  4 | import ReactMarkdown, { type Components } from "react-markdown"
  5 | import remarkBreaks from "remark-breaks"
  6 | import remarkGfm from "remark-gfm"
  7 | import { CodeBlock, CodeBlockCode } from "./code-block"
  8 | 
  9 | export type MarkdownProps = {
 10 |   children: string
 11 |   id?: string
 12 |   className?: string
 13 |   components?: Partial<Components>
 14 | }
 15 | 
 16 | function parseMarkdownIntoBlocks(markdown: string): string[] {
 17 |   const tokens = marked.lexer(markdown)
 18 |   return tokens.map((token) => token.raw)
 19 | }
 20 | 
 21 | function extractLanguage(className?: string): string {
 22 |   if (!className) return "plaintext"
 23 |   const match = className.match(/language-(\w+)/)
 24 |   return match ? match[1] : "plaintext"
 25 | }
 26 | 
 27 | const INITIAL_COMPONENTS: Partial<Components> = {
 28 |   code: function CodeComponent({ className, children, ...props }) {
 29 |     const isInline =
 30 |       !props.node?.position?.start.line ||
 31 |       props.node?.position?.start.line === props.node?.position?.end.line
 32 | 
 33 |     if (isInline) {
 34 |       return (
 35 |         <span
 36 |           className={cn(
 37 |             "bg-primary-foreground rounded-sm px-1 font-mono text-sm",
 38 |             className
 39 |           )}
 40 |           {...props}
 41 |         >
 42 |           {children}
 43 |         </span>
 44 |       )
 45 |     }
 46 | 
 47 |     const language = extractLanguage(className)
 48 | 
 49 |     return (
 50 |       <CodeBlock className={className}>
 51 |         <CodeBlockCode code={children as string} language={language} />
 52 |       </CodeBlock>
 53 |     )
 54 |   },
 55 |   pre: function PreComponent({ children }) {
 56 |     return <>{children}</>
 57 |   },
 58 | }
 59 | 
 60 | const MemoizedMarkdownBlock = memo(
 61 |   function MarkdownBlock({
 62 |     content,
 63 |     components = INITIAL_COMPONENTS,
 64 |   }: {
 65 |     content: string
 66 |     components?: Partial<Components>
 67 |   }) {
 68 |     return (
 69 |       <ReactMarkdown
 70 |         remarkPlugins={[remarkGfm, remarkBreaks]}
 71 |         components={components}
 72 |       >
 73 |         {content}
 74 |       </ReactMarkdown>
 75 |     )
 76 |   },
 77 |   function propsAreEqual(prevProps, nextProps) {
 78 |     return prevProps.content === nextProps.content
 79 |   }
 80 | )
 81 | 
 82 | MemoizedMarkdownBlock.displayName = "MemoizedMarkdownBlock"
 83 | 
 84 | function MarkdownComponent({
 85 |   children,
 86 |   id,
 87 |   className,
 88 |   components = INITIAL_COMPONENTS,
 89 | }: MarkdownProps) {
 90 |   const generatedId = useId()
 91 |   const blockId = id ?? generatedId
 92 |   const blocks = useMemo(() => parseMarkdownIntoBlocks(children), [children])
 93 | 
 94 |   return (
 95 |     <div className={className}>
 96 |       {blocks.map((block, index) => (
 97 |         <MemoizedMarkdownBlock
 98 |           key={`${blockId}-block-${index}`}
 99 |           content={block}
100 |           components={components}
101 |         />
102 |       ))}
103 |     </div>
104 |   )
105 | }
106 | 
107 | const Markdown = memo(MarkdownComponent)
108 | Markdown.displayName = "Markdown"
109 | 
110 | export { Markdown }
111 |

--------------------------------------------------------------------------------
/ui/src/components/chat-source-placeholder.tsx:
--------------------------------------------------------------------------------
 1 | import { Card, CardContent } from "@/components/ui/card";
 2 | import { ScrollArea } from "@/components/ui/scroll-area";
 3 | 
 4 | export function ChatSourcePlaceholder() {
 5 |     return (
 6 |         <div className="w-full bg-white backdrop-blur-sm rounded-lg border border-border px-4 pt-2 pb-1">
 7 |             {/* Header Section Shimmer */}
 8 |             <div className="flex items-center justify-between my-2">
 9 |                 <div className="h-6 bg-muted rounded animate-pulse w-32" />
10 |                 <div className="h-4 bg-muted rounded animate-pulse w-16" />
11 |             </div>
12 | 
13 |             {/* Filter Tags Shimmer */}
14 |             <div className="flex gap-2 mb-4 overflow-x-auto scrollbar-hidden">
15 |                 {[...Array(2)].map((_, index) => (
16 |                     <div
17 |                         key={index}
18 |                         className="h-6 bg-muted rounded-full animate-pulse w-16"
19 |                     />
20 |                 ))}
21 |             </div>
22 | 
23 |             {/* Scrollable Results Shimmer */}
24 |             <ScrollArea className="w-full whitespace-nowrap rounded-md">
25 |                 <div className="flex w-max space-x-4 mb-3">
26 |                     {[...Array(3)].map((_, index) => (
27 |                         <Card
28 |                             key={index}
29 |                             className="w-[340px] bg-accent border-accent shadow-none py-3"
30 |                         >
31 |                             <CardContent className="p-0">
32 |                                 <div className="flex px-2">
33 |                                     {/* Favicon shimmer */}
34 |                                     <div className="flex gap-2">
35 |                                         <div className="flex-shrink-0 w-8 h-8 bg-muted rounded-md animate-pulse" />
36 |                                         <div>
37 |                                             {/* Title shimmer */}
38 |                                             <div className="h-4 bg-muted rounded mb-1 animate-pulse w-40" />
39 |                                             {/* URL shimmer */}
40 |                                             <div className="h-3 bg-muted rounded animate-pulse w-24" />
41 |                                         </div>
42 |                                     </div>
43 |                                 </div>
44 | 
45 |                                 {/* Content shimmer */}
46 |                                 <div className="flex-1 min-w-0 space-y-2 py-4 px-3">
47 |                                     {/* Preview shimmer */}
48 |                                     <div className="space-y-1">
49 |                                         <div className="h-3 bg-muted rounded animate-pulse w-full" />
50 |                                         <div className="h-3 bg-muted rounded animate-pulse w-3/4" />
51 |                                         <div className="h-3 bg-muted rounded animate-pulse w-1/2" />
52 |                                     </div>
53 |                                 </div>
54 |                             </CardContent>
55 |                         </Card>
56 |                     ))}
57 |                 </div>
58 |             </ScrollArea>
59 |         </div>
60 |     );
61 | }
62 |

--------------------------------------------------------------------------------
/Knowmore/settings.py:
--------------------------------------------------------------------------------
  1 | 
  2 | from pathlib import Path
  3 | import environ
  4 | 
  5 | env = environ.Env(
  6 |     SECRET_KEY=str,
  7 | )
  8 | 
  9 | BASE_DIR = Path(__file__).resolve().parent.parent
 10 | SECRET_KEY = env("SECRET_KEY")
 11 | DEBUG = True
 12 | ALLOWED_HOSTS = []
 13 | 
 14 | 
 15 | # Application definition
 16 | 
 17 | INSTALLED_APPS = [
 18 |     'daphne',
 19 |     'django.contrib.admin',
 20 |     'django.contrib.auth',
 21 |     'django.contrib.contenttypes',
 22 |     'django.contrib.sessions',
 23 |     'django.contrib.messages',
 24 |     'django.contrib.staticfiles',
 25 |     'compressor',
 26 | ]
 27 | 
 28 | MIDDLEWARE = [
 29 |     'django.middleware.security.SecurityMiddleware',
 30 |     'django.contrib.sessions.middleware.SessionMiddleware',
 31 |     'django.middleware.common.CommonMiddleware',
 32 |     # 'django.middleware.csrf.CsrfViewMiddleware',
 33 |     'django.contrib.auth.middleware.AuthenticationMiddleware',
 34 |     'django.contrib.messages.middleware.MessageMiddleware',
 35 |     'django.middleware.clickjacking.XFrameOptionsMiddleware',
 36 | ]
 37 | 
 38 | ASGI_APPLICATION = "Knowmore.asgi.application"
 39 | 
 40 | ROOT_URLCONF = 'Knowmore.urls'
 41 | 
 42 | TEMPLATES = [
 43 |     {
 44 |         'BACKEND': 'django.template.backends.django.DjangoTemplates',
 45 |         'DIRS': [BASE_DIR / 'templates'],
 46 |         'APP_DIRS': True,
 47 |         'OPTIONS': {
 48 |             'context_processors': [
 49 |                 'django.template.context_processors.debug',
 50 |                 'django.template.context_processors.request',
 51 |                 'django.contrib.auth.context_processors.auth',
 52 |                 'django.contrib.messages.context_processors.messages',
 53 |             ],
 54 |         },
 55 |     },
 56 | ]
 57 | 
 58 | WSGI_APPLICATION = 'Knowmore.wsgi.application'
 59 | 
 60 | 
 61 | # Database
 62 | # https://docs.djangoproject.com/en/4.2/ref/settings/#databases
 63 | 
 64 | DATABASES = {
 65 |     'default': {
 66 |         'ENGINE': 'django.db.backends.sqlite3',
 67 |         'NAME': BASE_DIR / 'db.sqlite3',
 68 |     }
 69 | }
 70 | 
 71 | 
 72 | # Password validation
 73 | # https://docs.djangoproject.com/en/4.2/ref/settings/#auth-password-validators
 74 | 
 75 | AUTH_PASSWORD_VALIDATORS = [
 76 |     {
 77 |         'NAME': 'django.contrib.auth.password_validation.UserAttributeSimilarityValidator',
 78 |     },
 79 |     {
 80 |         'NAME': 'django.contrib.auth.password_validation.MinimumLengthValidator',
 81 |     },
 82 |     {
 83 |         'NAME': 'django.contrib.auth.password_validation.CommonPasswordValidator',
 84 |     },
 85 |     {
 86 |         'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator',
 87 |     },
 88 | ]
 89 | 
 90 | 
 91 | # Internationalization
 92 | # https://docs.djangoproject.com/en/4.2/topics/i18n/
 93 | 
 94 | LANGUAGE_CODE = 'en-us'
 95 | 
 96 | TIME_ZONE = 'UTC'
 97 | 
 98 | USE_I18N = True
 99 | 
100 | USE_TZ = True
101 | 
102 | STATIC_URL = '/static/'
103 | STATICFILES_DIRS = [
104 |     BASE_DIR / 'static',
105 | ]
106 | STATIC_ROOT = BASE_DIR / 'staticfiles'
107 | 
108 | DEFAULT_AUTO_FIELD = 'django.db.models.BigAutoField'
109 | 
110 | COMPRESS_ROOT = BASE_DIR / 'static'
111 | 
112 | COMPRESS_ENABLED = True
113 | 
114 | STATICFILES_FINDERS = (
115 |     'django.contrib.staticfiles.finders.FileSystemFinder',
116 |     'django.contrib.staticfiles.finders.AppDirectoriesFinder',
117 |     'compressor.finders.CompressorFinder',
118 | )
119 |

--------------------------------------------------------------------------------
/ui/src/chat/chat-source.tsx:
--------------------------------------------------------------------------------
 1 | import { Card, CardContent } from "@/components/ui/card";
 2 | import { Badge } from "@/components/ui/badge";
 3 | import { ScrollArea, ScrollBar } from "@/components/ui/scroll-area";
 4 | import { ChatSourcePlaceholder } from "@/components/chat-source-placeholder";
 5 | 
 6 | interface SearchResult {
 7 |     id: string;
 8 |     title: string;
 9 |     url: string;
10 |     preview: string;
11 |     favicon: string;
12 |     domain: string;
13 | }
14 | 
15 | type ChatSourceProps = {
16 |     filterTags: string[];
17 |     searchResults: SearchResult[]
18 | }
19 | 
20 | export function ChatSource({ filterTags, searchResults }: ChatSourceProps) {
21 |     if (searchResults.length === 0) {
22 |         return <ChatSourcePlaceholder />
23 |     }
24 | 
25 |     return (
26 |         <div className="w-full bg-white backdrop-blur-sm rounded-lg border border-border px-4 pt-2 pb-1">
27 |             {/* Header Section */}
28 |             <div className="flex items-center justify-between mb-2">
29 |                 <h2 className="text-lg font-semibold text-foreground">Sources</h2>
30 |                 <span className="text-sm text-muted-foreground">{searchResults.length} Results</span>
31 |             </div>
32 | 
33 |             {/* Filter Tags */}
34 |             <div className="flex gap-2 mb-4 overflow-x-auto scrollbar-hidden">
35 |                 {filterTags.map((tag, index) => (
36 |                     <Badge
37 |                         key={index}
38 |                         variant="secondary"
39 |                         className="text-secondary-foreground hover:bg-accent border-border"
40 |                     >
41 |                         {tag}
42 |                     </Badge>
43 |                 ))}
44 |             </div>
45 | 
46 |             {/* Scrollable Results */}
47 |             <ScrollArea className="w-full whitespace-nowrap rounded-md">
48 |                 <div className="flex w-max space-x-4 mb-3">
49 |                     {searchResults.length === 0 ? (
50 |                         <ChatSourcePlaceholder />
51 |                     ) : (
52 |                         searchResults.map((result) => (
53 |                             <Card
54 |                                 key={result.id}
55 |                                 className="w-[340px] bg-accent shadow-none py-3 cursor-pointer hover:bg-accent/80 transition-colors"
56 |                                 onClick={() => window.open(result.url, '_blank', 'noopener,noreferrer')}
57 |                             >
58 |                                 <CardContent className="p-0">
59 |                                     <div className="flex px-2">
60 |                                         {/* Favicon */}
61 |                                         <div className="flex gap-2">
62 |                                             <div className="flex-shrink-0 w-8 h-8 bg-white rounded-md flex items-center justify-center text-muted-foreground">
63 |                                                 {result.favicon}
64 |                                             </div>
65 |                                             <div>
66 |                                                 {/* Title */}
67 |                                                 <h3 className="text-sm font-medium text-foreground truncate whitespace-normal line-clamp-1">
68 |                                                     {result.title}
69 |                                                 </h3>
70 | 
71 |                                                 {/* URL */}
72 |                                                 <p className="text-xs text-primary truncate">
73 |                                                     {result.domain}
74 |                                                 </p>
75 |                                             </div>
76 |                                         </div>
77 |                                     </div>
78 | 
79 |                                     {/* Content */}
80 |                                     <div className="flex-1 min-w-0 space-y-2 py-4 px-3">
81 |                                         {/* Preview */}
82 |                                         <p className="text-xs text-muted-foreground whitespace-normal line-clamp-3">
83 |                                             {result.preview}
84 |                                         </p>
85 |                                     </div>
86 |                                 </CardContent>
87 |                             </Card>
88 |                         ))
89 |                     )}
90 |                 </div>
91 |                 <ScrollBar orientation="horizontal" />
92 |             </ScrollArea>
93 |         </div>
94 |     );
95 | }

--------------------------------------------------------------------------------
/Knowmore/services/web_search_firecrawl.py:
--------------------------------------------------------------------------------
  1 | import json
  2 | import requests
  3 | import environ
  4 | from typing import Dict, Any, List
  5 | 
  6 | env = environ.Env(
  7 |     FIRE_CRAWL_API_TOKEN=str,
  8 | )
  9 | 
 10 | class FirecrawlWebSearch:    
 11 |     def __init__(self):
 12 |         super().__init__()
 13 |         self.api_key = env("FIRE_CRAWL_API_TOKEN")
 14 |         self.base_url = "https://api.firecrawl.dev/v1"
 15 |     
 16 |     def get_name(self) -> str:
 17 |         return "web_search"
 18 |     
 19 |     def get_description(self) -> str:
 20 |         return "Search the web and optionally scrape content from search results. Returns titles, descriptions, URLs and optional content."
 21 |     
 22 |     def get_parameters(self) -> Dict[str, Any]:
 23 |         return {
 24 |             "type": "object",
 25 |             "properties": {
 26 |                 "query": {
 27 |                     "type": "string",
 28 |                     "description": "The search query"
 29 |                 },
 30 |                 "limit": {
 31 |                     "type": "integer",
 32 |                     "description": "Number of search results to return",
 33 |                     "default": 5,
 34 |                     "minimum": 1,
 35 |                     "maximum": 20
 36 |                 },
 37 |                 "location": {
 38 |                     "type": "string",
 39 |                     "description": "Location for localized search results (e.g., 'Germany', 'United States')"
 40 |                 },
 41 |                 "tbs": {
 42 |                     "type": "string",
 43 |                     "description": "Time-based search filter",
 44 |                     "enum": ["qdr:h", "qdr:d", "qdr:w", "qdr:m", "qdr:y"]
 45 |                 },
 46 |                 "scrape_content": {
 47 |                     "type": "boolean",
 48 |                     "description": "Whether to scrape content from search results",
 49 |                     "default": False
 50 |                 },
 51 |                 "formats": {
 52 |                     "type": "array",
 53 |                     "items": {
 54 |                         "type": "string",
 55 |                         "enum": ["markdown", "html", "rawHtml", "links", "screenshot"]
 56 |                     },
 57 |                     "description": "Content formats to scrape (only used if scrape_content is true)",
 58 |                     "default": ["markdown"]
 59 |                 }
 60 |             },
 61 |             "required": ["query"]
 62 |         }
 63 |     
 64 |     async def execute(self, **kwargs) -> Dict[str, Any]:
 65 |         """Execute Firecrawl web search"""
 66 |         if not self.api_key:
 67 |             return {
 68 |                 "error": "Firecrawl API key not configured",
 69 |                 "results": []
 70 |             }
 71 |         
 72 |         # Prepare request payload
 73 |         payload = {
 74 |             "query": kwargs["query"],
 75 |             "limit": kwargs.get("limit", 5)
 76 |         }
 77 |         
 78 |         # Add optional parameters
 79 |         if "location" in kwargs:
 80 |             payload["location"] = kwargs["location"]
 81 |         if "tbs" in kwargs:
 82 |             payload["tbs"] = kwargs["tbs"]
 83 |         
 84 |         # Add scraping options if requested
 85 |         if kwargs.get("scrape_content", False):
 86 |             payload["scrapeOptions"] = {
 87 |                 "formats": kwargs.get("formats", ["markdown"])
 88 |             }
 89 |         
 90 |         try:
 91 |             response = requests.post(
 92 |                 f"{self.base_url}/search",
 93 |                 headers={
 94 |                     "Content-Type": "application/json",
 95 |                     "Authorization": f"Bearer {self.api_key}"
 96 |                 },
 97 |                 json=payload,
 98 |                 timeout=30
 99 |             )
100 | 
101 |             if response.status_code == 200:
102 |                 data = response.json()
103 |                 return {
104 |                     "success": True,
105 |                     "results": data.get("data", []),
106 |                     "query": kwargs["query"]
107 |                 }
108 |             else:
109 |                 print("firecrawl:search failed with status", f"Bearer {self.api_key}")
110 |                 return {
111 |                     "error": f"Search failed with status {response.status_code}",
112 |                     "details": response.text,
113 |                     "results": []
114 |                 }
115 |                 
116 |         except requests.RequestException as e:
117 |             return {
118 |                 "error": f"Network error: {str(e)}",
119 |                 "results": []
120 |             }
121 |         except Exception as e:
122 |             return {
123 |                 "error": f"Unexpected error: {str(e)}",
124 |                 "results": []
125 |             }
126 |

--------------------------------------------------------------------------------
/ui/src/components/ui/prompt-input.tsx:
--------------------------------------------------------------------------------
  1 | import { Textarea } from "@/components/ui/textarea"
  2 | import {
  3 |   Tooltip,
  4 |   TooltipContent,
  5 |   TooltipProvider,
  6 |   TooltipTrigger,
  7 | } from "@/components/ui/tooltip"
  8 | import { cn } from "@/lib/utils"
  9 | import React, {
 10 |   createContext,
 11 |   useContext,
 12 |   useEffect,
 13 |   useRef,
 14 |   useState,
 15 | } from "react"
 16 | 
 17 | type PromptInputContextType = {
 18 |   isLoading: boolean
 19 |   value: string
 20 |   setValue: (value: string) => void
 21 |   maxHeight: number | string
 22 |   onSubmit?: () => void
 23 |   disabled?: boolean
 24 | }
 25 | 
 26 | const PromptInputContext = createContext<PromptInputContextType>({
 27 |   isLoading: false,
 28 |   value: "",
 29 |   setValue: () => {},
 30 |   maxHeight: 240,
 31 |   onSubmit: undefined,
 32 |   disabled: false,
 33 | })
 34 | 
 35 | function usePromptInput() {
 36 |   const context = useContext(PromptInputContext)
 37 |   if (!context) {
 38 |     throw new Error("usePromptInput must be used within a PromptInput")
 39 |   }
 40 |   return context
 41 | }
 42 | 
 43 | type PromptInputProps = {
 44 |   isLoading?: boolean
 45 |   value?: string
 46 |   onValueChange?: (value: string) => void
 47 |   maxHeight?: number | string
 48 |   onSubmit?: () => void
 49 |   children: React.ReactNode
 50 |   className?: string
 51 | }
 52 | 
 53 | function PromptInput({
 54 |   className,
 55 |   isLoading = false,
 56 |   maxHeight = 240,
 57 |   value,
 58 |   onValueChange,
 59 |   onSubmit,
 60 |   children,
 61 | }: PromptInputProps) {
 62 |   const [internalValue, setInternalValue] = useState(value || "")
 63 | 
 64 |   const handleChange = (newValue: string) => {
 65 |     setInternalValue(newValue)
 66 |     onValueChange?.(newValue)
 67 |   }
 68 | 
 69 |   return (
 70 |     <TooltipProvider>
 71 |       <PromptInputContext.Provider
 72 |         value={{
 73 |           isLoading,
 74 |           value: value ?? internalValue,
 75 |           setValue: onValueChange ?? handleChange,
 76 |           maxHeight,
 77 |           onSubmit,
 78 |         }}
 79 |       >
 80 |         <div
 81 |           className={cn(
 82 |             "border-input bg-background rounded-3xl border p-2 shadow-xs",
 83 |             className
 84 |           )}
 85 |         >
 86 |           {children}
 87 |         </div>
 88 |       </PromptInputContext.Provider>
 89 |     </TooltipProvider>
 90 |   )
 91 | }
 92 | 
 93 | export type PromptInputTextareaProps = {
 94 |   disableAutosize?: boolean
 95 | } & React.ComponentProps<typeof Textarea>
 96 | 
 97 | function PromptInputTextarea({
 98 |   className,
 99 |   onKeyDown,
100 |   disableAutosize = false,
101 |   ...props
102 | }: PromptInputTextareaProps) {
103 |   const { value, setValue, maxHeight, onSubmit, disabled } = usePromptInput()
104 |   const textareaRef = useRef<HTMLTextAreaElement>(null)
105 | 
106 |   useEffect(() => {
107 |     if (disableAutosize) return
108 | 
109 |     if (!textareaRef.current) return
110 |     textareaRef.current.style.height = "auto"
111 |     textareaRef.current.style.height =
112 |       typeof maxHeight === "number"
113 |         ? `${Math.min(textareaRef.current.scrollHeight, maxHeight)}px`
114 |         : `min(${textareaRef.current.scrollHeight}px, ${maxHeight})`
115 |   }, [value, maxHeight, disableAutosize])
116 | 
117 |   const handleKeyDown = (e: React.KeyboardEvent<HTMLTextAreaElement>) => {
118 |     if (e.key === "Enter" && !e.shiftKey) {
119 |       e.preventDefault()
120 |       onSubmit?.()
121 |     }
122 |     onKeyDown?.(e)
123 |   }
124 | 
125 |   return (
126 |     <Textarea
127 |       ref={textareaRef}
128 |       value={value}
129 |       onChange={(e) => setValue(e.target.value)}
130 |       onKeyDown={handleKeyDown}
131 |       className={cn(
132 |         "text-primary min-h-[44px] w-full resize-none border-none bg-transparent shadow-none outline-none focus-visible:ring-0 focus-visible:ring-offset-0",
133 |         className
134 |       )}
135 |       rows={1}
136 |       disabled={disabled}
137 |       {...props}
138 |     />
139 |   )
140 | }
141 | 
142 | type PromptInputActionsProps = React.HTMLAttributes<HTMLDivElement>
143 | 
144 | function PromptInputActions({
145 |   children,
146 |   className,
147 |   ...props
148 | }: PromptInputActionsProps) {
149 |   return (
150 |     <div className={cn("flex items-center gap-2", className)} {...props}>
151 |       {children}
152 |     </div>
153 |   )
154 | }
155 | 
156 | type PromptInputActionProps = {
157 |   className?: string
158 |   tooltip: React.ReactNode
159 |   children: React.ReactNode
160 |   side?: "top" | "bottom" | "left" | "right"
161 | } & React.ComponentProps<typeof Tooltip>
162 | 
163 | function PromptInputAction({
164 |   tooltip,
165 |   children,
166 |   className,
167 |   side = "top",
168 |   ...props
169 | }: PromptInputActionProps) {
170 |   const { disabled } = usePromptInput()
171 | 
172 |   return (
173 |     <Tooltip {...props}>
174 |       <TooltipTrigger asChild disabled={disabled}>
175 |         {children}
176 |       </TooltipTrigger>
177 |       <TooltipContent side={side} className={className}>
178 |         {tooltip}
179 |       </TooltipContent>
180 |     </Tooltip>
181 |   )
182 | }
183 | 
184 | export {
185 |   PromptInput,
186 |   PromptInputTextarea,
187 |   PromptInputActions,
188 |   PromptInputAction,
189 | }
190 |

--------------------------------------------------------------------------------
/ui/src/index.css:
--------------------------------------------------------------------------------
  1 | @import url('https://fonts.googleapis.com/css2?family=Instrument+Serif:ital,wght@0,400;0,500;0,600;0,700;1,400;1,500;1,600;1,700&display=swap');
  2 | @import "tailwindcss";
  3 | @import "tw-animate-css";
  4 | 
  5 | @plugin 'tailwindcss-animate';
  6 | @plugin '@tailwindcss/typography';
  7 | 
  8 | @custom-variant dark (&:is(.dark *));
  9 | 
 10 | @theme inline {
 11 |   --radius-sm: calc(var(--radius) - 4px);
 12 |   --radius-md: calc(var(--radius) - 2px);
 13 |   --radius-lg: var(--radius);
 14 |   --radius-xl: calc(var(--radius) + 4px);
 15 |   --color-background: var(--background);
 16 |   --color-foreground: var(--foreground);
 17 |   --color-card: var(--card);
 18 |   --color-card-foreground: var(--card-foreground);
 19 |   --color-popover: var(--popover);
 20 |   --color-popover-foreground: var(--popover-foreground);
 21 |   --color-primary: var(--primary);
 22 |   --color-primary-foreground: var(--primary-foreground);
 23 |   --color-secondary: var(--secondary);
 24 |   --color-secondary-foreground: var(--secondary-foreground);
 25 |   --color-muted: var(--muted);
 26 |   --color-muted-foreground: var(--muted-foreground);
 27 |   --color-accent: var(--accent);
 28 |   --color-accent-foreground: var(--accent-foreground);
 29 |   --color-destructive: var(--destructive);
 30 |   --color-border: var(--border);
 31 |   --color-input: var(--input);
 32 |   --color-ring: var(--ring);
 33 |   --color-chart-1: var(--chart-1);
 34 |   --color-chart-2: var(--chart-2);
 35 |   --color-chart-3: var(--chart-3);
 36 |   --color-chart-4: var(--chart-4);
 37 |   --color-chart-5: var(--chart-5);
 38 |   --color-sidebar: var(--sidebar);
 39 |   --color-sidebar-foreground: var(--sidebar-foreground);
 40 |   --color-sidebar-primary: var(--sidebar-primary);
 41 |   --color-sidebar-primary-foreground: var(--sidebar-primary-foreground);
 42 |   --color-sidebar-accent: var(--sidebar-accent);
 43 |   --color-sidebar-accent-foreground: var(--sidebar-accent-foreground);
 44 |   --color-sidebar-border: var(--sidebar-border);
 45 |   --color-sidebar-ring: var(--sidebar-ring);
 46 | }
 47 | 
 48 | :root {
 49 |   --radius: 0.625rem;
 50 |   --background: oklch(1 0 0);
 51 |   --foreground: oklch(0.145 0 0);
 52 |   --card: oklch(1 0 0);
 53 |   --card-foreground: oklch(0.145 0 0);
 54 |   --popover: oklch(1 0 0);
 55 |   --popover-foreground: oklch(0.145 0 0);
 56 |   --primary: oklch(0.205 0 0);
 57 |   --primary-foreground: oklch(0.985 0 0);
 58 |   --secondary: oklch(0.97 0 0);
 59 |   --secondary-foreground: oklch(0.205 0 0);
 60 |   --muted: oklch(0.97 0 0);
 61 |   --muted-foreground: oklch(0.556 0 0);
 62 |   --accent: oklch(0.97 0 0);
 63 |   --accent-foreground: oklch(0.205 0 0);
 64 |   --destructive: oklch(0.577 0.245 27.325);
 65 |   --border: oklch(0.922 0 0);
 66 |   --input: oklch(0.922 0 0);
 67 |   --ring: oklch(0.708 0 0);
 68 |   --chart-1: oklch(0.646 0.222 41.116);
 69 |   --chart-2: oklch(0.6 0.118 184.704);
 70 |   --chart-3: oklch(0.398 0.07 227.392);
 71 |   --chart-4: oklch(0.828 0.189 84.429);
 72 |   --chart-5: oklch(0.769 0.188 70.08);
 73 |   --sidebar: oklch(0.985 0 0);
 74 |   --sidebar-foreground: oklch(0.145 0 0);
 75 |   --sidebar-primary: oklch(0.205 0 0);
 76 |   --sidebar-primary-foreground: oklch(0.985 0 0);
 77 |   --sidebar-accent: oklch(0.97 0 0);
 78 |   --sidebar-accent-foreground: oklch(0.205 0 0);
 79 |   --sidebar-border: oklch(0.922 0 0);
 80 |   --sidebar-ring: oklch(0.708 0 0);
 81 | }
 82 | 
 83 | .dark {
 84 |   --background: oklch(0.145 0 0);
 85 |   --foreground: oklch(0.985 0 0);
 86 |   --card: oklch(0.205 0 0);
 87 |   --card-foreground: oklch(0.985 0 0);
 88 |   --popover: oklch(0.205 0 0);
 89 |   --popover-foreground: oklch(0.985 0 0);
 90 |   --primary: oklch(0.922 0 0);
 91 |   --primary-foreground: oklch(0.205 0 0);
 92 |   --secondary: oklch(0.269 0 0);
 93 |   --secondary-foreground: oklch(0.985 0 0);
 94 |   --muted: oklch(0.269 0 0);
 95 |   --muted-foreground: oklch(0.708 0 0);
 96 |   --accent: oklch(0.269 0 0);
 97 |   --accent-foreground: oklch(0.985 0 0);
 98 |   --destructive: oklch(0.704 0.191 22.216);
 99 |   --border: oklch(1 0 0 / 10%);
100 |   --input: oklch(1 0 0 / 15%);
101 |   --ring: oklch(0.556 0 0);
102 |   --chart-1: oklch(0.488 0.243 264.376);
103 |   --chart-2: oklch(0.696 0.17 162.48);
104 |   --chart-3: oklch(0.769 0.188 70.08);
105 |   --chart-4: oklch(0.627 0.265 303.9);
106 |   --chart-5: oklch(0.645 0.246 16.439);
107 |   --sidebar: oklch(0.205 0 0);
108 |   --sidebar-foreground: oklch(0.985 0 0);
109 |   --sidebar-primary: oklch(0.488 0.243 264.376);
110 |   --sidebar-primary-foreground: oklch(0.985 0 0);
111 |   --sidebar-accent: oklch(0.269 0 0);
112 |   --sidebar-accent-foreground: oklch(0.985 0 0);
113 |   --sidebar-border: oklch(1 0 0 / 10%);
114 |   --sidebar-ring: oklch(0.556 0 0);
115 | }
116 | 
117 | @layer base {
118 |   * {
119 |     @apply border-border outline-ring/50;
120 |   }
121 |   body {
122 |     @apply bg-background text-foreground;
123 |   }
124 | }
125 | 
126 | @layer utilities {
127 |   .font-elegant {
128 |     font-family: 'Instrument Serif', serif;
129 |   }
130 | }
131 | 
132 | .scrollbar-hidden {
133 |   scrollbar-width: none;
134 |   -ms-overflow-style: none;
135 | }
136 | 
137 | .scrollbar-hidden::-webkit-scrollbar {
138 |   display: none;
139 | }
140 | 
141 | /* Hide scrollbar for StickToBottom component's internal div */
142 | .scrollbar-hidden > div {
143 |   scrollbar-width: none;
144 |   -ms-overflow-style: none;
145 | }
146 | 
147 | .scrollbar-hidden > div::-webkit-scrollbar {
148 |   display: none;
149 | }
150 | 
151 | 
152 | 
153 |   /* keyframes for loaders */
154 |   @keyframes typing {
155 |     0%,
156 |     100% {
157 |       transform: translateY(0);
158 |       opacity: 0.5;
159 |     }
160 |     50% {
161 |       transform: translateY(-2px);
162 |       opacity: 1;
163 |     }
164 |   }
165 | 
166 |   @keyframes loading-dots {
167 |     0%,
168 |     100% {
169 |       opacity: 0;
170 |     }
171 |     50% {
172 |       opacity: 1;
173 |     }
174 |   }
175 | 
176 |   @keyframes wave {
177 |     0%,
178 |     100% {
179 |       transform: scaleY(1);
180 |     }
181 |     50% {
182 |       transform: scaleY(0.6);
183 |     }
184 |   }
185 | 
186 |   @keyframes blink {
187 |     0%,
188 |     100% {
189 |       opacity: 1;
190 |     }
191 |     50% {
192 |       opacity: 0;
193 |     }
194 |   }
195 | 
196 |   @keyframes text-blink {
197 |     0%,
198 |     100% {
199 |       color: var(--primary);
200 |     }
201 |     50% {
202 |       color: var(--muted-foreground);
203 |     }
204 |   }
205 | 
206 |   @keyframes bounce-dots {
207 |     0%,
208 |     100% {
209 |       transform: scale(0.8);
210 |       opacity: 0.5;
211 |     }
212 |     50% {
213 |       transform: scale(1.2);
214 |       opacity: 1;
215 |     }
216 |   }
217 | 
218 |   @keyframes thin-pulse {
219 |     0%,
220 |     100% {
221 |       transform: scale(0.95);
222 |       opacity: 0.8;
223 |     }
224 |     50% {
225 |       transform: scale(1.05);
226 |       opacity: 0.4;
227 |     }
228 |   }
229 | 
230 |   @keyframes pulse-dot {
231 |     0%,
232 |     100% {
233 |       transform: scale(1);
234 |       opacity: 0.8;
235 |     }
236 |     50% {
237 |       transform: scale(1.5);
238 |       opacity: 1;
239 |     }
240 |   }
241 | 
242 |   @keyframes shimmer-text {
243 |     0% {
244 |       background-position: 150% center;
245 |     }
246 |     100% {
247 |       background-position: -150% center;
248 |     }
249 |   }
250 | 
251 |   @keyframes wave-bars {
252 |     0%,
253 |     100% {
254 |       transform: scaleY(1);
255 |       opacity: 0.5;
256 |     }
257 |     50% {
258 |       transform: scaleY(0.6);
259 |       opacity: 1;
260 |     }
261 |   }
262 | 
263 |   @keyframes shimmer {
264 |     0% {
265 |       background-position: 200% 50%;
266 |     }
267 |     100% {
268 |       background-position: -200% 50%;
269 |     }
270 |   }
271 | 
272 |   @keyframes spinner-fade {
273 |     0% {
274 |       opacity: 0;
275 |     }
276 |     100% {
277 |       opacity: 1;
278 |     }
279 |   }
280 |

--------------------------------------------------------------------------------
/ui/src/chat/chat-message.tsx:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | import {
  4 |     Message,
  5 |     MessageAction,
  6 |     MessageActions,
  7 |     MessageContent,
  8 | } from "@/components/ui/message"
  9 | import { Button } from "@/components/ui/button"
 10 | import { Copy, ThumbsDown, ThumbsUp } from "lucide-react"
 11 | import { useState, useMemo } from "react"
 12 | import { type Message as MessageItem } from "@ai-sdk/react"
 13 | import { ChatSource } from "./chat-source"
 14 | import { ChatSourcePlaceholder } from "@/components/chat-source-placeholder"
 15 | import { Loader } from '@/components/ui/loader';
 16 | 
 17 | function extractSearchQuery(args: any): string {
 18 |     if (typeof args === 'string') {
 19 |         try {
 20 |             const parsed = JSON.parse(args);
 21 |             return parsed.query || '';
 22 |         } catch {
 23 |             return args;
 24 |         }
 25 |     }
 26 |     return args?.query || '';
 27 | }
 28 | 
 29 | function transformSearchResults(results: any, idPrefix: string = ''): any[] {
 30 |     if (!Array.isArray(results)) {
 31 |         console.warn('transformSearchResults: Expected array but received:', typeof results, results);
 32 |         return [];
 33 |     }
 34 |     
 35 |     return results.map((result, index) => ({
 36 |         id: `${idPrefix}result-${index}`,
 37 |         title: result.title || 'Untitled',
 38 |         url: result.url || '',
 39 |         preview: result.description || result.snippet || 'No preview available',
 40 |         favicon: result.url ? new URL(result.url).hostname.charAt(0).toUpperCase() : '?',
 41 |         domain: result.url ? new URL(result.url).hostname : 'unknown',
 42 |     }));
 43 | }
 44 | 
 45 | function AssistantMessage({ message }: { message: MessageItem }) {
 46 |     const [liked, setLiked] = useState<boolean | null>(null)
 47 |     const [copied, setCopied] = useState(false)
 48 | 
 49 |     const handleCopy = (text: string) => {
 50 |         navigator.clipboard.writeText(text)
 51 |         setCopied(true)
 52 |         setTimeout(() => setCopied(false), 2000)
 53 |     }
 54 | 
 55 |     const renderedContent = useMemo(() => {
 56 |         if (!message.parts || message.parts.length === 0) {
 57 |             return (
 58 |                 <MessageContent markdown className="bg-transparent p-0">
 59 |                     {message.content}
 60 |                 </MessageContent>
 61 |             );
 62 |         }
 63 | 
 64 |         // Collect all search results first
 65 |         const allSearchResults: any[] = [];
 66 |         const allFilterTags: string[] = [];
 67 |         let hasSearches = false;
 68 |         let searchesLoading = false;
 69 | 
 70 |         // First pass: collect all search data
 71 |         message.parts.forEach((part, partIndex) => {
 72 |             if (part.type === 'tool-invocation') {
 73 |                 const invocation = part.toolInvocation as any;
 74 |                 const { toolName, state, args } = invocation;
 75 | 
 76 |                 if (toolName === 'web_search') {
 77 |                     hasSearches = true;
 78 |                     
 79 |                     if (!invocation.result) {
 80 |                         searchesLoading = true;
 81 |                     } else if (state === 'result' && invocation.result) {
 82 |                         const resultsArray = invocation.result?.results || invocation.result || [];
 83 |                         const searchResults = transformSearchResults(resultsArray, `search-${partIndex}-`);
 84 |                         allSearchResults.push(...searchResults);
 85 |                         
 86 |                         const query = extractSearchQuery(args);
 87 |                         if (query && !allFilterTags.includes(query)) {
 88 |                             allFilterTags.push(query);
 89 |                         }
 90 |                     }
 91 |                 }
 92 |             }
 93 |         });
 94 | 
 95 |         // Second pass: render content
 96 |         const content: React.ReactNode[] = [];
 97 |         
 98 |         // If we have searches, show the combined source component first
 99 |         if (hasSearches) {
100 |             if (searchesLoading) {
101 |                 content.push(
102 |                     <div key="search-loading" className="mb-4">
103 |                         <ChatSourcePlaceholder />
104 |                     </div>
105 |                 );
106 |             } else if (allSearchResults.length > 0) {
107 |                 content.push(
108 |                     <div key="search-results" className="mb-4">
109 |                         <ChatSource 
110 |                             filterTags={allFilterTags} 
111 |                             searchResults={allSearchResults} 
112 |                         />
113 |                     </div>
114 |                 );
115 |             }
116 |         }
117 | 
118 |         // Render text parts
119 |         message.parts.forEach((part, partIndex) => {
120 |             if (part.type === 'text') {
121 |                 content.push(
122 |                     <MessageContent key={`text-${partIndex}`} markdown className="bg-transparent p-0">
123 |                         {part.text}
124 |                     </MessageContent>
125 |                 );
126 |             }
127 |         });
128 | 
129 |         return content;
130 |     }, [message.parts, message.content]);
131 | 
132 |     return (
133 |         <Message className="justify-start px-2">
134 |             <div className="flex w-full flex-col gap-4">
135 |                 {renderedContent}
136 | 
137 |                 <MessageActions className="self-start">
138 |                     <MessageAction tooltip="Copy to clipboard">
139 |                         <Button
140 |                             variant="ghost"
141 |                             size="icon"
142 |                             className="h-8 w-8 rounded-full"
143 |                             onClick={() => handleCopy(message.content)}
144 |                         >
145 |                             <Copy className={`size-4 ${copied ? "text-green-500" : ""}`} />
146 |                         </Button>
147 |                     </MessageAction>
148 | 
149 |                     <MessageAction tooltip="Helpful">
150 |                         <Button
151 |                             variant="ghost"
152 |                             size="icon"
153 |                             className={`h-8 w-8 rounded-full ${liked === true ? "bg-green-100 text-green-500" : ""}`}
154 |                             onClick={() => setLiked(true)}
155 |                         >
156 |                             <ThumbsUp className="size-4" />
157 |                         </Button>
158 |                     </MessageAction>
159 | 
160 |                     <MessageAction tooltip="Not helpful">
161 |                         <Button
162 |                             variant="ghost"
163 |                             size="icon"
164 |                             className={`h-8 w-8 rounded-full ${liked === false ? "bg-red-100 text-red-500" : ""}`}
165 |                             onClick={() => setLiked(false)}
166 |                         >
167 |                             <ThumbsDown className="size-4" />
168 |                         </Button>
169 |                     </MessageAction>
170 |                 </MessageActions>
171 |             </div>
172 |         </Message>
173 |     );
174 | }
175 | 
176 | export function ChatMessage({ messages, isLoading }: { messages: MessageItem[], isLoading: boolean }) {
177 |     return (
178 |         <div className="flex flex-col gap-8">
179 |             {messages.map((message) => (
180 |                 message.role === "user" ? (
181 |                     <Message key={message.id} className="justify-start">
182 |                         <MessageContent>{message.content}</MessageContent>
183 |                     </Message>
184 |                 ) : (
185 |                     <AssistantMessage key={message.id} message={message} />
186 |                 )
187 |             ))}
188 |             {isLoading && (
189 |                 <div>
190 |                     <div className="mb-4">
191 |                         <ChatSourcePlaceholder />
192 |                     </div>
193 |                     <div className="px-4">
194 |                         <Loader variant='typing' />
195 |                     </div>
196 |                 </div>
197 |             )}
198 |         </div>
199 |     )
200 | }
201 |

--------------------------------------------------------------------------------
/ui/src/components/ui/dropdown-menu.tsx:
--------------------------------------------------------------------------------
  1 | import * as React from "react"
  2 | import * as DropdownMenuPrimitive from "@radix-ui/react-dropdown-menu"
  3 | import { CheckIcon, ChevronRightIcon, CircleIcon } from "lucide-react"
  4 | 
  5 | import { cn } from "@/lib/utils"
  6 | 
  7 | function DropdownMenu({
  8 |   ...props
  9 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Root>) {
 10 |   return <DropdownMenuPrimitive.Root data-slot="dropdown-menu" {...props} />
 11 | }
 12 | 
 13 | function DropdownMenuPortal({
 14 |   ...props
 15 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Portal>) {
 16 |   return (
 17 |     <DropdownMenuPrimitive.Portal data-slot="dropdown-menu-portal" {...props} />
 18 |   )
 19 | }
 20 | 
 21 | function DropdownMenuTrigger({
 22 |   ...props
 23 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Trigger>) {
 24 |   return (
 25 |     <DropdownMenuPrimitive.Trigger
 26 |       data-slot="dropdown-menu-trigger"
 27 |       {...props}
 28 |     />
 29 |   )
 30 | }
 31 | 
 32 | function DropdownMenuContent({
 33 |   className,
 34 |   sideOffset = 4,
 35 |   ...props
 36 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Content>) {
 37 |   return (
 38 |     <DropdownMenuPrimitive.Portal>
 39 |       <DropdownMenuPrimitive.Content
 40 |         data-slot="dropdown-menu-content"
 41 |         sideOffset={sideOffset}
 42 |         className={cn(
 43 |           "bg-popover text-popover-foreground data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 z-50 max-h-(--radix-dropdown-menu-content-available-height) min-w-[8rem] origin-(--radix-dropdown-menu-content-transform-origin) overflow-x-hidden overflow-y-auto rounded-md border p-1 shadow-md",
 44 |           className
 45 |         )}
 46 |         {...props}
 47 |       />
 48 |     </DropdownMenuPrimitive.Portal>
 49 |   )
 50 | }
 51 | 
 52 | function DropdownMenuGroup({
 53 |   ...props
 54 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Group>) {
 55 |   return (
 56 |     <DropdownMenuPrimitive.Group data-slot="dropdown-menu-group" {...props} />
 57 |   )
 58 | }
 59 | 
 60 | function DropdownMenuItem({
 61 |   className,
 62 |   inset,
 63 |   variant = "default",
 64 |   ...props
 65 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Item> & {
 66 |   inset?: boolean
 67 |   variant?: "default" | "destructive"
 68 | }) {
 69 |   return (
 70 |     <DropdownMenuPrimitive.Item
 71 |       data-slot="dropdown-menu-item"
 72 |       data-inset={inset}
 73 |       data-variant={variant}
 74 |       className={cn(
 75 |         "focus:bg-accent focus:text-accent-foreground data-[variant=destructive]:text-destructive data-[variant=destructive]:focus:bg-destructive/10 dark:data-[variant=destructive]:focus:bg-destructive/20 data-[variant=destructive]:focus:text-destructive data-[variant=destructive]:*:[svg]:!text-destructive [&_svg:not([class*='text-'])]:text-muted-foreground relative flex cursor-default items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-hidden select-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 data-[inset]:pl-8 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
 76 |         className
 77 |       )}
 78 |       {...props}
 79 |     />
 80 |   )
 81 | }
 82 | 
 83 | function DropdownMenuCheckboxItem({
 84 |   className,
 85 |   children,
 86 |   checked,
 87 |   ...props
 88 | }: React.ComponentProps<typeof DropdownMenuPrimitive.CheckboxItem>) {
 89 |   return (
 90 |     <DropdownMenuPrimitive.CheckboxItem
 91 |       data-slot="dropdown-menu-checkbox-item"
 92 |       className={cn(
 93 |         "focus:bg-accent focus:text-accent-foreground relative flex cursor-default items-center gap-2 rounded-sm py-1.5 pr-2 pl-8 text-sm outline-hidden select-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
 94 |         className
 95 |       )}
 96 |       checked={checked}
 97 |       {...props}
 98 |     >
 99 |       <span className="pointer-events-none absolute left-2 flex size-3.5 items-center justify-center">
100 |         <DropdownMenuPrimitive.ItemIndicator>
101 |           <CheckIcon className="size-4" />
102 |         </DropdownMenuPrimitive.ItemIndicator>
103 |       </span>
104 |       {children}
105 |     </DropdownMenuPrimitive.CheckboxItem>
106 |   )
107 | }
108 | 
109 | function DropdownMenuRadioGroup({
110 |   ...props
111 | }: React.ComponentProps<typeof DropdownMenuPrimitive.RadioGroup>) {
112 |   return (
113 |     <DropdownMenuPrimitive.RadioGroup
114 |       data-slot="dropdown-menu-radio-group"
115 |       {...props}
116 |     />
117 |   )
118 | }
119 | 
120 | function DropdownMenuRadioItem({
121 |   className,
122 |   children,
123 |   ...props
124 | }: React.ComponentProps<typeof DropdownMenuPrimitive.RadioItem>) {
125 |   return (
126 |     <DropdownMenuPrimitive.RadioItem
127 |       data-slot="dropdown-menu-radio-item"
128 |       className={cn(
129 |         "focus:bg-accent focus:text-accent-foreground relative flex cursor-default items-center gap-2 rounded-sm py-1.5 pr-2 pl-8 text-sm outline-hidden select-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
130 |         className
131 |       )}
132 |       {...props}
133 |     >
134 |       <span className="pointer-events-none absolute left-2 flex size-3.5 items-center justify-center">
135 |         <DropdownMenuPrimitive.ItemIndicator>
136 |           <CircleIcon className="size-2 fill-current" />
137 |         </DropdownMenuPrimitive.ItemIndicator>
138 |       </span>
139 |       {children}
140 |     </DropdownMenuPrimitive.RadioItem>
141 |   )
142 | }
143 | 
144 | function DropdownMenuLabel({
145 |   className,
146 |   inset,
147 |   ...props
148 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Label> & {
149 |   inset?: boolean
150 | }) {
151 |   return (
152 |     <DropdownMenuPrimitive.Label
153 |       data-slot="dropdown-menu-label"
154 |       data-inset={inset}
155 |       className={cn(
156 |         "px-2 py-1.5 text-sm font-medium data-[inset]:pl-8",
157 |         className
158 |       )}
159 |       {...props}
160 |     />
161 |   )
162 | }
163 | 
164 | function DropdownMenuSeparator({
165 |   className,
166 |   ...props
167 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Separator>) {
168 |   return (
169 |     <DropdownMenuPrimitive.Separator
170 |       data-slot="dropdown-menu-separator"
171 |       className={cn("bg-border -mx-1 my-1 h-px", className)}
172 |       {...props}
173 |     />
174 |   )
175 | }
176 | 
177 | function DropdownMenuShortcut({
178 |   className,
179 |   ...props
180 | }: React.ComponentProps<"span">) {
181 |   return (
182 |     <span
183 |       data-slot="dropdown-menu-shortcut"
184 |       className={cn(
185 |         "text-muted-foreground ml-auto text-xs tracking-widest",
186 |         className
187 |       )}
188 |       {...props}
189 |     />
190 |   )
191 | }
192 | 
193 | function DropdownMenuSub({
194 |   ...props
195 | }: React.ComponentProps<typeof DropdownMenuPrimitive.Sub>) {
196 |   return <DropdownMenuPrimitive.Sub data-slot="dropdown-menu-sub" {...props} />
197 | }
198 | 
199 | function DropdownMenuSubTrigger({
200 |   className,
201 |   inset,
202 |   children,
203 |   ...props
204 | }: React.ComponentProps<typeof DropdownMenuPrimitive.SubTrigger> & {
205 |   inset?: boolean
206 | }) {
207 |   return (
208 |     <DropdownMenuPrimitive.SubTrigger
209 |       data-slot="dropdown-menu-sub-trigger"
210 |       data-inset={inset}
211 |       className={cn(
212 |         "focus:bg-accent focus:text-accent-foreground data-[state=open]:bg-accent data-[state=open]:text-accent-foreground flex cursor-default items-center rounded-sm px-2 py-1.5 text-sm outline-hidden select-none data-[inset]:pl-8",
213 |         className
214 |       )}
215 |       {...props}
216 |     >
217 |       {children}
218 |       <ChevronRightIcon className="ml-auto size-4" />
219 |     </DropdownMenuPrimitive.SubTrigger>
220 |   )
221 | }
222 | 
223 | function DropdownMenuSubContent({
224 |   className,
225 |   ...props
226 | }: React.ComponentProps<typeof DropdownMenuPrimitive.SubContent>) {
227 |   return (
228 |     <DropdownMenuPrimitive.SubContent
229 |       data-slot="dropdown-menu-sub-content"
230 |       className={cn(
231 |         "bg-popover text-popover-foreground data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 z-50 min-w-[8rem] origin-(--radix-dropdown-menu-content-transform-origin) overflow-hidden rounded-md border p-1 shadow-lg",
232 |         className
233 |       )}
234 |       {...props}
235 |     />
236 |   )
237 | }
238 | 
239 | export {
240 |   DropdownMenu,
241 |   DropdownMenuPortal,
242 |   DropdownMenuTrigger,
243 |   DropdownMenuContent,
244 |   DropdownMenuGroup,
245 |   DropdownMenuLabel,
246 |   DropdownMenuItem,
247 |   DropdownMenuCheckboxItem,
248 |   DropdownMenuRadioGroup,
249 |   DropdownMenuRadioItem,
250 |   DropdownMenuSeparator,
251 |   DropdownMenuShortcut,
252 |   DropdownMenuSub,
253 |   DropdownMenuSubTrigger,
254 |   DropdownMenuSubContent,
255 | }
256 |

--------------------------------------------------------------------------------
/ui/src/components/ui/loader.tsx:
--------------------------------------------------------------------------------
  1 | import { cn } from "@/lib/utils"
  2 | 
  3 | export interface LoaderProps {
  4 |   variant?:
  5 |     | "circular"
  6 |     | "classic"
  7 |     | "pulse"
  8 |     | "pulse-dot"
  9 |     | "dots"
 10 |     | "typing"
 11 |     | "wave"
 12 |     | "bars"
 13 |     | "terminal"
 14 |     | "text-blink"
 15 |     | "text-shimmer"
 16 |     | "loading-dots"
 17 |   size?: "sm" | "md" | "lg"
 18 |   text?: string
 19 |   className?: string
 20 | }
 21 | 
 22 | export function CircularLoader({
 23 |   className,
 24 |   size = "md",
 25 | }: {
 26 |   className?: string
 27 |   size?: "sm" | "md" | "lg"
 28 | }) {
 29 |   const sizeClasses = {
 30 |     sm: "size-4",
 31 |     md: "size-5",
 32 |     lg: "size-6",
 33 |   }
 34 | 
 35 |   return (
 36 |     <div
 37 |       className={cn(
 38 |         "border-primary animate-spin rounded-full border-2 border-t-transparent",
 39 |         sizeClasses[size],
 40 |         className
 41 |       )}
 42 |     >
 43 |       <span className="sr-only">Loading</span>
 44 |     </div>
 45 |   )
 46 | }
 47 | 
 48 | export function ClassicLoader({
 49 |   className,
 50 |   size = "md",
 51 | }: {
 52 |   className?: string
 53 |   size?: "sm" | "md" | "lg"
 54 | }) {
 55 |   const sizeClasses = {
 56 |     sm: "size-4",
 57 |     md: "size-5",
 58 |     lg: "size-6",
 59 |   }
 60 | 
 61 |   const barSizes = {
 62 |     sm: { height: "6px", width: "1.5px" },
 63 |     md: { height: "8px", width: "2px" },
 64 |     lg: { height: "10px", width: "2.5px" },
 65 |   }
 66 | 
 67 |   return (
 68 |     <div className={cn("relative", sizeClasses[size], className)}>
 69 |       <div className="absolute h-full w-full">
 70 |         {[...Array(12)].map((_, i) => (
 71 |           <div
 72 |             key={i}
 73 |             className="bg-primary absolute animate-[spinner-fade_1.2s_linear_infinite] rounded-full"
 74 |             style={{
 75 |               top: "0",
 76 |               left: "50%",
 77 |               marginLeft:
 78 |                 size === "sm" ? "-0.75px" : size === "lg" ? "-1.25px" : "-1px",
 79 |               transformOrigin: `${size === "sm" ? "0.75px" : size === "lg" ? "1.25px" : "1px"} ${size === "sm" ? "10px" : size === "lg" ? "14px" : "12px"}`,
 80 |               transform: `rotate(${i * 30}deg)`,
 81 |               opacity: 0,
 82 |               animationDelay: `${i * 0.1}s`,
 83 |               height: barSizes[size].height,
 84 |               width: barSizes[size].width,
 85 |             }}
 86 |           />
 87 |         ))}
 88 |       </div>
 89 |       <span className="sr-only">Loading</span>
 90 |     </div>
 91 |   )
 92 | }
 93 | 
 94 | export function PulseLoader({
 95 |   className,
 96 |   size = "md",
 97 | }: {
 98 |   className?: string
 99 |   size?: "sm" | "md" | "lg"
100 | }) {
101 |   const sizeClasses = {
102 |     sm: "size-4",
103 |     md: "size-5",
104 |     lg: "size-6",
105 |   }
106 | 
107 |   return (
108 |     <div className={cn("relative", sizeClasses[size], className)}>
109 |       <div className="border-primary absolute inset-0 animate-[thin-pulse_1.5s_ease-in-out_infinite] rounded-full border-2" />
110 |       <span className="sr-only">Loading</span>
111 |     </div>
112 |   )
113 | }
114 | 
115 | export function PulseDotLoader({
116 |   className,
117 |   size = "md",
118 | }: {
119 |   className?: string
120 |   size?: "sm" | "md" | "lg"
121 | }) {
122 |   const sizeClasses = {
123 |     sm: "size-1",
124 |     md: "size-2",
125 |     lg: "size-3",
126 |   }
127 | 
128 |   return (
129 |     <div
130 |       className={cn(
131 |         "bg-primary animate-[pulse-dot_1.2s_ease-in-out_infinite] rounded-full",
132 |         sizeClasses[size],
133 |         className
134 |       )}
135 |     >
136 |       <span className="sr-only">Loading</span>
137 |     </div>
138 |   )
139 | }
140 | 
141 | export function DotsLoader({
142 |   className,
143 |   size = "md",
144 | }: {
145 |   className?: string
146 |   size?: "sm" | "md" | "lg"
147 | }) {
148 |   const dotSizes = {
149 |     sm: "h-1.5 w-1.5",
150 |     md: "h-2 w-2",
151 |     lg: "h-2.5 w-2.5",
152 |   }
153 | 
154 |   const containerSizes = {
155 |     sm: "h-4",
156 |     md: "h-5",
157 |     lg: "h-6",
158 |   }
159 | 
160 |   return (
161 |     <div
162 |       className={cn(
163 |         "flex items-center space-x-1",
164 |         containerSizes[size],
165 |         className
166 |       )}
167 |     >
168 |       {[...Array(3)].map((_, i) => (
169 |         <div
170 |           key={i}
171 |           className={cn(
172 |             "bg-primary animate-[bounce-dots_1.4s_ease-in-out_infinite] rounded-full",
173 |             dotSizes[size]
174 |           )}
175 |           style={{
176 |             animationDelay: `${i * 160}ms`,
177 |           }}
178 |         />
179 |       ))}
180 |       <span className="sr-only">Loading</span>
181 |     </div>
182 |   )
183 | }
184 | 
185 | export function TypingLoader({
186 |   className,
187 |   size = "md",
188 | }: {
189 |   className?: string
190 |   size?: "sm" | "md" | "lg"
191 | }) {
192 |   const dotSizes = {
193 |     sm: "h-1 w-1",
194 |     md: "h-1.5 w-1.5",
195 |     lg: "h-2 w-2",
196 |   }
197 | 
198 |   const containerSizes = {
199 |     sm: "h-4",
200 |     md: "h-5",
201 |     lg: "h-6",
202 |   }
203 | 
204 |   return (
205 |     <div
206 |       className={cn(
207 |         "flex items-center space-x-1",
208 |         containerSizes[size],
209 |         className
210 |       )}
211 |     >
212 |       {[...Array(3)].map((_, i) => (
213 |         <div
214 |           key={i}
215 |           className={cn(
216 |             "bg-primary animate-[typing_1s_infinite] rounded-full",
217 |             dotSizes[size]
218 |           )}
219 |           style={{
220 |             animationDelay: `${i * 250}ms`,
221 |           }}
222 |         />
223 |       ))}
224 |       <span className="sr-only">Loading</span>
225 |     </div>
226 |   )
227 | }
228 | 
229 | export function WaveLoader({
230 |   className,
231 |   size = "md",
232 | }: {
233 |   className?: string
234 |   size?: "sm" | "md" | "lg"
235 | }) {
236 |   const barWidths = {
237 |     sm: "w-0.5",
238 |     md: "w-0.5",
239 |     lg: "w-1",
240 |   }
241 | 
242 |   const containerSizes = {
243 |     sm: "h-4",
244 |     md: "h-5",
245 |     lg: "h-6",
246 |   }
247 | 
248 |   const heights = {
249 |     sm: ["6px", "9px", "12px", "9px", "6px"],
250 |     md: ["8px", "12px", "16px", "12px", "8px"],
251 |     lg: ["10px", "15px", "20px", "15px", "10px"],
252 |   }
253 | 
254 |   return (
255 |     <div
256 |       className={cn(
257 |         "flex items-center gap-0.5",
258 |         containerSizes[size],
259 |         className
260 |       )}
261 |     >
262 |       {[...Array(5)].map((_, i) => (
263 |         <div
264 |           key={i}
265 |           className={cn(
266 |             "bg-primary animate-[wave_1s_ease-in-out_infinite] rounded-full",
267 |             barWidths[size]
268 |           )}
269 |           style={{
270 |             animationDelay: `${i * 100}ms`,
271 |             height: heights[size][i],
272 |           }}
273 |         />
274 |       ))}
275 |       <span className="sr-only">Loading</span>
276 |     </div>
277 |   )
278 | }
279 | 
280 | export function BarsLoader({
281 |   className,
282 |   size = "md",
283 | }: {
284 |   className?: string
285 |   size?: "sm" | "md" | "lg"
286 | }) {
287 |   const barWidths = {
288 |     sm: "w-1",
289 |     md: "w-1.5",
290 |     lg: "w-2",
291 |   }
292 | 
293 |   const containerSizes = {
294 |     sm: "h-4 gap-1",
295 |     md: "h-5 gap-1.5",
296 |     lg: "h-6 gap-2",
297 |   }
298 | 
299 |   return (
300 |     <div className={cn("flex", containerSizes[size], className)}>
301 |       {[...Array(3)].map((_, i) => (
302 |         <div
303 |           key={i}
304 |           className={cn(
305 |             "bg-primary h-full animate-[wave-bars_1.2s_ease-in-out_infinite]",
306 |             barWidths[size]
307 |           )}
308 |           style={{
309 |             animationDelay: `${i * 0.2}s`,
310 |           }}
311 |         />
312 |       ))}
313 |       <span className="sr-only">Loading</span>
314 |     </div>
315 |   )
316 | }
317 | 
318 | export function TerminalLoader({
319 |   className,
320 |   size = "md",
321 | }: {
322 |   className?: string
323 |   size?: "sm" | "md" | "lg"
324 | }) {
325 |   const cursorSizes = {
326 |     sm: "h-3 w-1.5",
327 |     md: "h-4 w-2",
328 |     lg: "h-5 w-2.5",
329 |   }
330 | 
331 |   const textSizes = {
332 |     sm: "text-xs",
333 |     md: "text-sm",
334 |     lg: "text-base",
335 |   }
336 | 
337 |   const containerSizes = {
338 |     sm: "h-4",
339 |     md: "h-5",
340 |     lg: "h-6",
341 |   }
342 | 
343 |   return (
344 |     <div
345 |       className={cn(
346 |         "flex items-center space-x-1",
347 |         containerSizes[size],
348 |         className
349 |       )}
350 |     >
351 |       <span className={cn("text-primary font-mono", textSizes[size])}>
352 |         {">"}
353 |       </span>
354 |       <div
355 |         className={cn(
356 |           "bg-primary animate-[blink_1s_step-end_infinite]",
357 |           cursorSizes[size]
358 |         )}
359 |       />
360 |       <span className="sr-only">Loading</span>
361 |     </div>
362 |   )
363 | }
364 | 
365 | export function TextBlinkLoader({
366 |   text = "Thinking",
367 |   className,
368 |   size = "md",
369 | }: {
370 |   text?: string
371 |   className?: string
372 |   size?: "sm" | "md" | "lg"
373 | }) {
374 |   const textSizes = {
375 |     sm: "text-xs",
376 |     md: "text-sm",
377 |     lg: "text-base",
378 |   }
379 | 
380 |   return (
381 |     <div
382 |       className={cn(
383 |         "animate-[text-blink_2s_ease-in-out_infinite] font-medium",
384 |         textSizes[size],
385 |         className
386 |       )}
387 |     >
388 |       {text}
389 |     </div>
390 |   )
391 | }
392 | 
393 | export function TextShimmerLoader({
394 |   text = "Thinking",
395 |   className,
396 |   size = "md",
397 | }: {
398 |   text?: string
399 |   className?: string
400 |   size?: "sm" | "md" | "lg"
401 | }) {
402 |   const textSizes = {
403 |     sm: "text-xs",
404 |     md: "text-sm",
405 |     lg: "text-base",
406 |   }
407 | 
408 |   return (
409 |     <div
410 |       className={cn(
411 |         "bg-[linear-gradient(to_right,var(--muted-foreground)_40%,var(--foreground)_60%,var(--muted-foreground)_80%)]",
412 |         "bg-size-[200%_auto] bg-clip-text font-medium text-transparent",
413 |         "animate-[shimmer_4s_infinite_linear]",
414 |         textSizes[size],
415 |         className
416 |       )}
417 |     >
418 |       {text}
419 |     </div>
420 |   )
421 | }
422 | 
423 | export function TextDotsLoader({
424 |   className,
425 |   text = "Thinking",
426 |   size = "md",
427 | }: {
428 |   className?: string
429 |   text?: string
430 |   size?: "sm" | "md" | "lg"
431 | }) {
432 |   const textSizes = {
433 |     sm: "text-xs",
434 |     md: "text-sm",
435 |     lg: "text-base",
436 |   }
437 | 
438 |   return (
439 |     <div
440 |       className={cn("inline-flex items-center", className)}
441 |     >
442 |       <span className={cn("text-primary font-medium", textSizes[size])}>
443 |         {text}
444 |       </span>
445 |       <span className="inline-flex">
446 |         <span className="text-primary animate-[loading-dots_1.4s_infinite_0.2s]">
447 |           .
448 |         </span>
449 |         <span className="text-primary animate-[loading-dots_1.4s_infinite_0.4s]">
450 |           .
451 |         </span>
452 |         <span className="text-primary animate-[loading-dots_1.4s_infinite_0.6s]">
453 |           .
454 |         </span>
455 |       </span>
456 |     </div>
457 |   )
458 | }
459 | 
460 | function Loader({
461 |   variant = "circular",
462 |   size = "md",
463 |   text,
464 |   className,
465 | }: LoaderProps) {
466 |   switch (variant) {
467 |     case "circular":
468 |       return <CircularLoader size={size} className={className} />
469 |     case "classic":
470 |       return <ClassicLoader size={size} className={className} />
471 |     case "pulse":
472 |       return <PulseLoader size={size} className={className} />
473 |     case "pulse-dot":
474 |       return <PulseDotLoader size={size} className={className} />
475 |     case "dots":
476 |       return <DotsLoader size={size} className={className} />
477 |     case "typing":
478 |       return <TypingLoader size={size} className={className} />
479 |     case "wave":
480 |       return <WaveLoader size={size} className={className} />
481 |     case "bars":
482 |       return <BarsLoader size={size} className={className} />
483 |     case "terminal":
484 |       return <TerminalLoader size={size} className={className} />
485 |     case "text-blink":
486 |       return <TextBlinkLoader text={text} size={size} className={className} />
487 |     case "text-shimmer":
488 |       return <TextShimmerLoader text={text} size={size} className={className} />
489 |     case "loading-dots":
490 |       return <TextDotsLoader text={text} size={size} className={className} />
491 |     default:
492 |       return <CircularLoader size={size} className={className} />
493 |   }
494 | }
495 | 
496 | export { Loader }
497 |

--------------------------------------------------------------------------------
/Knowmore/services/search_orchestrator.py:
--------------------------------------------------------------------------------
  1 | import json
  2 | import asyncio
  3 | from datetime import datetime
  4 | from typing import Optional, Dict, Any, List
  5 | from .claude_service import ClaudeService
  6 | from .web_search_firecrawl import FirecrawlWebSearch
  7 | 
  8 | 
  9 | class SearchOrchestrator:
 10 |     """
 11 |     Orchestrates web search functionality including query extraction and search execution.
 12 |     Separates search logic from streaming concerns.
 13 |     """
 14 |     
 15 |     def __init__(self):
 16 |         self.claude_service = ClaudeService()
 17 |         self.search_tool = FirecrawlWebSearch()
 18 |     
 19 |     
 20 |     async def generate_search_queries(self, messages: List[Dict[str, Any]]) -> List[str]:
 21 |         """Generate 3 different web search queries based on user message using LLM"""
 22 |         last_message = self._get_last_user_message(messages)
 23 |         if not last_message:
 24 |             return []
 25 |         
 26 |         # Get conversation context (last 3-5 messages)
 27 |         conversation_context = self._get_conversation_context(messages, max_messages=5)
 28 |         
 29 |         # Get current date information
 30 |         current_date = datetime.now()
 31 |         date_context = f"""Current date: {current_date.strftime('%B %d, %Y')}
 32 | Current year: {current_date.year}
 33 | Current month: {current_date.strftime('%B')}"""
 34 |         
 35 |         query_messages = [
 36 |             {
 37 |                 "role": "user",
 38 |                 "content": f"""Generate 3 different web search queries based on the user's latest message within the context of the conversation. 
 39 | The queries should approach the topic from different angles or search for different aspects.
 40 | 
 41 | {date_context}
 42 | 
 43 | Conversation context:
 44 | {conversation_context}
 45 | 
 46 | Latest user message: {last_message}
 47 | 
 48 | Important guidelines:
 49 | - Consider the conversation context to understand what the user is really asking about
 50 | - If the latest message refers to "it", "this", "that", etc., use the context to understand what is being referenced
 51 | - If the user asks about "latest", "recent", "current", or "new" information, include the current year ({current_date.year}) or month in relevant queries
 52 | - For time-sensitive topics, add appropriate year or date qualifiers
 53 | - For comparisons or updates, consider including the current year
 54 | - Only add date/year when it makes the search more relevant
 55 | 
 56 | Return ONLY the 3 search queries, one per line, no explanation or numbering:"""
 57 |             }
 58 |         ]
 59 |         
 60 |         try:
 61 |             response = await self.claude_service.client.messages.create(
 62 |                 model="claude-3-5-haiku-latest",
 63 |                 max_tokens=150,
 64 |                 messages=query_messages
 65 |             )
 66 |             
 67 |             queries_text = response.content[0].text.strip()
 68 |             # Split by newlines and clean up
 69 |             queries = [q.strip() for q in queries_text.split('\n') if q.strip()]
 70 |             
 71 |             # Basic validation and limit to 3
 72 |             valid_queries = []
 73 |             for query in queries[:3]:
 74 |                 if 5 < len(query) < 100:
 75 |                     valid_queries.append(query)
 76 |             
 77 |             # Log generated search queries
 78 |             print("\n" + "="*60)
 79 |             print("🔍 GENERATED SEARCH QUERIES:")
 80 |             for i, query in enumerate(valid_queries, 1):
 81 |                 print(f"  {i}. {query}")
 82 |             print("="*60 + "\n")
 83 |             
 84 |             return valid_queries
 85 |                 
 86 |         except Exception as e:
 87 |             print(f"Query generation failed: {e}")
 88 |         
 89 |         return []
 90 |     
 91 |     async def execute_search(self, query: str) -> Dict[str, Any]:
 92 |         """Execute web search and return formatted results"""
 93 |         try:
 94 |             result = await self.search_tool.execute(
 95 |                 query=query,
 96 |                 limit=3,  # Reduced from 5 to 3 since we're doing 3 searches
 97 |                 scrape_content=True,
 98 |                 formats=["markdown"]
 99 |             )
100 |             
101 |             if result.get("success"):
102 |                 search_results = result.get("results", [])
103 |                 return {
104 |                     "results": search_results,
105 |                     "filterTags": [],
106 |                     "summary": f"Found {len(search_results)} relevant sources about {query}",
107 |                     "success": True,
108 |                     "query": query
109 |                 }
110 |             else:
111 |                 return {
112 |                     "results": [],
113 |                     "filterTags": [],
114 |                     "summary": f"Search failed: {result.get('error', 'Unknown error')}",
115 |                     "success": False,
116 |                     "query": query
117 |                 }
118 |                 
119 |         except Exception as e:
120 |             return {
121 |                 "results": [],
122 |                 "filterTags": [],
123 |                 "summary": f"Search error: {str(e)}",
124 |                 "success": False,
125 |                 "query": query
126 |             }
127 |     
128 |     async def execute_multiple_searches(self, queries: List[str]) -> List[Dict[str, Any]]:
129 |         """Execute multiple searches concurrently"""
130 |         if not queries:
131 |             return []
132 |         
133 |         print("\n" + "="*60)
134 |         print("🚀 EXECUTING CONCURRENT SEARCHES...")
135 |         print("="*60)
136 |         
137 |         # Create tasks for all queries
138 |         tasks = [self.execute_search(query) for query in queries]
139 |         
140 |         # Execute all searches concurrently
141 |         results = await asyncio.gather(*tasks, return_exceptions=True)
142 |         
143 |         # Filter out exceptions and return valid results
144 |         valid_results = []
145 |         for i, (query, result) in enumerate(zip(queries, results), 1):
146 |             if isinstance(result, dict) and not isinstance(result, Exception):
147 |                 valid_results.append(result)
148 |                 
149 |                 # Log search results
150 |                 print(f"\n📊 Search #{i}: '{query}'")
151 |                 print(f"   Status: {'✅ Success' if result.get('success') else '❌ Failed'}")
152 |                 if result.get('success'):
153 |                     results_count = len(result.get('results', []))
154 |                     print(f"   Results: {results_count} sources found")
155 |                     
156 |                     # Log top 3 results
157 |                     for j, res in enumerate(result.get('results', [])[:3], 1):
158 |                         print(f"\n   Result {j}:")
159 |                         print(f"     Title: {res.get('title', 'N/A')}")
160 |                         print(f"     URL: {res.get('url', 'N/A')}")
161 |                         if res.get('description'):
162 |                             desc = res.get('description', '')[:100] + '...' if len(res.get('description', '')) > 100 else res.get('description', '')
163 |                             print(f"     Preview: {desc}")
164 |                 else:
165 |                     print(f"   Error: {result.get('summary', 'Unknown error')}")
166 |             else:
167 |                 print(f"\n❌ Search #{i}: '{queries[i-1]}' - Failed with exception: {result}")
168 |         
169 |         print("\n" + "="*60)
170 |         print(f"✨ SEARCH COMPLETE: {len(valid_results)}/{len(queries)} searches successful")
171 |         print("="*60 + "\n")
172 |         
173 |         return valid_results
174 |     
175 |     def enhance_messages_with_search(
176 |         self, 
177 |         messages: List[Dict[str, Any]], 
178 |         search_results: Dict[str, Any], 
179 |         query: str
180 |     ) -> List[Dict[str, Any]]:
181 |         """Inject search results into conversation context"""
182 |         if not search_results.get("results"):
183 |             return messages
184 |         
185 |         # Build search context
186 |         context_parts = [f"Search results for '{query}':"]
187 |         
188 |         for i, result in enumerate(search_results["results"][:5]):
189 |             title = result.get("title", "")
190 |             url = result.get("url", "")
191 |             content = result.get("markdown", result.get("description", ""))
192 |             
193 |             context_parts.append(f"\n{i+1}. {title}")
194 |             context_parts.append(f"   URL: {url}")
195 |             if content:
196 |                 # Limit content length
197 |                 content = content[:500] + "..." if len(content) > 500 else content
198 |                 context_parts.append(f"   Content: {content}")
199 |         
200 |         search_context = "\n".join(context_parts)
201 |         
202 |         enhanced_messages = []
203 |         
204 |         # Copy all messages except the last user message
205 |         for i, msg in enumerate(messages[:-1]):
206 |             enhanced_messages.append(msg)
207 |         
208 |         # Get the last user message and enhance it with search context
209 |         last_message = messages[-1]
210 |         enhanced_content = f"{last_message['content']}\n\n---\n{search_context}"
211 |         
212 |         enhanced_messages.append({
213 |             "role": last_message["role"],
214 |             "content": enhanced_content
215 |         })
216 |         
217 |         return enhanced_messages
218 |     
219 |     def enhance_messages_with_multiple_searches(
220 |         self, 
221 |         messages: List[Dict[str, Any]], 
222 |         search_results_list: List[Dict[str, Any]]
223 |     ) -> List[Dict[str, Any]]:
224 |         """Inject multiple search results into conversation context"""
225 |         if not search_results_list or not any(sr.get("results") for sr in search_results_list):
226 |             return messages
227 |         
228 |         # Build combined search context
229 |         context_parts = ["Web search results from multiple queries:"]
230 |         
231 |         result_counter = 1
232 |         for search_result in search_results_list:
233 |             if not search_result.get("results"):
234 |                 continue
235 |                 
236 |             query = search_result.get("query", "Unknown query")
237 |             context_parts.append(f"\n## Search: '{query}'")
238 |             
239 |             for result in search_result["results"][:3]:  # Limit to 3 results per search
240 |                 title = result.get("title", "")
241 |                 url = result.get("url", "")
242 |                 content = result.get("markdown", result.get("description", ""))
243 |                 
244 |                 context_parts.append(f"\n{result_counter}. {title}")
245 |                 context_parts.append(f"   URL: {url}")
246 |                 if content:
247 |                     # Limit content length
248 |                     content = content[:300] + "..." if len(content) > 300 else content
249 |                     context_parts.append(f"   Content: {content}")
250 |                 
251 |                 result_counter += 1
252 |         
253 |         search_context = "\n".join(context_parts)
254 |         
255 |         enhanced_messages = []
256 |         
257 |         # Copy all messages except the last user message
258 |         for i, msg in enumerate(messages[:-1]):
259 |             enhanced_messages.append(msg)
260 |         
261 |         # Get the last user message and enhance it with search context
262 |         last_message = messages[-1]
263 |         enhanced_content = f"{last_message['content']}\n\n---\n{search_context}"
264 |         
265 |         enhanced_messages.append({
266 |             "role": last_message["role"],
267 |             "content": enhanced_content
268 |         })
269 |         
270 |         return enhanced_messages
271 |     
272 |     def _get_last_user_message(self, messages: List[Dict[str, Any]]) -> Optional[str]:
273 |         """Get the last user message content"""
274 |         for msg in reversed(messages):
275 |             if msg.get('role') == 'user':
276 |                 return msg.get('content', '')
277 |         return None
278 |     
279 |     def _get_conversation_context(self, messages: List[Dict[str, Any]], max_messages: int = 5) -> str:
280 |         """Get recent conversation context for better search query generation"""
281 |         if not messages:
282 |             return ""
283 |         
284 |         # Get the last N messages (excluding system messages)
285 |         recent_messages = []
286 |         for msg in reversed(messages):
287 |             if msg.get('role') in ['user', 'assistant']:
288 |                 # Truncate long messages for context
289 |                 content = msg.get('content', '')
290 |                 if len(content) > 200:
291 |                     content = content[:200] + "..."
292 |                 recent_messages.append(f"{msg.get('role').capitalize()}: {content}")
293 |                 
294 |                 if len(recent_messages) >= max_messages:
295 |                     break
296 |         
297 |         # Reverse to show in chronological order
298 |         recent_messages.reverse()
299 |         
300 |         return "\n".join(recent_messages)
301 |

--------------------------------------------------------------------------------