├── frontend
    ├── src
    │   ├── vite-env.d.ts
    │   ├── lib
    │   │   └── utils.ts
    │   ├── main.tsx
    │   ├── components
    │   │   ├── ui
    │   │   │   ├── textarea.tsx
    │   │   │   ├── input.tsx
    │   │   │   ├── badge.tsx
    │   │   │   ├── scroll-area.tsx
    │   │   │   ├── tabs.tsx
    │   │   │   ├── button.tsx
    │   │   │   ├── card.tsx
    │   │   │   └── select.tsx
    │   │   ├── WelcomeScreen.tsx
    │   │   ├── ActivityTimeline.tsx
    │   │   ├── InputForm.tsx
    │   │   └── ChatMessagesView.tsx
    │   ├── global.css
    │   └── App.tsx
    ├── .gitignore
    ├── index.html
    ├── components.json
    ├── tsconfig.node.json
    ├── eslint.config.js
    ├── vite.config.ts
    ├── tsconfig.json
    ├── public
    │   └── vite.svg
    └── package.json
├── backend
    ├── src
    │   └── agent
    │   │   ├── __init__.py
    │   │   ├── tools_and_schemas.py
    │   │   ├── state.py
    │   │   ├── configuration.py
    │   │   ├── app.py
    │   │   ├── prompts.py
    │   │   ├── utils.py
    │   │   └── graph.py
    ├── .env.example
    ├── langgraph.json
    ├── LICENSE
    ├── pyproject.toml
    ├── Makefile
    └── .gitignore
├── .DS_Store
├── demo.jpeg
├── demo2.jpeg
├── Makefile
├── README.md
└── LICENSE


/frontend/src/vite-env.d.ts:
--------------------------------------------------------------------------------
1 | /// <reference types="vite/client" />
2 | 


--------------------------------------------------------------------------------
/backend/src/agent/__init__.py:
--------------------------------------------------------------------------------
1 | from agent.graph import graph
2 | 
3 | __all__ = ["graph"]
4 | 


--------------------------------------------------------------------------------
/.DS_Store:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/daiwk/anyllm-fullstack-langgraph-quickstart/HEAD/.DS_Store


--------------------------------------------------------------------------------
/demo.jpeg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/daiwk/anyllm-fullstack-langgraph-quickstart/HEAD/demo.jpeg


--------------------------------------------------------------------------------
/demo2.jpeg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/daiwk/anyllm-fullstack-langgraph-quickstart/HEAD/demo2.jpeg


--------------------------------------------------------------------------------
/backend/.env.example:
--------------------------------------------------------------------------------
1 | # OPENAI_API_BASE=http://localhost:1234/v1
2 | # OPENAI_API_KEY=lm-studio
3 | # LANGSMITH_API_KEY=xx
4 | # GOOGLE_SEARCH_API_KEY=xx
5 | # GOOGLE_SEARCH_CX=xx
6 | 


--------------------------------------------------------------------------------
/frontend/src/lib/utils.ts:
--------------------------------------------------------------------------------
1 | import { type ClassValue, clsx } from "clsx";
2 | import { twMerge } from "tailwind-merge";
3 | 
4 | export function cn(...inputs: ClassValue[]) {
5 |   return twMerge(clsx(inputs));
6 | }
7 | 


--------------------------------------------------------------------------------
/backend/langgraph.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "dependencies": ["."],
 3 |   "graphs": {
 4 |     "agent": "./src/agent/graph.py:graph"
 5 |   },
 6 |   "http": {
 7 |     "app": "./src/agent/app.py:app"
 8 |   },
 9 |   "env": ".env"
10 | }
11 | 


--------------------------------------------------------------------------------
/frontend/.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 | 


--------------------------------------------------------------------------------
/frontend/src/main.tsx:
--------------------------------------------------------------------------------
 1 | import { StrictMode } from "react";
 2 | import { createRoot } from "react-dom/client";
 3 | import { BrowserRouter } from "react-router-dom";
 4 | import "./global.css";
 5 | import App from "./App.tsx";
 6 | 
 7 | createRoot(document.getElementById("root")!).render(
 8 |   <StrictMode>
 9 |     <BrowserRouter>
10 |       <App />
11 |     </BrowserRouter>
12 |   </StrictMode>
13 | );
14 | 


--------------------------------------------------------------------------------
/frontend/index.html:
--------------------------------------------------------------------------------
 1 | <!doctype html>
 2 | <html lang="en">
 3 |   <head>
 4 |     <meta charset="UTF-8" />
 5 |     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
 6 |     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
 7 |     <title>Vite + React + TS</title>
 8 |   </head>
 9 |   <body>
10 |     <div id="root"></div>
11 |     <script type="module" src="/src/main.tsx"></script>
12 |   </body>
13 | </html>
14 | 


--------------------------------------------------------------------------------
/frontend/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/app.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 | }


--------------------------------------------------------------------------------
/frontend/tsconfig.node.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "compilerOptions": {
 3 |     "tsBuildInfoFile": "./node_modules/.tmp/tsconfig.node.tsbuildinfo",
 4 |     "target": "ES2022",
 5 |     "lib": ["ES2023"],
 6 |     "module": "ESNext",
 7 |     "skipLibCheck": true,
 8 | 
 9 |     /* Bundler mode */
10 |     "moduleResolution": "bundler",
11 |     "allowImportingTsExtensions": true,
12 |     "isolatedModules": true,
13 |     "moduleDetection": "force",
14 |     "noEmit": true,
15 | 
16 |     /* Linting */
17 |     "strict": true,
18 |     "noUnusedLocals": true,
19 |     "noUnusedParameters": true,
20 |     "noFallthroughCasesInSwitch": true,
21 |     "noUncheckedSideEffectImports": true
22 |   },
23 |   "include": ["vite.config.ts"]
24 | }
25 | 


--------------------------------------------------------------------------------
/Makefile:
--------------------------------------------------------------------------------
 1 | .PHONY: help dev-frontend dev-backend dev
 2 | 
 3 | help:
 4 | 	@echo "Available commands:"
 5 | 	@echo "  make dev-frontend    - Starts the frontend development server (Vite)"
 6 | 	@echo "  make dev-backend     - Starts the backend development server (Uvicorn with reload)"
 7 | 	@echo "  make dev             - Starts both frontend and backend development servers"
 8 | 
 9 | dev-frontend:
10 | 	@echo "Starting frontend development server..."
11 | 	@cd frontend && npm run dev
12 | 
13 | dev-backend:
14 | 	@echo "Starting backend development server..."
15 | 	@cd backend && langgraph dev
16 | 
17 | # Run frontend and backend concurrently
18 | dev:
19 | 	@echo "Starting both frontend and backend development servers..."
20 | 	@make dev-frontend & make dev-backend 


--------------------------------------------------------------------------------
/frontend/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 |     <textarea
 8 |       data-slot="textarea"
 9 |       className={cn(
10 |         "border-input placeholder:text-muted-foreground focus-visible:border-ring focus-visible:ring-ring/50 aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive dark:bg-input/30 flex field-sizing-content min-h-16 w-full rounded-md border bg-transparent px-3 py-2 text-base shadow-xs transition-[color,box-shadow] outline-none focus-visible:ring-[3px] disabled:cursor-not-allowed disabled:opacity-50 md:text-sm",
11 |         className
12 |       )}
13 |       {...props}
14 |     />
15 |   )
16 | }
17 | 
18 | export { Textarea }
19 | 


--------------------------------------------------------------------------------
/frontend/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 | 
 7 | export default tseslint.config(
 8 |   { ignores: ['dist'] },
 9 |   {
10 |     extends: [js.configs.recommended, ...tseslint.configs.recommended],
11 |     files: ['**/*.{ts,tsx}'],
12 |     languageOptions: {
13 |       ecmaVersion: 2020,
14 |       globals: globals.browser,
15 |     },
16 |     plugins: {
17 |       'react-hooks': reactHooks,
18 |       'react-refresh': reactRefresh,
19 |     },
20 |     rules: {
21 |       ...reactHooks.configs.recommended.rules,
22 |       'react-refresh/only-export-components': [
23 |         'warn',
24 |         { allowConstantExport: true },
25 |       ],
26 |     },
27 |   },
28 | )
29 | 


--------------------------------------------------------------------------------
/frontend/vite.config.ts:
--------------------------------------------------------------------------------
 1 | import path from "node:path";
 2 | import { defineConfig } from "vite";
 3 | import react from "@vitejs/plugin-react-swc";
 4 | import tailwindcss from "@tailwindcss/vite";
 5 | 
 6 | // https://vitejs.dev/config/
 7 | export default defineConfig({
 8 |   plugins: [react(), tailwindcss()],
 9 |   base: "/app/",
10 |   resolve: {
11 |     alias: {
12 |       "@": path.resolve(new URL(".", import.meta.url).pathname, "./src"),
13 |     },
14 |   },
15 |   server: {
16 |     proxy: {
17 |       // Proxy API requests to the backend server
18 |       "/api": {
19 |         target: "http://127.0.0.1:8000", // Default backend address
20 |         changeOrigin: true,
21 |         // Optionally rewrite path if needed (e.g., remove /api prefix if backend doesn't expect it)
22 |         // rewrite: (path) => path.replace(/^\/api/, ''),
23 |       },
24 |     },
25 |   },
26 | });
27 | 


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


--------------------------------------------------------------------------------
/backend/src/agent/tools_and_schemas.py:
--------------------------------------------------------------------------------
 1 | from typing import List
 2 | from pydantic import BaseModel, Field
 3 | 
 4 | 
 5 | class SearchQueryList(BaseModel):
 6 |     query: List[str] = Field(
 7 |         description="A list of search queries to be used for web research."
 8 |     )
 9 |     rationale: str = Field(
10 |         description="A brief explanation of why these queries are relevant to the research topic."
11 |     )
12 | 
13 | 
14 | class Reflection(BaseModel):
15 |     is_sufficient: bool = Field(
16 |         description="Whether the provided summaries are sufficient to answer the user's question."
17 |     )
18 |     knowledge_gap: str = Field(
19 |         description="A description of what information is missing or needs clarification."
20 |     )
21 |     follow_up_queries: List[str] = Field(
22 |         description="A list of follow-up queries to address the knowledge gap."
23 |     )
24 | 


--------------------------------------------------------------------------------
/frontend/src/components/ui/input.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | 
 3 | import { cn } from "@/lib/utils"
 4 | 
 5 | function Input({ className, type, ...props }: React.ComponentProps<"input">) {
 6 |   return (
 7 |     <input
 8 |       type={type}
 9 |       data-slot="input"
10 |       className={cn(
11 |         "file:text-foreground placeholder:text-muted-foreground selection:bg-primary selection:text-primary-foreground dark:bg-input/30 border-input flex h-9 w-full min-w-0 rounded-md border bg-transparent px-3 py-1 text-base shadow-xs transition-[color,box-shadow] outline-none file:inline-flex file:h-7 file:border-0 file:bg-transparent file:text-sm file:font-medium disabled:pointer-events-none disabled:cursor-not-allowed disabled:opacity-50 md:text-sm",
12 |         "focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px]",
13 |         "aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
14 |         className
15 |       )}
16 |       {...props}
17 |     />
18 |   )
19 | }
20 | 
21 | export { Input }
22 | 


--------------------------------------------------------------------------------
/backend/LICENSE:
--------------------------------------------------------------------------------
 1 | MIT License
 2 | 
 3 | Copyright (c) 2025 Philipp Schmid
 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.
22 | 


--------------------------------------------------------------------------------
/frontend/src/components/WelcomeScreen.tsx:
--------------------------------------------------------------------------------
 1 | import { InputForm } from "./InputForm";
 2 | 
 3 | interface WelcomeScreenProps {
 4 |   handleSubmit: (
 5 |     submittedInputValue: string,
 6 |     effort: string,
 7 |     model: string
 8 |   ) => void;
 9 |   onCancel: () => void;
10 |   isLoading: boolean;
11 | }
12 | 
13 | export const WelcomeScreen: React.FC<WelcomeScreenProps> = ({
14 |   handleSubmit,
15 |   onCancel,
16 |   isLoading,
17 | }) => (
18 |   <div className="flex flex-col items-center justify-center text-center px-4 flex-1 w-full max-w-3xl mx-auto gap-4">
19 |     <div>
20 |       <h1 className="text-5xl md:text-6xl font-semibold text-neutral-100 mb-3">
21 |         Welcome.
22 |       </h1>
23 |       <p className="text-xl md:text-2xl text-neutral-400">
24 |         How can I help you today?
25 |       </p>
26 |     </div>
27 |     <div className="w-full mt-4">
28 |       <InputForm
29 |         onSubmit={handleSubmit}
30 |         isLoading={isLoading}
31 |         onCancel={onCancel}
32 |         hasHistory={false}
33 |       />
34 |     </div>
35 |     <p className="text-xs text-neutral-500">
36 |       Powered by LM Studio (Qwen) and LangChain LangGraph.
37 |     </p>
38 |   </div>
39 | );
40 | 


--------------------------------------------------------------------------------
/backend/src/agent/state.py:
--------------------------------------------------------------------------------
 1 | from __future__ import annotations
 2 | 
 3 | from dataclasses import dataclass, field
 4 | from typing import TypedDict
 5 | 
 6 | from langgraph.graph import add_messages
 7 | from typing_extensions import Annotated
 8 | 
 9 | 
10 | import operator
11 | from dataclasses import dataclass, field
12 | from typing_extensions import Annotated
13 | 
14 | 
15 | class OverallState(TypedDict):
16 |     messages: Annotated[list, add_messages]
17 |     search_query: Annotated[list, operator.add]
18 |     web_research_result: Annotated[list, operator.add]
19 |     sources_gathered: Annotated[list, operator.add]
20 |     initial_search_query_count: int
21 |     max_research_loops: int
22 |     research_loop_count: int
23 |     reasoning_model: str
24 | 
25 | 
26 | class ReflectionState(TypedDict):
27 |     is_sufficient: bool
28 |     knowledge_gap: str
29 |     follow_up_queries: Annotated[list, operator.add]
30 |     research_loop_count: int
31 |     number_of_ran_queries: int
32 | 
33 | 
34 | class Query(TypedDict):
35 |     query: str
36 |     rationale: str
37 | 
38 | 
39 | class QueryGenerationState(TypedDict):
40 |     query_list: list[Query]
41 | 
42 | 
43 | class WebSearchState(TypedDict):
44 |     search_query: str
45 |     id: str
46 | 
47 | 
48 | @dataclass(kw_only=True)
49 | class SearchStateOutput:
50 |     running_summary: str = field(default=None)  # Final report
51 | 


--------------------------------------------------------------------------------
/frontend/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>


--------------------------------------------------------------------------------
/frontend/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 |     "@langchain/core": "^0.3.55",
14 |     "@langchain/langgraph-sdk": "^0.0.74",
15 |     "@radix-ui/react-scroll-area": "^1.2.8",
16 |     "@radix-ui/react-select": "^2.2.4",
17 |     "@radix-ui/react-slot": "^1.2.2",
18 |     "@radix-ui/react-tabs": "^1.1.11",
19 |     "@radix-ui/react-tooltip": "^1.2.6",
20 |     "@tailwindcss/vite": "^4.1.5",
21 |     "class-variance-authority": "^0.7.1",
22 |     "clsx": "^2.1.1",
23 |     "lucide-react": "^0.508.0",
24 |     "react": "^19.0.0",
25 |     "react-dom": "^19.0.0",
26 |     "react-markdown": "^9.0.3",
27 |     "react-router-dom": "^7.5.3",
28 |     "tailwind-merge": "^3.2.0",
29 |     "tailwindcss": "^4.1.5"
30 |   },
31 |   "devDependencies": {
32 |     "@eslint/js": "^9.22.0",
33 |     "@types/node": "^22.15.17",
34 |     "@types/react": "^19.1.2",
35 |     "@types/react-dom": "^19.1.3",
36 |     "@vitejs/plugin-react-swc": "^3.9.0",
37 |     "eslint": "^9.22.0",
38 |     "eslint-plugin-react-hooks": "^5.2.0",
39 |     "eslint-plugin-react-refresh": "^0.4.19",
40 |     "globals": "^16.0.0",
41 |     "tw-animate-css": "^1.2.9",
42 |     "typescript": "~5.7.2",
43 |     "typescript-eslint": "^8.26.1",
44 |     "vite": "^6.3.4"
45 |   }
46 | }
47 | 


--------------------------------------------------------------------------------
/backend/pyproject.toml:
--------------------------------------------------------------------------------
 1 | [project]
 2 | name = "agent"
 3 | version = "0.0.1"
 4 | description = "Backend for the LangGraph agent"
 5 | authors = [
 6 |     { name = "Philipp Schmid", email = "schmidphilipp1995@gmail.com" },
 7 | ]
 8 | readme = "README.md"
 9 | license = { text = "MIT" }
10 | requires-python = ">=3.11,<4.0"
11 | dependencies = [
12 |     "langgraph>=0.2.6",
13 |     "langchain>=0.3.19",
14 |     "langchain-openai",
15 |     "openai",
16 |     "duckduckgo-search>=5.0",
17 |     "python-dotenv>=1.0.1",
18 |     "langgraph-sdk>=0.1.57",
19 |     "langgraph-cli",
20 |     "langgraph-api",
21 |     "fastapi",
22 | ]
23 | 
24 | 
25 | [project.optional-dependencies]
26 | dev = ["mypy>=1.11.1", "ruff>=0.6.1"]
27 | 
28 | [build-system]
29 | requires = ["setuptools>=73.0.0", "wheel"]
30 | build-backend = "setuptools.build_meta"
31 | 
32 | [tool.ruff]
33 | lint.select = [
34 |     "E",    # pycodestyle
35 |     "F",    # pyflakes
36 |     "I",    # isort
37 |     "D",    # pydocstyle
38 |     "D401", # First line should be in imperative mood
39 |     "T201",
40 |     "UP",
41 | ]
42 | lint.ignore = [
43 |     "UP006",
44 |     "UP007",
45 |     # We actually do want to import from typing_extensions
46 |     "UP035",
47 |     # Relax the convention by _not_ requiring documentation for every function parameter.
48 |     "D417",
49 |     "E501",
50 | ]
51 | [tool.ruff.lint.per-file-ignores]
52 | "tests/*" = ["D", "UP"]
53 | [tool.ruff.lint.pydocstyle]
54 | convention = "google"
55 | 
56 | [dependency-groups]
57 | dev = [
58 |     "langgraph-cli[inmem]>=0.1.71",
59 |     "pytest>=8.3.5",
60 | ]
61 | 


--------------------------------------------------------------------------------
/frontend/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 | 


--------------------------------------------------------------------------------
/frontend/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 | 


--------------------------------------------------------------------------------
/backend/src/agent/configuration.py:
--------------------------------------------------------------------------------
 1 | import os
 2 | from typing import Any, Optional
 3 | 
 4 | from langchain_core.runnables import RunnableConfig
 5 | from pydantic import BaseModel, Field
 6 | 
 7 | 
 8 | class Configuration(BaseModel):
 9 |     """The configuration for the agent."""
10 | 
11 |     query_generator_model: str = Field(
12 |         default="qwen2.5-coder-7b-instruct",
13 |         metadata={
14 |             "description": "The name of the language model to use for the agent's query generation."
15 |         },
16 |     )
17 | 
18 |     reflection_model: str = Field(
19 |         default="qwen2.5-coder-7b-instruct",
20 |         metadata={
21 |             "description": "The name of the language model to use for the agent's reflection."
22 |         },
23 |     )
24 | 
25 |     answer_model: str = Field(
26 |         default="qwen2.5-coder-7b-instruct",
27 |         metadata={
28 |             "description": "The name of the language model to use for the agent's answer."
29 |         },
30 |     )
31 | 
32 |     number_of_initial_queries: int = Field(
33 |         default=3,
34 |         metadata={"description": "The number of initial search queries to generate."},
35 |     )
36 | 
37 |     max_research_loops: int = Field(
38 |         default=2,
39 |         metadata={"description": "The maximum number of research loops to perform."},
40 |     )
41 | 
42 |     @classmethod
43 |     def from_runnable_config(
44 |         cls, config: Optional[RunnableConfig] = None
45 |     ) -> "Configuration":
46 |         """Create a Configuration instance from a RunnableConfig."""
47 |         configurable = (
48 |             config["configurable"] if config and "configurable" in config else {}
49 |         )
50 | 
51 |         # Get raw values from environment or config
52 |         raw_values: dict[str, Any] = {
53 |             name: os.environ.get(name.upper(), configurable.get(name))
54 |             for name in cls.model_fields.keys()
55 |         }
56 | 
57 |         # Filter out None values
58 |         values = {k: v for k, v in raw_values.items() if v is not None}
59 | 
60 |         return cls(**values)
61 | 


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | # anyllm-fullstack-langgraph-quickstart
 2 | 
 3 | ## 简介
 4 | 
 5 | 基于[google-gemini/gemini-fullstack-langgraph-quickstart](https://github.com/google-gemini/gemini-fullstack-langgraph-quickstart)
 6 | 
 7 | 主要做了如下几个改动：
 8 | 1. 支持本地llm(例如lm-studio启动一个qwen2.5-coder-7b-instruct),同时把gemini的api接口改成了openai形式的接口【这个代码全是codex帮改的】
 9 | 2. 工具调用改成google search api进行搜索(比较慢，duckduckgo也可以，代码里DDGS相关的代码，只是会经常超限…)【这个代码是在codex改完的基础上，参考[https://github.com/leptonai/search_with_lepton/blob/main/search_with_lepton.py](https://github.com/leptonai/search_with_lepton/blob/main/search_with_lepton.py)里的google search api改的，只是我之前试过了google的所以复制了这个，当然里面还有别的search api，可以后续再试试】
10 | 3. 为了方便完全的本地调试，删掉了一些docker相关的东西
11 | 
12 | ## 使用方式
13 | 
14 | 1. python版本需要是3.11以上
15 | 2. 获取google search api的两个参数
16 |     + api-key：[link](https://developers.google.com/custom-search/v1/introduction?hl=zh-cn#identify_your_application_to_google_with_api_key)
17 |     + cx：[link](https://stackoverflow.com/questions/6562125/getting-a-cx-id-for-custom-search-google-api-python)
18 | 3. LANGSMITH_API_KEY这个可选，去langchain官网可以搞到
19 | 4. 参考```backend/.env.example```，搞一个```backend/.env```，里面的所有注释放开，改成你的base_url（lm-studio的话就是默认的那个1234端口的）、openai_api_key（本地服务就是随便写的）、google api-key、cx和LANGSMITH_API_KEY
20 | 5. 去backend目录，执行```pip3 install .```
21 | 6. 安装node.js，去frontend目录，执行```npm install```
22 | 7. 回到主目录，执行```make dev```，打开```http://localhost:5173/app/```就能用了
23 | 
24 | demo1
25 | 
26 | ![](./demo.jpeg)
27 | 
28 | demo2
29 | 
30 | ![](./demo2.jpeg)
31 | 
32 | 
33 | ## Star History
34 | 
35 | <a href="https://star-history.com/#daiwk/anyllm-fullstack-langgraph-quickstart&Date">
36 |  <picture>
37 |    <source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/svg?repos=daiwk/anyllm-fullstack-langgraph-quickstart&type=Date&theme=dark" />
38 |    <source media="(prefers-color-scheme: light)" srcset="https://api.star-history.com/svg?repos=daiwk/anyllm-fullstack-langgraph-quickstart&type=Date" />
39 |    <img alt="Star History Chart" src="https://api.star-history.com/svg?repos=daiwk/anyllm-fullstack-langgraph-quickstart&type=Date" />
40 |  </picture>
41 | </a>
42 | 
43 | 


--------------------------------------------------------------------------------
/backend/src/agent/app.py:
--------------------------------------------------------------------------------
 1 | # mypy: disable - error - code = "no-untyped-def,misc"
 2 | import pathlib
 3 | from fastapi import FastAPI, Request, Response
 4 | from fastapi.staticfiles import StaticFiles
 5 | import fastapi.exceptions
 6 | 
 7 | # Define the FastAPI app
 8 | app = FastAPI()
 9 | 
10 | 
11 | def create_frontend_router(build_dir="../frontend/dist"):
12 |     """Creates a router to serve the React frontend.
13 | 
14 |     Args:
15 |         build_dir: Path to the React build directory relative to this file.
16 | 
17 |     Returns:
18 |         A Starlette application serving the frontend.
19 |     """
20 |     build_path = pathlib.Path(__file__).parent.parent.parent / build_dir
21 |     static_files_path = build_path / "assets"  # Vite uses 'assets' subdir
22 | 
23 |     if not build_path.is_dir() or not (build_path / "index.html").is_file():
24 |         print(
25 |             f"WARN: Frontend build directory not found or incomplete at {build_path}. Serving frontend will likely fail."
26 |         )
27 |         # Return a dummy router if build isn't ready
28 |         from starlette.routing import Route
29 | 
30 |         async def dummy_frontend(request):
31 |             return Response(
32 |                 "Frontend not built. Run 'npm run build' in the frontend directory.",
33 |                 media_type="text/plain",
34 |                 status_code=503,
35 |             )
36 | 
37 |         return Route("/{path:path}", endpoint=dummy_frontend)
38 | 
39 |     build_dir = pathlib.Path(build_dir)
40 | 
41 |     react = FastAPI(openapi_url="")
42 |     react.mount(
43 |         "/assets", StaticFiles(directory=static_files_path), name="static_assets"
44 |     )
45 | 
46 |     @react.get("/{path:path}")
47 |     async def handle_catch_all(request: Request, path: str):
48 |         fp = build_path / path
49 |         if not fp.exists() or not fp.is_file():
50 |             fp = build_path / "index.html"
51 |         return fastapi.responses.FileResponse(fp)
52 | 
53 |     return react
54 | 
55 | 
56 | # Mount the frontend under /app to not conflict with the LangGraph API routes
57 | app.mount(
58 |     "/app",
59 |     create_frontend_router(),
60 |     name="frontend",
61 | )
62 | 


--------------------------------------------------------------------------------
/frontend/src/components/ui/tabs.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | import * as TabsPrimitive from "@radix-ui/react-tabs"
 3 | 
 4 | import { cn } from "@/lib/utils"
 5 | 
 6 | function Tabs({
 7 |   className,
 8 |   ...props
 9 | }: React.ComponentProps<typeof TabsPrimitive.Root>) {
10 |   return (
11 |     <TabsPrimitive.Root
12 |       data-slot="tabs"
13 |       className={cn("flex flex-col gap-2", className)}
14 |       {...props}
15 |     />
16 |   )
17 | }
18 | 
19 | function TabsList({
20 |   className,
21 |   ...props
22 | }: React.ComponentProps<typeof TabsPrimitive.List>) {
23 |   return (
24 |     <TabsPrimitive.List
25 |       data-slot="tabs-list"
26 |       className={cn(
27 |         "bg-muted text-muted-foreground inline-flex h-9 w-fit items-center justify-center rounded-lg p-[3px]",
28 |         className
29 |       )}
30 |       {...props}
31 |     />
32 |   )
33 | }
34 | 
35 | function TabsTrigger({
36 |   className,
37 |   ...props
38 | }: React.ComponentProps<typeof TabsPrimitive.Trigger>) {
39 |   return (
40 |     <TabsPrimitive.Trigger
41 |       data-slot="tabs-trigger"
42 |       className={cn(
43 |         "data-[state=active]:bg-background dark:data-[state=active]:text-foreground focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:outline-ring dark:data-[state=active]:border-input dark:data-[state=active]:bg-input/30 text-foreground dark:text-muted-foreground inline-flex h-[calc(100%-1px)] flex-1 items-center justify-center gap-1.5 rounded-md border border-transparent px-2 py-1 text-sm font-medium whitespace-nowrap transition-[color,box-shadow] focus-visible:ring-[3px] focus-visible:outline-1 disabled:pointer-events-none disabled:opacity-50 data-[state=active]:shadow-sm [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
44 |         className
45 |       )}
46 |       {...props}
47 |     />
48 |   )
49 | }
50 | 
51 | function TabsContent({
52 |   className,
53 |   ...props
54 | }: React.ComponentProps<typeof TabsPrimitive.Content>) {
55 |   return (
56 |     <TabsPrimitive.Content
57 |       data-slot="tabs-content"
58 |       className={cn("flex-1 outline-none", className)}
59 |       {...props}
60 |     />
61 |   )
62 | }
63 | 
64 | export { Tabs, TabsList, TabsTrigger, TabsContent }
65 | 


--------------------------------------------------------------------------------
/backend/Makefile:
--------------------------------------------------------------------------------
 1 | .PHONY: all format lint test tests test_watch integration_tests docker_tests help extended_tests
 2 | 
 3 | # Default target executed when no arguments are given to make.
 4 | all: help
 5 | 
 6 | # Define a variable for the test file path.
 7 | TEST_FILE ?= tests/unit_tests/
 8 | 
 9 | test:
10 | 	uv run --with-editable . pytest $(TEST_FILE)
11 | 
12 | test_watch:
13 | 	uv run --with-editable . ptw --snapshot-update --now . -- -vv tests/unit_tests
14 | 
15 | test_profile:
16 | 	uv run --with-editable . pytest -vv tests/unit_tests/ --profile-svg
17 | 
18 | extended_tests:
19 | 	uv run --with-editable . pytest --only-extended $(TEST_FILE)
20 | 
21 | 
22 | ######################
23 | # LINTING AND FORMATTING
24 | ######################
25 | 
26 | # Define a variable for Python and notebook files.
27 | PYTHON_FILES=src/
28 | MYPY_CACHE=.mypy_cache
29 | lint format: PYTHON_FILES=.
30 | lint_diff format_diff: PYTHON_FILES=$(shell git diff --name-only --diff-filter=d main | grep -E '\.py$$|\.ipynb$$')
31 | lint_package: PYTHON_FILES=src
32 | lint_tests: PYTHON_FILES=tests
33 | lint_tests: MYPY_CACHE=.mypy_cache_test
34 | 
35 | lint lint_diff lint_package lint_tests:
36 | 	uv run ruff check .
37 | 	[ "$(PYTHON_FILES)" = "" ] || uv run ruff format $(PYTHON_FILES) --diff
38 | 	[ "$(PYTHON_FILES)" = "" ] || uv run ruff check --select I $(PYTHON_FILES)
39 | 	[ "$(PYTHON_FILES)" = "" ] || uv run mypy --strict $(PYTHON_FILES)
40 | 	[ "$(PYTHON_FILES)" = "" ] || mkdir -p $(MYPY_CACHE) && uv run mypy --strict $(PYTHON_FILES) --cache-dir $(MYPY_CACHE)
41 | 
42 | format format_diff:
43 | 	uv run ruff format $(PYTHON_FILES)
44 | 	uv run ruff check --select I --fix $(PYTHON_FILES)
45 | 
46 | spell_check:
47 | 	codespell --toml pyproject.toml
48 | 
49 | spell_fix:
50 | 	codespell --toml pyproject.toml -w
51 | 
52 | ######################
53 | # HELP
54 | ######################
55 | 
56 | help:
57 | 	@echo '----'
58 | 	@echo 'format                       - run code formatters'
59 | 	@echo 'lint                         - run linters'
60 | 	@echo 'test                         - run unit tests'
61 | 	@echo 'tests                        - run unit tests'
62 | 	@echo 'test TEST_FILE=<test_file>   - run all tests in file'
63 | 	@echo 'test_watch                   - run unit tests in watch mode'
64 | 
65 | 


--------------------------------------------------------------------------------
/frontend/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 |   "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 | 


--------------------------------------------------------------------------------
/frontend/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 | 


--------------------------------------------------------------------------------
/backend/.gitignore:
--------------------------------------------------------------------------------
  1 | # Byte-compiled / optimized / DLL files
  2 | __pycache__/
  3 | *.py[cod]
  4 | *$py.class
  5 | uv.lock
  6 | 
  7 | # C extensions
  8 | *.so
  9 | 
 10 | # Distribution / packaging
 11 | .Python
 12 | build/
 13 | develop-eggs/
 14 | dist/
 15 | downloads/
 16 | eggs/
 17 | .eggs/
 18 | lib/
 19 | lib64/
 20 | parts/
 21 | sdist/
 22 | var/
 23 | wheels/
 24 | share/python-wheels/
 25 | *.egg-info/
 26 | .installed.cfg
 27 | *.egg
 28 | MANIFEST
 29 | 
 30 | # PyInstaller
 31 | #  Usually these files are written by a python script from a template
 32 | #  before PyInstaller builds the exe, so as to inject date/other infos into it.
 33 | *.manifest
 34 | *.spec
 35 | 
 36 | # Installer logs
 37 | pip-log.txt
 38 | pip-delete-this-directory.txt
 39 | 
 40 | # Unit test / coverage reports
 41 | htmlcov/
 42 | .tox/
 43 | .nox/
 44 | .coverage
 45 | .coverage.*
 46 | .cache
 47 | nosetests.xml
 48 | coverage.xml
 49 | *.cover
 50 | *.py,cover
 51 | .hypothesis/
 52 | .pytest_cache/
 53 | cover/
 54 | 
 55 | # Translations
 56 | *.mo
 57 | *.pot
 58 | 
 59 | # Django stuff:
 60 | *.log
 61 | local_settings.py
 62 | db.sqlite3
 63 | db.sqlite3-journal
 64 | 
 65 | # Flask stuff:
 66 | instance/
 67 | .webassets-cache
 68 | 
 69 | # Scrapy stuff:
 70 | .scrapy
 71 | 
 72 | # Sphinx documentation
 73 | docs/_build/
 74 | 
 75 | # PyBuilder
 76 | .pybuilder/
 77 | target/
 78 | 
 79 | # Jupyter Notebook
 80 | .ipynb_checkpoints
 81 | 
 82 | # IPython
 83 | profile_default/
 84 | ipython_config.py
 85 | 
 86 | # pyenv
 87 | #   For a library or package, you might want to ignore these files since the code is
 88 | #   intended to run in multiple environments; otherwise, check them in:
 89 | # .python-version
 90 | 
 91 | # pipenv
 92 | #   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
 93 | #   However, in case of collaboration, if having platform-specific dependencies or dependencies
 94 | #   having no cross-platform support, pipenv may install dependencies that don't work, or not
 95 | #   install all needed dependencies.
 96 | #Pipfile.lock
 97 | 
 98 | # poetry
 99 | #   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
100 | #   This is especially recommended for binary packages to ensure reproducibility, and is more
101 | #   commonly ignored for libraries.
102 | #   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
103 | #poetry.lock
104 | 
105 | # pdm
106 | #   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
107 | #pdm.lock
108 | #   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
109 | #   in version control.
110 | #   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
111 | .pdm.toml
112 | .pdm-python
113 | .pdm-build/
114 | 
115 | # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
116 | __pypackages__/
117 | 
118 | # Celery stuff
119 | celerybeat-schedule
120 | celerybeat.pid
121 | 
122 | # SageMath parsed files
123 | *.sage.py
124 | 
125 | # Environments
126 | .env
127 | .venv
128 | env/
129 | venv/
130 | ENV/
131 | env.bak/
132 | venv.bak/
133 | 
134 | # Spyder project settings
135 | .spyderproject
136 | .spyproject
137 | 
138 | # Rope project settings
139 | .ropeproject
140 | 
141 | # mkdocs documentation
142 | /site
143 | 
144 | # mypy
145 | .mypy_cache/
146 | .dmypy.json
147 | dmypy.json
148 | 
149 | # Pyre type checker
150 | .pyre/
151 | 
152 | # pytype static type analyzer
153 | .pytype/
154 | 
155 | # Cython debug symbols
156 | cython_debug/
157 | 
158 | # PyCharm
159 | #  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
160 | #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
161 | #  and can be added to the global gitignore or merged into this file.  For a more nuclear
162 | #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
163 | #.idea/
164 | 


--------------------------------------------------------------------------------
/backend/src/agent/prompts.py:
--------------------------------------------------------------------------------
 1 | from datetime import datetime
 2 | 
 3 | 
 4 | # Get current date in a readable format
 5 | def get_current_date():
 6 |     return datetime.now().strftime("%B %d, %Y")
 7 | 
 8 | 
 9 | query_writer_instructions = """Your goal is to generate sophisticated and diverse web search queries. These queries are intended for an advanced automated web research tool capable of analyzing complex results, following links, and synthesizing information.
10 | 
11 | Instructions:
12 | - Always prefer a single search query, only add another query if the original question requests multiple aspects or elements and one query is not enough.
13 | - Each query should focus on one specific aspect of the original question.
14 | - Don't produce more than {number_queries} queries.
15 | - Queries should be diverse, if the topic is broad, generate more than 1 query.
16 | - Don't generate multiple similar queries, 1 is enough.
17 | - Query should ensure that the most current information is gathered. The current date is {current_date}.
18 | 
19 | Format: 
20 | - Format your response as a JSON object with ALL three of these exact keys:
21 |    - "rationale": Brief explanation of why these queries are relevant
22 |    - "query": A list of search queries
23 | 
24 | Example:
25 | 
26 | Topic: What revenue grew more last year apple stock or the number of people buying an iphone
27 | ```json
28 | {{
29 |     "rationale": "To answer this comparative growth question accurately, we need specific data points on Apple's stock performance and iPhone sales metrics. These queries target the precise financial information needed: company revenue trends, product-specific unit sales figures, and stock price movement over the same fiscal period for direct comparison.",
30 |     "query": ["Apple total revenue growth fiscal year 2024", "iPhone unit sales growth fiscal year 2024", "Apple stock price growth fiscal year 2024"],
31 | }}
32 | ```
33 | 
34 | Context: {research_topic}"""
35 | 
36 | 
37 | web_searcher_instructions = """Conduct targeted web searches to gather the most recent, credible information on "{research_topic}" and synthesize it into a verifiable text artifact.
38 | 
39 | Instructions:
40 | - Query should ensure that the most current information is gathered. The current date is {current_date}.
41 | - Conduct multiple, diverse searches to gather comprehensive information.
42 | - Consolidate key findings while meticulously tracking the source(s) for each specific piece of information.
43 | - The output should be a well-written summary or report based on your search findings. 
44 | - Only include the information found in the search results, don't make up any information.
45 | 
46 | Research Topic:
47 | {research_topic}
48 | """
49 | 
50 | reflection_instructions = """You are an expert research assistant analyzing summaries about "{research_topic}".
51 | 
52 | Instructions:
53 | - Identify knowledge gaps or areas that need deeper exploration and generate a follow-up query. (1 or multiple).
54 | - If provided summaries are sufficient to answer the user's question, don't generate a follow-up query.
55 | - If there is a knowledge gap, generate a follow-up query that would help expand your understanding.
56 | - Focus on technical details, implementation specifics, or emerging trends that weren't fully covered.
57 | 
58 | Requirements:
59 | - Ensure the follow-up query is self-contained and includes necessary context for web search.
60 | 
61 | Output Format:
62 | - Format your response as a JSON object with these exact keys:
63 |    - "is_sufficient": true or false
64 |    - "knowledge_gap": Describe what information is missing or needs clarification
65 |    - "follow_up_queries": Write a specific question to address this gap
66 | 
67 | Example:
68 | ```json
69 | {{
70 |     "is_sufficient": true, // or false
71 |     "knowledge_gap": "The summary lacks information about performance metrics and benchmarks", // "" if is_sufficient is true
72 |     "follow_up_queries": ["What are typical performance benchmarks and metrics used to evaluate [specific technology]?"] // [] if is_sufficient is true
73 | }}
74 | ```
75 | 
76 | Reflect carefully on the Summaries to identify knowledge gaps and produce a follow-up query. Then, produce your output following this JSON format:
77 | 
78 | Summaries:
79 | {summaries}
80 | """
81 | 
82 | answer_instructions = """Generate a high-quality answer to the user's question based on the provided summaries.
83 | 
84 | Instructions:
85 | - The current date is {current_date}.
86 | - You are the final step of a multi-step research process, don't mention that you are the final step. 
87 | - You have access to all the information gathered from the previous steps.
88 | - You have access to the user's question.
89 | - Generate a high-quality answer to the user's question based on the provided summaries and the user's question.
90 | - you MUST include all the citations from the summaries in the answer correctly.
91 | 
92 | User Context:
93 | - {research_topic}
94 | 
95 | Summaries:
96 | {summaries}"""
97 | 


--------------------------------------------------------------------------------
/frontend/src/global.css:
--------------------------------------------------------------------------------
  1 | @import "tailwindcss";
  2 | @import "tw-animate-css";
  3 | 
  4 | @custom-variant dark (&:is(.dark *));
  5 | 
  6 | @theme inline {
  7 |   --radius-sm: calc(var(--radius) - 4px);
  8 |   --radius-md: calc(var(--radius) - 2px);
  9 |   --radius-lg: var(--radius);
 10 |   --radius-xl: calc(var(--radius) + 4px);
 11 |   --color-background: var(--background);
 12 |   --color-foreground: var(--foreground);
 13 |   --color-card: var(--card);
 14 |   --color-card-foreground: var(--card-foreground);
 15 |   --color-popover: var(--popover);
 16 |   --color-popover-foreground: var(--popover-foreground);
 17 |   --color-primary: var(--primary);
 18 |   --color-primary-foreground: var(--primary-foreground);
 19 |   --color-secondary: var(--secondary);
 20 |   --color-secondary-foreground: var(--secondary-foreground);
 21 |   --color-muted: var(--muted);
 22 |   --color-muted-foreground: var(--muted-foreground);
 23 |   --color-accent: var(--accent);
 24 |   --color-accent-foreground: var(--accent-foreground);
 25 |   --color-destructive: var(--destructive);
 26 |   --color-border: var(--border);
 27 |   --color-input: var(--input);
 28 |   --color-ring: var(--ring);
 29 |   --color-chart-1: var(--chart-1);
 30 |   --color-chart-2: var(--chart-2);
 31 |   --color-chart-3: var(--chart-3);
 32 |   --color-chart-4: var(--chart-4);
 33 |   --color-chart-5: var(--chart-5);
 34 |   --color-sidebar: var(--sidebar);
 35 |   --color-sidebar-foreground: var(--sidebar-foreground);
 36 |   --color-sidebar-primary: var(--sidebar-primary);
 37 |   --color-sidebar-primary-foreground: var(--sidebar-primary-foreground);
 38 |   --color-sidebar-accent: var(--sidebar-accent);
 39 |   --color-sidebar-accent-foreground: var(--sidebar-accent-foreground);
 40 |   --color-sidebar-border: var(--sidebar-border);
 41 |   --color-sidebar-ring: var(--sidebar-ring);
 42 | }
 43 | 
 44 | :root {
 45 |   --radius: 0.625rem;
 46 |   --background: oklch(1 0 0);
 47 |   --foreground: oklch(0.145 0 0);
 48 |   --card: oklch(1 0 0);
 49 |   --card-foreground: oklch(0.145 0 0);
 50 |   --popover: oklch(1 0 0);
 51 |   --popover-foreground: oklch(0.145 0 0);
 52 |   --primary: oklch(0.205 0 0);
 53 |   --primary-foreground: oklch(0.985 0 0);
 54 |   --secondary: oklch(0.97 0 0);
 55 |   --secondary-foreground: oklch(0.205 0 0);
 56 |   --muted: oklch(0.97 0 0);
 57 |   --muted-foreground: oklch(0.556 0 0);
 58 |   --accent: oklch(0.97 0 0);
 59 |   --accent-foreground: oklch(0.205 0 0);
 60 |   --destructive: oklch(0.577 0.245 27.325);
 61 |   --border: oklch(0.922 0 0);
 62 |   --input: oklch(0.922 0 0);
 63 |   --ring: oklch(0.708 0 0);
 64 |   --chart-1: oklch(0.646 0.222 41.116);
 65 |   --chart-2: oklch(0.6 0.118 184.704);
 66 |   --chart-3: oklch(0.398 0.07 227.392);
 67 |   --chart-4: oklch(0.828 0.189 84.429);
 68 |   --chart-5: oklch(0.769 0.188 70.08);
 69 |   --sidebar: oklch(0.985 0 0);
 70 |   --sidebar-foreground: oklch(0.145 0 0);
 71 |   --sidebar-primary: oklch(0.205 0 0);
 72 |   --sidebar-primary-foreground: oklch(0.985 0 0);
 73 |   --sidebar-accent: oklch(0.97 0 0);
 74 |   --sidebar-accent-foreground: oklch(0.205 0 0);
 75 |   --sidebar-border: oklch(0.922 0 0);
 76 |   --sidebar-ring: oklch(0.708 0 0);
 77 | }
 78 | 
 79 | .dark {
 80 |   --background: oklch(0.145 0 0);
 81 |   --foreground: oklch(0.985 0 0);
 82 |   --card: oklch(0.205 0 0);
 83 |   --card-foreground: oklch(0.985 0 0);
 84 |   --popover: oklch(0.205 0 0);
 85 |   --popover-foreground: oklch(0.985 0 0);
 86 |   --primary: oklch(0.922 0 0);
 87 |   --primary-foreground: oklch(0.205 0 0);
 88 |   --secondary: oklch(0.269 0 0);
 89 |   --secondary-foreground: oklch(0.985 0 0);
 90 |   --muted: oklch(0.269 0 0);
 91 |   --muted-foreground: oklch(0.708 0 0);
 92 |   --accent: oklch(0.269 0 0);
 93 |   --accent-foreground: oklch(0.985 0 0);
 94 |   --destructive: oklch(0.704 0.191 22.216);
 95 |   --border: oklch(1 0 0 / 10%);
 96 |   --input: oklch(1 0 0 / 15%);
 97 |   --ring: oklch(0.556 0 0);
 98 |   --chart-1: oklch(0.488 0.243 264.376);
 99 |   --chart-2: oklch(0.696 0.17 162.48);
100 |   --chart-3: oklch(0.769 0.188 70.08);
101 |   --chart-4: oklch(0.627 0.265 303.9);
102 |   --chart-5: oklch(0.645 0.246 16.439);
103 |   --sidebar: oklch(0.205 0 0);
104 |   --sidebar-foreground: oklch(0.985 0 0);
105 |   --sidebar-primary: oklch(0.488 0.243 264.376);
106 |   --sidebar-primary-foreground: oklch(0.985 0 0);
107 |   --sidebar-accent: oklch(0.269 0 0);
108 |   --sidebar-accent-foreground: oklch(0.985 0 0);
109 |   --sidebar-border: oklch(1 0 0 / 10%);
110 |   --sidebar-ring: oklch(0.556 0 0);
111 | }
112 | 
113 | @layer base {
114 |   * {
115 |     @apply border-border outline-ring/50;
116 |   }
117 |   body {
118 |     @apply bg-background text-foreground;
119 |   }
120 | }
121 | 
122 | /* Animation Delays */
123 | .animation-delay-200 { animation-delay: 0.2s; }
124 | .animation-delay-400 { animation-delay: 0.4s; }
125 | .animation-delay-600 { animation-delay: 0.6s; }
126 | .animation-delay-800 { animation-delay: 0.8s; }
127 | 
128 | /* Keyframes */
129 | @keyframes fadeIn {
130 |   from { opacity: 0; }
131 |   to { opacity: 1; }
132 | }
133 | @keyframes fadeInUp {
134 |   from { opacity: 0; transform: translateY(20px); }
135 |   to { opacity: 1; transform: translateY(0); }
136 | }
137 | @keyframes fadeInUpSmooth {
138 |   from { opacity: 0; transform: translateY(10px); }
139 |   to { opacity: 1; transform: translateY(0); }
140 | }
141 | 
142 | /* Animation Classes */
143 | .animate-fadeIn {
144 |   animation: fadeIn 0.5s ease-out forwards;
145 | }
146 | .animate-fadeInUp {
147 |   animation: fadeInUp 0.5s ease-out forwards;
148 | }
149 | .animate-fadeInUpSmooth {
150 |   animation: fadeInUpSmooth 0.3s ease-out forwards;
151 | }
152 | 
153 | /* Ensure your body or html has a dark background if not already set, e.g.: */
154 | /* body { background-color: #0c0c0d; } */ /* This is similar to neutral-950 */
155 | 


--------------------------------------------------------------------------------
/frontend/src/components/ActivityTimeline.tsx:
--------------------------------------------------------------------------------
  1 | import {
  2 |   Card,
  3 |   CardContent,
  4 |   CardDescription,
  5 |   CardHeader,
  6 | } from "@/components/ui/card";
  7 | import { ScrollArea } from "@/components/ui/scroll-area";
  8 | import {
  9 |   Loader2,
 10 |   Activity,
 11 |   Info,
 12 |   Search,
 13 |   TextSearch,
 14 |   Brain,
 15 |   Pen,
 16 |   ChevronDown,
 17 |   ChevronUp,
 18 | } from "lucide-react";
 19 | import { useEffect, useState } from "react";
 20 | 
 21 | export interface ProcessedEvent {
 22 |   title: string;
 23 |   data: any;
 24 | }
 25 | 
 26 | interface ActivityTimelineProps {
 27 |   processedEvents: ProcessedEvent[];
 28 |   isLoading: boolean;
 29 | }
 30 | 
 31 | export function ActivityTimeline({
 32 |   processedEvents,
 33 |   isLoading,
 34 | }: ActivityTimelineProps) {
 35 |   const [isTimelineCollapsed, setIsTimelineCollapsed] =
 36 |     useState<boolean>(false);
 37 |   const getEventIcon = (title: string, index: number) => {
 38 |     if (index === 0 && isLoading && processedEvents.length === 0) {
 39 |       return <Loader2 className="h-4 w-4 text-neutral-400 animate-spin" />;
 40 |     }
 41 |     if (title.toLowerCase().includes("generating")) {
 42 |       return <TextSearch className="h-4 w-4 text-neutral-400" />;
 43 |     } else if (title.toLowerCase().includes("thinking")) {
 44 |       return <Loader2 className="h-4 w-4 text-neutral-400 animate-spin" />;
 45 |     } else if (title.toLowerCase().includes("reflection")) {
 46 |       return <Brain className="h-4 w-4 text-neutral-400" />;
 47 |     } else if (title.toLowerCase().includes("research")) {
 48 |       return <Search className="h-4 w-4 text-neutral-400" />;
 49 |     } else if (title.toLowerCase().includes("finalizing")) {
 50 |       return <Pen className="h-4 w-4 text-neutral-400" />;
 51 |     }
 52 |     return <Activity className="h-4 w-4 text-neutral-400" />;
 53 |   };
 54 | 
 55 |   useEffect(() => {
 56 |     if (!isLoading && processedEvents.length !== 0) {
 57 |       setIsTimelineCollapsed(true);
 58 |     }
 59 |   }, [isLoading, processedEvents]);
 60 | 
 61 |   return (
 62 |     <Card className="border-none rounded-lg bg-neutral-700 max-h-96">
 63 |       <CardHeader>
 64 |         <CardDescription className="flex items-center justify-between">
 65 |           <div
 66 |             className="flex items-center justify-start text-sm w-full cursor-pointer gap-2 text-neutral-100"
 67 |             onClick={() => setIsTimelineCollapsed(!isTimelineCollapsed)}
 68 |           >
 69 |             Research
 70 |             {isTimelineCollapsed ? (
 71 |               <ChevronDown className="h-4 w-4 mr-2" />
 72 |             ) : (
 73 |               <ChevronUp className="h-4 w-4 mr-2" />
 74 |             )}
 75 |           </div>
 76 |         </CardDescription>
 77 |       </CardHeader>
 78 |       {!isTimelineCollapsed && (
 79 |         <ScrollArea className="max-h-96 overflow-y-auto">
 80 |           <CardContent>
 81 |             {isLoading && processedEvents.length === 0 && (
 82 |               <div className="relative pl-8 pb-4">
 83 |                 <div className="absolute left-3 top-3.5 h-full w-0.5 bg-neutral-800" />
 84 |                 <div className="absolute left-0.5 top-2 h-5 w-5 rounded-full bg-neutral-800 flex items-center justify-center ring-4 ring-neutral-900">
 85 |                   <Loader2 className="h-3 w-3 text-neutral-400 animate-spin" />
 86 |                 </div>
 87 |                 <div>
 88 |                   <p className="text-sm text-neutral-300 font-medium">
 89 |                     Searching...
 90 |                   </p>
 91 |                 </div>
 92 |               </div>
 93 |             )}
 94 |             {processedEvents.length > 0 ? (
 95 |               <div className="space-y-0">
 96 |                 {processedEvents.map((eventItem, index) => (
 97 |                   <div key={index} className="relative pl-8 pb-4">
 98 |                     {index < processedEvents.length - 1 ||
 99 |                     (isLoading && index === processedEvents.length - 1) ? (
100 |                       <div className="absolute left-3 top-3.5 h-full w-0.5 bg-neutral-600" />
101 |                     ) : null}
102 |                     <div className="absolute left-0.5 top-2 h-6 w-6 rounded-full bg-neutral-600 flex items-center justify-center ring-4 ring-neutral-700">
103 |                       {getEventIcon(eventItem.title, index)}
104 |                     </div>
105 |                     <div>
106 |                       <p className="text-sm text-neutral-200 font-medium mb-0.5">
107 |                         {eventItem.title}
108 |                       </p>
109 |                       <p className="text-xs text-neutral-300 leading-relaxed">
110 |                         {typeof eventItem.data === "string"
111 |                           ? eventItem.data
112 |                           : Array.isArray(eventItem.data)
113 |                           ? (eventItem.data as string[]).join(", ")
114 |                           : JSON.stringify(eventItem.data)}
115 |                       </p>
116 |                     </div>
117 |                   </div>
118 |                 ))}
119 |                 {isLoading && processedEvents.length > 0 && (
120 |                   <div className="relative pl-8 pb-4">
121 |                     <div className="absolute left-0.5 top-2 h-5 w-5 rounded-full bg-neutral-600 flex items-center justify-center ring-4 ring-neutral-700">
122 |                       <Loader2 className="h-3 w-3 text-neutral-400 animate-spin" />
123 |                     </div>
124 |                     <div>
125 |                       <p className="text-sm text-neutral-300 font-medium">
126 |                         Searching...
127 |                       </p>
128 |                     </div>
129 |                   </div>
130 |                 )}
131 |               </div>
132 |             ) : !isLoading ? ( // Only show "No activity" if not loading and no events
133 |               <div className="flex flex-col items-center justify-center h-full text-neutral-500 pt-10">
134 |                 <Info className="h-6 w-6 mb-3" />
135 |                 <p className="text-sm">No activity to display.</p>
136 |                 <p className="text-xs text-neutral-600 mt-1">
137 |                   Timeline will update during processing.
138 |                 </p>
139 |               </div>
140 |             ) : null}
141 |           </CardContent>
142 |         </ScrollArea>
143 |       )}
144 |     </Card>
145 |   );
146 | }
147 | 


--------------------------------------------------------------------------------
/frontend/src/App.tsx:
--------------------------------------------------------------------------------
  1 | import { useStream } from "@langchain/langgraph-sdk/react";
  2 | import type { Message } from "@langchain/langgraph-sdk";
  3 | import { useState, useEffect, useRef, useCallback } from "react";
  4 | import { ProcessedEvent } from "@/components/ActivityTimeline";
  5 | import { WelcomeScreen } from "@/components/WelcomeScreen";
  6 | import { ChatMessagesView } from "@/components/ChatMessagesView";
  7 | 
  8 | export default function App() {
  9 |   const [processedEventsTimeline, setProcessedEventsTimeline] = useState<
 10 |     ProcessedEvent[]
 11 |   >([]);
 12 |   const [historicalActivities, setHistoricalActivities] = useState<
 13 |     Record<string, ProcessedEvent[]>
 14 |   >({});
 15 |   const scrollAreaRef = useRef<HTMLDivElement>(null);
 16 |   const hasFinalizeEventOccurredRef = useRef(false);
 17 | 
 18 |   const thread = useStream<{
 19 |     messages: Message[];
 20 |     initial_search_query_count: number;
 21 |     max_research_loops: number;
 22 |     reasoning_model: string;
 23 |   }>({
 24 |     apiUrl: import.meta.env.DEV
 25 |       ? "http://localhost:2024"
 26 |       : "http://localhost:8123",
 27 |     assistantId: "agent",
 28 |     messagesKey: "messages",
 29 |     onFinish: (event: any) => {
 30 |       console.log(event);
 31 |     },
 32 |     onUpdateEvent: (event: any) => {
 33 |       let processedEvent: ProcessedEvent | null = null;
 34 |       if (event.generate_query) {
 35 |         processedEvent = {
 36 |           title: "Generating Search Queries",
 37 |           data: event.generate_query.query_list.join(", "),
 38 |         };
 39 |       } else if (event.web_research) {
 40 |         const sources = event.web_research.sources_gathered || [];
 41 |         const numSources = sources.length;
 42 |         const uniqueLabels = [
 43 |           ...new Set(sources.map((s: any) => s.label).filter(Boolean)),
 44 |         ];
 45 |         const exampleLabels = uniqueLabels.slice(0, 3).join(", ");
 46 |         processedEvent = {
 47 |           title: "Web Research",
 48 |           data: `Gathered ${numSources} sources. Related to: ${
 49 |             exampleLabels || "N/A"
 50 |           }.`,
 51 |         };
 52 |       } else if (event.reflection) {
 53 |         processedEvent = {
 54 |           title: "Reflection",
 55 |           data: event.reflection.is_sufficient
 56 |             ? "Search successful, generating final answer."
 57 |             : `Need more information, searching for ${event.reflection.follow_up_queries.join(
 58 |                 ", "
 59 |               )}`,
 60 |         };
 61 |       } else if (event.finalize_answer) {
 62 |         processedEvent = {
 63 |           title: "Finalizing Answer",
 64 |           data: "Composing and presenting the final answer.",
 65 |         };
 66 |         hasFinalizeEventOccurredRef.current = true;
 67 |       }
 68 |       if (processedEvent) {
 69 |         setProcessedEventsTimeline((prevEvents) => [
 70 |           ...prevEvents,
 71 |           processedEvent!,
 72 |         ]);
 73 |       }
 74 |     },
 75 |   });
 76 | 
 77 |   useEffect(() => {
 78 |     if (scrollAreaRef.current) {
 79 |       const scrollViewport = scrollAreaRef.current.querySelector(
 80 |         "[data-radix-scroll-area-viewport]"
 81 |       );
 82 |       if (scrollViewport) {
 83 |         scrollViewport.scrollTop = scrollViewport.scrollHeight;
 84 |       }
 85 |     }
 86 |   }, [thread.messages]);
 87 | 
 88 |   useEffect(() => {
 89 |     if (
 90 |       hasFinalizeEventOccurredRef.current &&
 91 |       !thread.isLoading &&
 92 |       thread.messages.length > 0
 93 |     ) {
 94 |       const lastMessage = thread.messages[thread.messages.length - 1];
 95 |       if (lastMessage && lastMessage.type === "ai" && lastMessage.id) {
 96 |         setHistoricalActivities((prev) => ({
 97 |           ...prev,
 98 |           [lastMessage.id!]: [...processedEventsTimeline],
 99 |         }));
100 |       }
101 |       hasFinalizeEventOccurredRef.current = false;
102 |     }
103 |   }, [thread.messages, thread.isLoading, processedEventsTimeline]);
104 | 
105 |   const handleSubmit = useCallback(
106 |     (submittedInputValue: string, effort: string, model: string) => {
107 |       if (!submittedInputValue.trim()) return;
108 |       setProcessedEventsTimeline([]);
109 |       hasFinalizeEventOccurredRef.current = false;
110 | 
111 |       // convert effort to, initial_search_query_count and max_research_loops
112 |       // low means max 1 loop and 1 query
113 |       // medium means max 3 loops and 3 queries
114 |       // high means max 10 loops and 5 queries
115 |       let initial_search_query_count = 0;
116 |       let max_research_loops = 0;
117 |       switch (effort) {
118 |         case "low":
119 |           initial_search_query_count = 1;
120 |           max_research_loops = 1;
121 |           break;
122 |         case "medium":
123 |           initial_search_query_count = 3;
124 |           max_research_loops = 3;
125 |           break;
126 |         case "high":
127 |           initial_search_query_count = 5;
128 |           max_research_loops = 10;
129 |           break;
130 |       }
131 | 
132 |       const newMessages: Message[] = [
133 |         ...(thread.messages || []),
134 |         {
135 |           type: "human",
136 |           content: submittedInputValue,
137 |           id: Date.now().toString(),
138 |         },
139 |       ];
140 |       thread.submit({
141 |         messages: newMessages,
142 |         initial_search_query_count: initial_search_query_count,
143 |         max_research_loops: max_research_loops,
144 |         reasoning_model: model,
145 |       });
146 |     },
147 |     [thread]
148 |   );
149 | 
150 |   const handleCancel = useCallback(() => {
151 |     thread.stop();
152 |     window.location.reload();
153 |   }, [thread]);
154 | 
155 |   return (
156 |     <div className="flex h-screen bg-neutral-800 text-neutral-100 font-sans antialiased">
157 |       <main className="flex-1 flex flex-col overflow-hidden max-w-4xl mx-auto w-full">
158 |         <div
159 |           className={`flex-1 overflow-y-auto ${
160 |             thread.messages.length === 0 ? "flex" : ""
161 |           }`}
162 |         >
163 |           {thread.messages.length === 0 ? (
164 |             <WelcomeScreen
165 |               handleSubmit={handleSubmit}
166 |               isLoading={thread.isLoading}
167 |               onCancel={handleCancel}
168 |             />
169 |           ) : (
170 |             <ChatMessagesView
171 |               messages={thread.messages}
172 |               isLoading={thread.isLoading}
173 |               scrollAreaRef={scrollAreaRef}
174 |               onSubmit={handleSubmit}
175 |               onCancel={handleCancel}
176 |               liveActivityEvents={processedEventsTimeline}
177 |               historicalActivities={historicalActivities}
178 |             />
179 |           )}
180 |         </div>
181 |       </main>
182 |     </div>
183 |   );
184 | }
185 | 


--------------------------------------------------------------------------------
/frontend/src/components/InputForm.tsx:
--------------------------------------------------------------------------------
  1 | import { useState } from "react";
  2 | import { Button } from "@/components/ui/button";
  3 | import { SquarePen, Brain, Send, StopCircle, Zap, Cpu } from "lucide-react";
  4 | import { Textarea } from "@/components/ui/textarea";
  5 | import {
  6 |   Select,
  7 |   SelectContent,
  8 |   SelectItem,
  9 |   SelectTrigger,
 10 |   SelectValue,
 11 | } from "@/components/ui/select";
 12 | 
 13 | // Updated InputFormProps
 14 | interface InputFormProps {
 15 |   onSubmit: (inputValue: string, effort: string, model: string) => void;
 16 |   onCancel: () => void;
 17 |   isLoading: boolean;
 18 |   hasHistory: boolean;
 19 | }
 20 | 
 21 | export const InputForm: React.FC<InputFormProps> = ({
 22 |   onSubmit,
 23 |   onCancel,
 24 |   isLoading,
 25 |   hasHistory,
 26 | }) => {
 27 |   const [internalInputValue, setInternalInputValue] = useState("");
 28 |   const [effort, setEffort] = useState("medium");
 29 |   const [model, setModel] = useState("qwen2.5-coder-7b-instruct");
 30 | 
 31 |   const handleInternalSubmit = (e?: React.FormEvent) => {
 32 |     if (e) e.preventDefault();
 33 |     if (!internalInputValue.trim()) return;
 34 |     onSubmit(internalInputValue, effort, model);
 35 |     setInternalInputValue("");
 36 |   };
 37 | 
 38 |   const handleInternalKeyDown = (
 39 |     e: React.KeyboardEvent<HTMLTextAreaElement>
 40 |   ) => {
 41 |     if (e.key === "Enter" && !e.shiftKey) {
 42 |       e.preventDefault();
 43 |       handleInternalSubmit();
 44 |     }
 45 |   };
 46 | 
 47 |   const isSubmitDisabled = !internalInputValue.trim() || isLoading;
 48 | 
 49 |   return (
 50 |     <form
 51 |       onSubmit={handleInternalSubmit}
 52 |       className={`flex flex-col gap-2 p-3 `}
 53 |     >
 54 |       <div
 55 |         className={`flex flex-row items-center justify-between text-white rounded-3xl rounded-bl-sm ${
 56 |           hasHistory ? "rounded-br-sm" : ""
 57 |         } break-words min-h-7 bg-neutral-700 px-4 pt-3 `}
 58 |       >
 59 |         <Textarea
 60 |           value={internalInputValue}
 61 |           onChange={(e) => setInternalInputValue(e.target.value)}
 62 |           onKeyDown={handleInternalKeyDown}
 63 |           placeholder="Who won the Euro 2024 and scored the most goals?"
 64 |           className={`w-full text-neutral-100 placeholder-neutral-500 resize-none border-0 focus:outline-none focus:ring-0 outline-none focus-visible:ring-0 shadow-none 
 65 |                         md:text-base  min-h-[56px] max-h-[200px]`}
 66 |           rows={1}
 67 |         />
 68 |         <div className="-mt-3">
 69 |           {isLoading ? (
 70 |             <Button
 71 |               type="button"
 72 |               variant="ghost"
 73 |               size="icon"
 74 |               className="text-red-500 hover:text-red-400 hover:bg-red-500/10 p-2 cursor-pointer rounded-full transition-all duration-200"
 75 |               onClick={onCancel}
 76 |             >
 77 |               <StopCircle className="h-5 w-5" />
 78 |             </Button>
 79 |           ) : (
 80 |             <Button
 81 |               type="submit"
 82 |               variant="ghost"
 83 |               className={`${
 84 |                 isSubmitDisabled
 85 |                   ? "text-neutral-500"
 86 |                   : "text-blue-500 hover:text-blue-400 hover:bg-blue-500/10"
 87 |               } p-2 cursor-pointer rounded-full transition-all duration-200 text-base`}
 88 |               disabled={isSubmitDisabled}
 89 |             >
 90 |               Search
 91 |               <Send className="h-5 w-5" />
 92 |             </Button>
 93 |           )}
 94 |         </div>
 95 |       </div>
 96 |       <div className="flex items-center justify-between">
 97 |         <div className="flex flex-row gap-2">
 98 |           <div className="flex flex-row gap-2 bg-neutral-700 border-neutral-600 text-neutral-300 focus:ring-neutral-500 rounded-xl rounded-t-sm pl-2  max-w-[100%] sm:max-w-[90%]">
 99 |             <div className="flex flex-row items-center text-sm">
100 |               <Brain className="h-4 w-4 mr-2" />
101 |               Effort
102 |             </div>
103 |             <Select value={effort} onValueChange={setEffort}>
104 |               <SelectTrigger className="w-[120px] bg-transparent border-none cursor-pointer">
105 |                 <SelectValue placeholder="Effort" />
106 |               </SelectTrigger>
107 |               <SelectContent className="bg-neutral-700 border-neutral-600 text-neutral-300 cursor-pointer">
108 |                 <SelectItem
109 |                   value="low"
110 |                   className="hover:bg-neutral-600 focus:bg-neutral-600 cursor-pointer"
111 |                 >
112 |                   Low
113 |                 </SelectItem>
114 |                 <SelectItem
115 |                   value="medium"
116 |                   className="hover:bg-neutral-600 focus:bg-neutral-600 cursor-pointer"
117 |                 >
118 |                   Medium
119 |                 </SelectItem>
120 |                 <SelectItem
121 |                   value="high"
122 |                   className="hover:bg-neutral-600 focus:bg-neutral-600 cursor-pointer"
123 |                 >
124 |                   High
125 |                 </SelectItem>
126 |               </SelectContent>
127 |             </Select>
128 |           </div>
129 |           <div className="flex flex-row gap-2 bg-neutral-700 border-neutral-600 text-neutral-300 focus:ring-neutral-500 rounded-xl rounded-t-sm pl-2  max-w-[100%] sm:max-w-[90%]">
130 |             <div className="flex flex-row items-center text-sm ml-2">
131 |               <Cpu className="h-4 w-4 mr-2" />
132 |               Model
133 |             </div>
134 |             <Select value={model} onValueChange={setModel}>
135 |               <SelectTrigger className="w-[150px] bg-transparent border-none cursor-pointer">
136 |                 <SelectValue placeholder="Model" />
137 |               </SelectTrigger>
138 |               <SelectContent className="bg-neutral-700 border-neutral-600 text-neutral-300 cursor-pointer">
139 |                 <SelectItem
140 |                   value="qwen2.5-coder-7b-instruct"
141 |                   className="hover:bg-neutral-600 focus:bg-neutral-600 cursor-pointer"
142 |                 >
143 |                   <div className="flex items-center">
144 |                     <Cpu className="h-4 w-4 mr-2 text-purple-400" /> qwen2.5-coder-7b-instruct
145 |                   </div>
146 |                 </SelectItem>
147 |               </SelectContent>
148 |             </Select>
149 |           </div>
150 |         </div>
151 |         {hasHistory && (
152 |           <Button
153 |             className="bg-neutral-700 border-neutral-600 text-neutral-300 cursor-pointer rounded-xl rounded-t-sm pl-2 "
154 |             variant="default"
155 |             onClick={() => window.location.reload()}
156 |           >
157 |             <SquarePen size={16} />
158 |             New Search
159 |           </Button>
160 |         )}
161 |       </div>
162 |     </form>
163 |   );
164 | };
165 | 


--------------------------------------------------------------------------------
/frontend/src/components/ui/select.tsx:
--------------------------------------------------------------------------------
  1 | import * as React from "react"
  2 | import * as SelectPrimitive from "@radix-ui/react-select"
  3 | import { CheckIcon, ChevronDownIcon, ChevronUpIcon } from "lucide-react"
  4 | 
  5 | import { cn } from "@/lib/utils"
  6 | 
  7 | function Select({
  8 |   ...props
  9 | }: React.ComponentProps<typeof SelectPrimitive.Root>) {
 10 |   return <SelectPrimitive.Root data-slot="select" {...props} />
 11 | }
 12 | 
 13 | function SelectGroup({
 14 |   ...props
 15 | }: React.ComponentProps<typeof SelectPrimitive.Group>) {
 16 |   return <SelectPrimitive.Group data-slot="select-group" {...props} />
 17 | }
 18 | 
 19 | function SelectValue({
 20 |   ...props
 21 | }: React.ComponentProps<typeof SelectPrimitive.Value>) {
 22 |   return <SelectPrimitive.Value data-slot="select-value" {...props} />
 23 | }
 24 | 
 25 | function SelectTrigger({
 26 |   className,
 27 |   size = "default",
 28 |   children,
 29 |   ...props
 30 | }: React.ComponentProps<typeof SelectPrimitive.Trigger> & {
 31 |   size?: "sm" | "default"
 32 | }) {
 33 |   return (
 34 |     <SelectPrimitive.Trigger
 35 |       data-slot="select-trigger"
 36 |       data-size={size}
 37 |       className={cn(
 38 |         "border-input data-[placeholder]:text-muted-foreground [&_svg:not([class*='text-'])]:text-muted-foreground focus-visible:border-ring focus-visible:ring-ring/50 aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive dark:bg-input/30 dark:hover:bg-input/50 flex w-fit items-center justify-between gap-2 rounded-md border bg-transparent px-3 py-2 text-sm whitespace-nowrap shadow-xs transition-[color,box-shadow] outline-none focus-visible:ring-[3px] disabled:cursor-not-allowed disabled:opacity-50 data-[size=default]:h-9 data-[size=sm]:h-8 *:data-[slot=select-value]:line-clamp-1 *:data-[slot=select-value]:flex *:data-[slot=select-value]:items-center *:data-[slot=select-value]:gap-2 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
 39 |         className
 40 |       )}
 41 |       {...props}
 42 |     >
 43 |       {children}
 44 |       <SelectPrimitive.Icon asChild>
 45 |         <ChevronDownIcon className="size-4 opacity-50" />
 46 |       </SelectPrimitive.Icon>
 47 |     </SelectPrimitive.Trigger>
 48 |   )
 49 | }
 50 | 
 51 | function SelectContent({
 52 |   className,
 53 |   children,
 54 |   position = "popper",
 55 |   ...props
 56 | }: React.ComponentProps<typeof SelectPrimitive.Content>) {
 57 |   return (
 58 |     <SelectPrimitive.Portal>
 59 |       <SelectPrimitive.Content
 60 |         data-slot="select-content"
 61 |         className={cn(
 62 |           "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 relative z-50 max-h-(--radix-select-content-available-height) min-w-[8rem] origin-(--radix-select-content-transform-origin) overflow-x-hidden overflow-y-auto rounded-md border shadow-md",
 63 |           position === "popper" &&
 64 |             "data-[side=bottom]:translate-y-1 data-[side=left]:-translate-x-1 data-[side=right]:translate-x-1 data-[side=top]:-translate-y-1",
 65 |           className
 66 |         )}
 67 |         position={position}
 68 |         {...props}
 69 |       >
 70 |         <SelectScrollUpButton />
 71 |         <SelectPrimitive.Viewport
 72 |           className={cn(
 73 |             "p-1",
 74 |             position === "popper" &&
 75 |               "h-[var(--radix-select-trigger-height)] w-full min-w-[var(--radix-select-trigger-width)] scroll-my-1"
 76 |           )}
 77 |         >
 78 |           {children}
 79 |         </SelectPrimitive.Viewport>
 80 |         <SelectScrollDownButton />
 81 |       </SelectPrimitive.Content>
 82 |     </SelectPrimitive.Portal>
 83 |   )
 84 | }
 85 | 
 86 | function SelectLabel({
 87 |   className,
 88 |   ...props
 89 | }: React.ComponentProps<typeof SelectPrimitive.Label>) {
 90 |   return (
 91 |     <SelectPrimitive.Label
 92 |       data-slot="select-label"
 93 |       className={cn("text-muted-foreground px-2 py-1.5 text-xs", className)}
 94 |       {...props}
 95 |     />
 96 |   )
 97 | }
 98 | 
 99 | function SelectItem({
100 |   className,
101 |   children,
102 |   ...props
103 | }: React.ComponentProps<typeof SelectPrimitive.Item>) {
104 |   return (
105 |     <SelectPrimitive.Item
106 |       data-slot="select-item"
107 |       className={cn(
108 |         "focus:bg-accent focus:text-accent-foreground [&_svg:not([class*='text-'])]:text-muted-foreground relative flex w-full cursor-default items-center gap-2 rounded-sm py-1.5 pr-8 pl-2 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 *:[span]:last:flex *:[span]:last:items-center *:[span]:last:gap-2",
109 |         className
110 |       )}
111 |       {...props}
112 |     >
113 |       <span className="absolute right-2 flex size-3.5 items-center justify-center">
114 |         <SelectPrimitive.ItemIndicator>
115 |           <CheckIcon className="size-4" />
116 |         </SelectPrimitive.ItemIndicator>
117 |       </span>
118 |       <SelectPrimitive.ItemText>{children}</SelectPrimitive.ItemText>
119 |     </SelectPrimitive.Item>
120 |   )
121 | }
122 | 
123 | function SelectSeparator({
124 |   className,
125 |   ...props
126 | }: React.ComponentProps<typeof SelectPrimitive.Separator>) {
127 |   return (
128 |     <SelectPrimitive.Separator
129 |       data-slot="select-separator"
130 |       className={cn("bg-border pointer-events-none -mx-1 my-1 h-px", className)}
131 |       {...props}
132 |     />
133 |   )
134 | }
135 | 
136 | function SelectScrollUpButton({
137 |   className,
138 |   ...props
139 | }: React.ComponentProps<typeof SelectPrimitive.ScrollUpButton>) {
140 |   return (
141 |     <SelectPrimitive.ScrollUpButton
142 |       data-slot="select-scroll-up-button"
143 |       className={cn(
144 |         "flex cursor-default items-center justify-center py-1",
145 |         className
146 |       )}
147 |       {...props}
148 |     >
149 |       <ChevronUpIcon className="size-4" />
150 |     </SelectPrimitive.ScrollUpButton>
151 |   )
152 | }
153 | 
154 | function SelectScrollDownButton({
155 |   className,
156 |   ...props
157 | }: React.ComponentProps<typeof SelectPrimitive.ScrollDownButton>) {
158 |   return (
159 |     <SelectPrimitive.ScrollDownButton
160 |       data-slot="select-scroll-down-button"
161 |       className={cn(
162 |         "flex cursor-default items-center justify-center py-1",
163 |         className
164 |       )}
165 |       {...props}
166 |     >
167 |       <ChevronDownIcon className="size-4" />
168 |     </SelectPrimitive.ScrollDownButton>
169 |   )
170 | }
171 | 
172 | export {
173 |   Select,
174 |   SelectContent,
175 |   SelectGroup,
176 |   SelectItem,
177 |   SelectLabel,
178 |   SelectScrollDownButton,
179 |   SelectScrollUpButton,
180 |   SelectSeparator,
181 |   SelectTrigger,
182 |   SelectValue,
183 | }
184 | 


--------------------------------------------------------------------------------
/backend/src/agent/utils.py:
--------------------------------------------------------------------------------
  1 | from typing import Any, Dict, List
  2 | from langchain_core.messages import AnyMessage, AIMessage, HumanMessage
  3 | 
  4 | 
  5 | def get_research_topic(messages: List[AnyMessage]) -> str:
  6 |     """
  7 |     Get the research topic from the messages.
  8 |     """
  9 |     # check if request has a history and combine the messages into a single string
 10 |     if len(messages) == 1:
 11 |         research_topic = messages[-1].content
 12 |     else:
 13 |         research_topic = ""
 14 |         for message in messages:
 15 |             if isinstance(message, HumanMessage):
 16 |                 research_topic += f"User: {message.content}\n"
 17 |             elif isinstance(message, AIMessage):
 18 |                 research_topic += f"Assistant: {message.content}\n"
 19 |     return research_topic
 20 | 
 21 | 
 22 | def resolve_urls(urls_to_resolve: List[Any], id: int) -> Dict[str, str]:
 23 |     """
 24 |     Create a map of the vertex ai search urls (very long) to a short url with a unique id for each url.
 25 |     Ensures each original URL gets a consistent shortened form while maintaining uniqueness.
 26 |     """
 27 |     prefix = f"https://vertexaisearch.cloud.google.com/id/"
 28 |     urls = [site.web.uri for site in urls_to_resolve]
 29 | 
 30 |     # Create a dictionary that maps each unique URL to its first occurrence index
 31 |     resolved_map = {}
 32 |     for idx, url in enumerate(urls):
 33 |         if url not in resolved_map:
 34 |             resolved_map[url] = f"{prefix}{id}-{idx}"
 35 | 
 36 |     return resolved_map
 37 | 
 38 | 
 39 | def insert_citation_markers(text, citations_list):
 40 |     """
 41 |     Inserts citation markers into a text string based on start and end indices.
 42 | 
 43 |     Args:
 44 |         text (str): The original text string.
 45 |         citations_list (list): A list of dictionaries, where each dictionary
 46 |                                contains 'start_index', 'end_index', and
 47 |                                'segment_string' (the marker to insert).
 48 |                                Indices are assumed to be for the original text.
 49 | 
 50 |     Returns:
 51 |         str: The text with citation markers inserted.
 52 |     """
 53 |     # Sort citations by end_index in descending order.
 54 |     # If end_index is the same, secondary sort by start_index descending.
 55 |     # This ensures that insertions at the end of the string don't affect
 56 |     # the indices of earlier parts of the string that still need to be processed.
 57 |     sorted_citations = sorted(
 58 |         citations_list, key=lambda c: (c["end_index"], c["start_index"]), reverse=True
 59 |     )
 60 | 
 61 |     modified_text = text
 62 |     for citation_info in sorted_citations:
 63 |         # These indices refer to positions in the *original* text,
 64 |         # but since we iterate from the end, they remain valid for insertion
 65 |         # relative to the parts of the string already processed.
 66 |         end_idx = citation_info["end_index"]
 67 |         marker_to_insert = ""
 68 |         for segment in citation_info["segments"]:
 69 |             marker_to_insert += f" [{segment['label']}]({segment['short_url']})"
 70 |         # Insert the citation marker at the original end_idx position
 71 |         modified_text = (
 72 |             modified_text[:end_idx] + marker_to_insert + modified_text[end_idx:]
 73 |         )
 74 | 
 75 |     return modified_text
 76 | 
 77 | 
 78 | def get_citations(response, resolved_urls_map):
 79 |     """
 80 |     Extracts and formats citation information from a Gemini model's response.
 81 | 
 82 |     This function processes the grounding metadata provided in the response to
 83 |     construct a list of citation objects. Each citation object includes the
 84 |     start and end indices of the text segment it refers to, and a string
 85 |     containing formatted markdown links to the supporting web chunks.
 86 | 
 87 |     Args:
 88 |         response: The response object from the Gemini model, expected to have
 89 |                   a structure including `candidates[0].grounding_metadata`.
 90 |                   It also relies on a `resolved_map` being available in its
 91 |                   scope to map chunk URIs to resolved URLs.
 92 | 
 93 |     Returns:
 94 |         list: A list of dictionaries, where each dictionary represents a citation
 95 |               and has the following keys:
 96 |               - "start_index" (int): The starting character index of the cited
 97 |                                      segment in the original text. Defaults to 0
 98 |                                      if not specified.
 99 |               - "end_index" (int): The character index immediately after the
100 |                                    end of the cited segment (exclusive).
101 |               - "segments" (list[str]): A list of individual markdown-formatted
102 |                                         links for each grounding chunk.
103 |               - "segment_string" (str): A concatenated string of all markdown-
104 |                                         formatted links for the citation.
105 |               Returns an empty list if no valid candidates or grounding supports
106 |               are found, or if essential data is missing.
107 |     """
108 |     citations = []
109 | 
110 |     # Ensure response and necessary nested structures are present
111 |     if not response or not response.candidates:
112 |         return citations
113 | 
114 |     candidate = response.candidates[0]
115 |     if (
116 |         not hasattr(candidate, "grounding_metadata")
117 |         or not candidate.grounding_metadata
118 |         or not hasattr(candidate.grounding_metadata, "grounding_supports")
119 |     ):
120 |         return citations
121 | 
122 |     for support in candidate.grounding_metadata.grounding_supports:
123 |         citation = {}
124 | 
125 |         # Ensure segment information is present
126 |         if not hasattr(support, "segment") or support.segment is None:
127 |             continue  # Skip this support if segment info is missing
128 | 
129 |         start_index = (
130 |             support.segment.start_index
131 |             if support.segment.start_index is not None
132 |             else 0
133 |         )
134 | 
135 |         # Ensure end_index is present to form a valid segment
136 |         if support.segment.end_index is None:
137 |             continue  # Skip if end_index is missing, as it's crucial
138 | 
139 |         # Add 1 to end_index to make it an exclusive end for slicing/range purposes
140 |         # (assuming the API provides an inclusive end_index)
141 |         citation["start_index"] = start_index
142 |         citation["end_index"] = support.segment.end_index
143 | 
144 |         citation["segments"] = []
145 |         if (
146 |             hasattr(support, "grounding_chunk_indices")
147 |             and support.grounding_chunk_indices
148 |         ):
149 |             for ind in support.grounding_chunk_indices:
150 |                 try:
151 |                     chunk = candidate.grounding_metadata.grounding_chunks[ind]
152 |                     resolved_url = resolved_urls_map.get(chunk.web.uri, None)
153 |                     label = chunk.web.title.split(".")[0] if chunk.web.title else ""
154 |                     citation["segments"].append(
155 |                         {
156 |                             "label": label,
157 |                             "short_url": resolved_url,
158 |                             "value": chunk.web.uri,
159 |                         }
160 |                     )
161 |                 except (IndexError, AttributeError, NameError):
162 |                     # Handle cases where chunk, web, uri, or resolved_map might be problematic
163 |                     # For simplicity, we'll just skip adding this particular segment link
164 |                     # In a production system, you might want to log this.
165 |                     pass
166 |         citations.append(citation)
167 |     return citations
168 | 


--------------------------------------------------------------------------------
/frontend/src/components/ChatMessagesView.tsx:
--------------------------------------------------------------------------------
  1 | import type React from "react";
  2 | import type { Message } from "@langchain/langgraph-sdk";
  3 | import { ScrollArea } from "@/components/ui/scroll-area";
  4 | import { Loader2, Copy, CopyCheck } from "lucide-react";
  5 | import { InputForm } from "@/components/InputForm";
  6 | import { Button } from "@/components/ui/button";
  7 | import { useState, ReactNode } from "react";
  8 | import ReactMarkdown from "react-markdown";
  9 | import { cn } from "@/lib/utils";
 10 | import { Badge } from "@/components/ui/badge";
 11 | import {
 12 |   ActivityTimeline,
 13 |   ProcessedEvent,
 14 | } from "@/components/ActivityTimeline"; // Assuming ActivityTimeline is in the same dir or adjust path
 15 | 
 16 | // Markdown component props type from former ReportView
 17 | type MdComponentProps = {
 18 |   className?: string;
 19 |   children?: ReactNode;
 20 |   [key: string]: any;
 21 | };
 22 | 
 23 | // Markdown components (from former ReportView.tsx)
 24 | const mdComponents = {
 25 |   h1: ({ className, children, ...props }: MdComponentProps) => (
 26 |     <h1 className={cn("text-2xl font-bold mt-4 mb-2", className)} {...props}>
 27 |       {children}
 28 |     </h1>
 29 |   ),
 30 |   h2: ({ className, children, ...props }: MdComponentProps) => (
 31 |     <h2 className={cn("text-xl font-bold mt-3 mb-2", className)} {...props}>
 32 |       {children}
 33 |     </h2>
 34 |   ),
 35 |   h3: ({ className, children, ...props }: MdComponentProps) => (
 36 |     <h3 className={cn("text-lg font-bold mt-3 mb-1", className)} {...props}>
 37 |       {children}
 38 |     </h3>
 39 |   ),
 40 |   p: ({ className, children, ...props }: MdComponentProps) => (
 41 |     <p className={cn("mb-3 leading-7", className)} {...props}>
 42 |       {children}
 43 |     </p>
 44 |   ),
 45 |   a: ({ className, children, href, ...props }: MdComponentProps) => (
 46 |     <Badge className="text-xs mx-0.5">
 47 |       <a
 48 |         className={cn("text-blue-400 hover:text-blue-300 text-xs", className)}
 49 |         href={href}
 50 |         target="_blank"
 51 |         rel="noopener noreferrer"
 52 |         {...props}
 53 |       >
 54 |         {children}
 55 |       </a>
 56 |     </Badge>
 57 |   ),
 58 |   ul: ({ className, children, ...props }: MdComponentProps) => (
 59 |     <ul className={cn("list-disc pl-6 mb-3", className)} {...props}>
 60 |       {children}
 61 |     </ul>
 62 |   ),
 63 |   ol: ({ className, children, ...props }: MdComponentProps) => (
 64 |     <ol className={cn("list-decimal pl-6 mb-3", className)} {...props}>
 65 |       {children}
 66 |     </ol>
 67 |   ),
 68 |   li: ({ className, children, ...props }: MdComponentProps) => (
 69 |     <li className={cn("mb-1", className)} {...props}>
 70 |       {children}
 71 |     </li>
 72 |   ),
 73 |   blockquote: ({ className, children, ...props }: MdComponentProps) => (
 74 |     <blockquote
 75 |       className={cn(
 76 |         "border-l-4 border-neutral-600 pl-4 italic my-3 text-sm",
 77 |         className
 78 |       )}
 79 |       {...props}
 80 |     >
 81 |       {children}
 82 |     </blockquote>
 83 |   ),
 84 |   code: ({ className, children, ...props }: MdComponentProps) => (
 85 |     <code
 86 |       className={cn(
 87 |         "bg-neutral-900 rounded px-1 py-0.5 font-mono text-xs",
 88 |         className
 89 |       )}
 90 |       {...props}
 91 |     >
 92 |       {children}
 93 |     </code>
 94 |   ),
 95 |   pre: ({ className, children, ...props }: MdComponentProps) => (
 96 |     <pre
 97 |       className={cn(
 98 |         "bg-neutral-900 p-3 rounded-lg overflow-x-auto font-mono text-xs my-3",
 99 |         className
100 |       )}
101 |       {...props}
102 |     >
103 |       {children}
104 |     </pre>
105 |   ),
106 |   hr: ({ className, ...props }: MdComponentProps) => (
107 |     <hr className={cn("border-neutral-600 my-4", className)} {...props} />
108 |   ),
109 |   table: ({ className, children, ...props }: MdComponentProps) => (
110 |     <div className="my-3 overflow-x-auto">
111 |       <table className={cn("border-collapse w-full", className)} {...props}>
112 |         {children}
113 |       </table>
114 |     </div>
115 |   ),
116 |   th: ({ className, children, ...props }: MdComponentProps) => (
117 |     <th
118 |       className={cn(
119 |         "border border-neutral-600 px-3 py-2 text-left font-bold",
120 |         className
121 |       )}
122 |       {...props}
123 |     >
124 |       {children}
125 |     </th>
126 |   ),
127 |   td: ({ className, children, ...props }: MdComponentProps) => (
128 |     <td
129 |       className={cn("border border-neutral-600 px-3 py-2", className)}
130 |       {...props}
131 |     >
132 |       {children}
133 |     </td>
134 |   ),
135 | };
136 | 
137 | // Props for HumanMessageBubble
138 | interface HumanMessageBubbleProps {
139 |   message: Message;
140 |   mdComponents: typeof mdComponents;
141 | }
142 | 
143 | // HumanMessageBubble Component
144 | const HumanMessageBubble: React.FC<HumanMessageBubbleProps> = ({
145 |   message,
146 |   mdComponents,
147 | }) => {
148 |   return (
149 |     <div
150 |       className={`text-white rounded-3xl break-words min-h-7 bg-neutral-700 max-w-[100%] sm:max-w-[90%] px-4 pt-3 rounded-br-lg`}
151 |     >
152 |       <ReactMarkdown components={mdComponents}>
153 |         {typeof message.content === "string"
154 |           ? message.content
155 |           : JSON.stringify(message.content)}
156 |       </ReactMarkdown>
157 |     </div>
158 |   );
159 | };
160 | 
161 | // Props for AiMessageBubble
162 | interface AiMessageBubbleProps {
163 |   message: Message;
164 |   historicalActivity: ProcessedEvent[] | undefined;
165 |   liveActivity: ProcessedEvent[] | undefined;
166 |   isLastMessage: boolean;
167 |   isOverallLoading: boolean;
168 |   mdComponents: typeof mdComponents;
169 |   handleCopy: (text: string, messageId: string) => void;
170 |   copiedMessageId: string | null;
171 | }
172 | 
173 | // AiMessageBubble Component
174 | const AiMessageBubble: React.FC<AiMessageBubbleProps> = ({
175 |   message,
176 |   historicalActivity,
177 |   liveActivity,
178 |   isLastMessage,
179 |   isOverallLoading,
180 |   mdComponents,
181 |   handleCopy,
182 |   copiedMessageId,
183 | }) => {
184 |   // Determine which activity events to show and if it's for a live loading message
185 |   const activityForThisBubble =
186 |     isLastMessage && isOverallLoading ? liveActivity : historicalActivity;
187 |   const isLiveActivityForThisBubble = isLastMessage && isOverallLoading;
188 | 
189 |   return (
190 |     <div className={`relative break-words flex flex-col`}>
191 |       {activityForThisBubble && activityForThisBubble.length > 0 && (
192 |         <div className="mb-3 border-b border-neutral-700 pb-3 text-xs">
193 |           <ActivityTimeline
194 |             processedEvents={activityForThisBubble}
195 |             isLoading={isLiveActivityForThisBubble}
196 |           />
197 |         </div>
198 |       )}
199 |       <ReactMarkdown components={mdComponents}>
200 |         {typeof message.content === "string"
201 |           ? message.content
202 |           : JSON.stringify(message.content)}
203 |       </ReactMarkdown>
204 |       <Button
205 |         variant="default"
206 |         className="cursor-pointer bg-neutral-700 border-neutral-600 text-neutral-300 self-end"
207 |         onClick={() =>
208 |           handleCopy(
209 |             typeof message.content === "string"
210 |               ? message.content
211 |               : JSON.stringify(message.content),
212 |             message.id!
213 |           )
214 |         }
215 |       >
216 |         {copiedMessageId === message.id ? "Copied" : "Copy"}
217 |         {copiedMessageId === message.id ? <CopyCheck /> : <Copy />}
218 |       </Button>
219 |     </div>
220 |   );
221 | };
222 | 
223 | interface ChatMessagesViewProps {
224 |   messages: Message[];
225 |   isLoading: boolean;
226 |   scrollAreaRef: React.RefObject<HTMLDivElement | null>;
227 |   onSubmit: (inputValue: string, effort: string, model: string) => void;
228 |   onCancel: () => void;
229 |   liveActivityEvents: ProcessedEvent[];
230 |   historicalActivities: Record<string, ProcessedEvent[]>;
231 | }
232 | 
233 | export function ChatMessagesView({
234 |   messages,
235 |   isLoading,
236 |   scrollAreaRef,
237 |   onSubmit,
238 |   onCancel,
239 |   liveActivityEvents,
240 |   historicalActivities,
241 | }: ChatMessagesViewProps) {
242 |   const [copiedMessageId, setCopiedMessageId] = useState<string | null>(null);
243 | 
244 |   const handleCopy = async (text: string, messageId: string) => {
245 |     try {
246 |       await navigator.clipboard.writeText(text);
247 |       setCopiedMessageId(messageId);
248 |       setTimeout(() => setCopiedMessageId(null), 2000); // Reset after 2 seconds
249 |     } catch (err) {
250 |       console.error("Failed to copy text: ", err);
251 |     }
252 |   };
253 | 
254 |   return (
255 |     <div className="flex flex-col h-full">
256 |       <ScrollArea className="flex-grow" ref={scrollAreaRef}>
257 |         <div className="p-4 md:p-6 space-y-2 max-w-4xl mx-auto pt-16">
258 |           {messages.map((message, index) => {
259 |             const isLast = index === messages.length - 1;
260 |             return (
261 |               <div key={message.id || `msg-${index}`} className="space-y-3">
262 |                 <div
263 |                   className={`flex items-start gap-3 ${
264 |                     message.type === "human" ? "justify-end" : ""
265 |                   }`}
266 |                 >
267 |                   {message.type === "human" ? (
268 |                     <HumanMessageBubble
269 |                       message={message}
270 |                       mdComponents={mdComponents}
271 |                     />
272 |                   ) : (
273 |                     <AiMessageBubble
274 |                       message={message}
275 |                       historicalActivity={historicalActivities[message.id!]}
276 |                       liveActivity={liveActivityEvents} // Pass global live events
277 |                       isLastMessage={isLast}
278 |                       isOverallLoading={isLoading} // Pass global loading state
279 |                       mdComponents={mdComponents}
280 |                       handleCopy={handleCopy}
281 |                       copiedMessageId={copiedMessageId}
282 |                     />
283 |                   )}
284 |                 </div>
285 |               </div>
286 |             );
287 |           })}
288 |           {isLoading &&
289 |             (messages.length === 0 ||
290 |               messages[messages.length - 1].type === "human") && (
291 |               <div className="flex items-start gap-3 mt-3">
292 |                 {" "}
293 |                 {/* AI message row structure */}
294 |                 <div className="relative group max-w-[85%] md:max-w-[80%] rounded-xl p-3 shadow-sm break-words bg-neutral-800 text-neutral-100 rounded-bl-none w-full min-h-[56px]">
295 |                   {liveActivityEvents.length > 0 ? (
296 |                     <div className="text-xs">
297 |                       <ActivityTimeline
298 |                         processedEvents={liveActivityEvents}
299 |                         isLoading={true}
300 |                       />
301 |                     </div>
302 |                   ) : (
303 |                     <div className="flex items-center justify-start h-full">
304 |                       <Loader2 className="h-5 w-5 animate-spin text-neutral-400 mr-2" />
305 |                       <span>Processing...</span>
306 |                     </div>
307 |                   )}
308 |                 </div>
309 |               </div>
310 |             )}
311 |         </div>
312 |       </ScrollArea>
313 |       <InputForm
314 |         onSubmit={onSubmit}
315 |         isLoading={isLoading}
316 |         onCancel={onCancel}
317 |         hasHistory={messages.length > 0}
318 |       />
319 |     </div>
320 |   );
321 | }
322 | 


--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
  1 |                                  Apache License
  2 |                            Version 2.0, January 2004
  3 |                         http://www.apache.org/licenses/
  4 | 
  5 |    TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
  6 | 
  7 |    1. Definitions.
  8 | 
  9 |       "License" shall mean the terms and conditions for use, reproduction,
 10 |       and distribution as defined by Sections 1 through 9 of this document.
 11 | 
 12 |       "Licensor" shall mean the copyright owner or entity authorized by
 13 |       the copyright owner that is granting the License.
 14 | 
 15 |       "Legal Entity" shall mean the union of the acting entity and all
 16 |       other entities that control, are controlled by, or are under common
 17 |       control with that entity. For the purposes of this definition,
 18 |       "control" means (i) the power, direct or indirect, to cause the
 19 |       direction or management of such entity, whether by contract or
 20 |       otherwise, or (ii) ownership of fifty percent (50%) or more of the
 21 |       outstanding shares, or (iii) beneficial ownership of such entity.
 22 | 
 23 |       "You" (or "Your") shall mean an individual or Legal Entity
 24 |       exercising permissions granted by this License.
 25 | 
 26 |       "Source" form shall mean the preferred form for making modifications,
 27 |       including but not limited to software source code, documentation
 28 |       source, and configuration files.
 29 | 
 30 |       "Object" form shall mean any form resulting from mechanical
 31 |       transformation or translation of a Source form, including but
 32 |       not limited to compiled object code, generated documentation,
 33 |       and conversions to other media types.
 34 | 
 35 |       "Work" shall mean the work of authorship, whether in Source or
 36 |       Object form, made available under the License, as indicated by a
 37 |       copyright notice that is included in or attached to the work
 38 |       (an example is provided in the Appendix below).
 39 | 
 40 |       "Derivative Works" shall mean any work, whether in Source or Object
 41 |       form, that is based on (or derived from) the Work and for which the
 42 |       editorial revisions, annotations, elaborations, or other modifications
 43 |       represent, as a whole, an original work of authorship. For the purposes
 44 |       of this License, Derivative Works shall not include works that remain
 45 |       separable from, or merely link (or bind by name) to the interfaces of,
 46 |       the Work and Derivative Works thereof.
 47 | 
 48 |       "Contribution" shall mean any work of authorship, including
 49 |       the original version of the Work and any modifications or additions
 50 |       to that Work or Derivative Works thereof, that is intentionally
 51 |       submitted to Licensor for inclusion in the Work by the copyright owner
 52 |       or by an individual or Legal Entity authorized to submit on behalf of
 53 |       the copyright owner. For the purposes of this definition, "submitted"
 54 |       means any form of electronic, verbal, or written communication sent
 55 |       to the Licensor or its representatives, including but not limited to
 56 |       communication on electronic mailing lists, source code control systems,
 57 |       and issue tracking systems that are managed by, or on behalf of, the
 58 |       Licensor for the purpose of discussing and improving the Work, but
 59 |       excluding communication that is conspicuously marked or otherwise
 60 |       designated in writing by the copyright owner as "Not a Contribution."
 61 | 
 62 |       "Contributor" shall mean Licensor and any individual or Legal Entity
 63 |       on behalf of whom a Contribution has been received by Licensor and
 64 |       subsequently incorporated within the Work.
 65 | 
 66 |    2. Grant of Copyright License. Subject to the terms and conditions of
 67 |       this License, each Contributor hereby grants to You a perpetual,
 68 |       worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 69 |       copyright license to reproduce, prepare Derivative Works of,
 70 |       publicly display, publicly perform, sublicense, and distribute the
 71 |       Work and such Derivative Works in Source or Object form.
 72 | 
 73 |    3. Grant of Patent License. Subject to the terms and conditions of
 74 |       this License, each Contributor hereby grants to You a perpetual,
 75 |       worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 76 |       (except as stated in this section) patent license to make, have made,
 77 |       use, offer to sell, sell, import, and otherwise transfer the Work,
 78 |       where such license applies only to those patent claims licensable
 79 |       by such Contributor that are necessarily infringed by their
 80 |       Contribution(s) alone or by combination of their Contribution(s)
 81 |       with the Work to which such Contribution(s) was submitted. If You
 82 |       institute patent litigation against any entity (including a
 83 |       cross-claim or counterclaim in a lawsuit) alleging that the Work
 84 |       or a Contribution incorporated within the Work constitutes direct
 85 |       or contributory patent infringement, then any patent licenses
 86 |       granted to You under this License for that Work shall terminate
 87 |       as of the date such litigation is filed.
 88 | 
 89 |    4. Redistribution. You may reproduce and distribute copies of the
 90 |       Work or Derivative Works thereof in any medium, with or without
 91 |       modifications, and in Source or Object form, provided that You
 92 |       meet the following conditions:
 93 | 
 94 |       (a) You must give any other recipients of the Work or
 95 |           Derivative Works a copy of this License; and
 96 | 
 97 |       (b) You must cause any modified files to carry prominent notices
 98 |           stating that You changed the files; and
 99 | 
100 |       (c) You must retain, in the Source form of any Derivative Works
101 |           that You distribute, all copyright, patent, trademark, and
102 |           attribution notices from the Source form of the Work,
103 |           excluding those notices that do not pertain to any part of
104 |           the Derivative Works; and
105 | 
106 |       (d) If the Work includes a "NOTICE" text file as part of its
107 |           distribution, then any Derivative Works that You distribute must
108 |           include a readable copy of the attribution notices contained
109 |           within such NOTICE file, excluding those notices that do not
110 |           pertain to any part of the Derivative Works, in at least one
111 |           of the following places: within a NOTICE text file distributed
112 |           as part of the Derivative Works; within the Source form or
113 |           documentation, if provided along with the Derivative Works; or,
114 |           within a display generated by the Derivative Works, if and
115 |           wherever such third-party notices normally appear. The contents
116 |           of the NOTICE file are for informational purposes only and
117 |           do not modify the License. You may add Your own attribution
118 |           notices within Derivative Works that You distribute, alongside
119 |           or as an addendum to the NOTICE text from the Work, provided
120 |           that such additional attribution notices cannot be construed
121 |           as modifying the License.
122 | 
123 |       You may add Your own copyright statement to Your modifications and
124 |       may provide additional or different license terms and conditions
125 |       for use, reproduction, or distribution of Your modifications, or
126 |       for any such Derivative Works as a whole, provided Your use,
127 |       reproduction, and distribution of the Work otherwise complies with
128 |       the conditions stated in this License.
129 | 
130 |    5. Submission of Contributions. Unless You explicitly state otherwise,
131 |       any Contribution intentionally submitted for inclusion in the Work
132 |       by You to the Licensor shall be under the terms and conditions of
133 |       this License, without any additional terms or conditions.
134 |       Notwithstanding the above, nothing herein shall supersede or modify
135 |       the terms of any separate license agreement you may have executed
136 |       with Licensor regarding such Contributions.
137 | 
138 |    6. Trademarks. This License does not grant permission to use the trade
139 |       names, trademarks, service marks, or product names of the Licensor,
140 |       except as required for reasonable and customary use in describing the
141 |       origin of the Work and reproducing the content of the NOTICE file.
142 | 
143 |    7. Disclaimer of Warranty. Unless required by applicable law or
144 |       agreed to in writing, Licensor provides the Work (and each
145 |       Contributor provides its Contributions) on an "AS IS" BASIS,
146 |       WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
147 |       implied, including, without limitation, any warranties or conditions
148 |       of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
149 |       PARTICULAR PURPOSE. You are solely responsible for determining the
150 |       appropriateness of using or redistributing the Work and assume any
151 |       risks associated with Your exercise of permissions under this License.
152 | 
153 |    8. Limitation of Liability. In no event and under no legal theory,
154 |       whether in tort (including negligence), contract, or otherwise,
155 |       unless required by applicable law (such as deliberate and grossly
156 |       negligent acts) or agreed to in writing, shall any Contributor be
157 |       liable to You for damages, including any direct, indirect, special,
158 |       incidental, or consequential damages of any character arising as a
159 |       result of this License or out of the use or inability to use the
160 |       Work (including but not limited to damages for loss of goodwill,
161 |       work stoppage, computer failure or malfunction, or any and all
162 |       other commercial damages or losses), even if such Contributor
163 |       has been advised of the possibility of such damages.
164 | 
165 |    9. Accepting Warranty or Additional Liability. While redistributing
166 |       the Work or Derivative Works thereof, You may choose to offer,
167 |       and charge a fee for, acceptance of support, warranty, indemnity,
168 |       or other liability obligations and/or rights consistent with this
169 |       License. However, in accepting such obligations, You may act only
170 |       on Your own behalf and on Your sole responsibility, not on behalf
171 |       of any other Contributor, and only if You agree to indemnify,
172 |       defend, and hold each Contributor harmless for any liability
173 |       incurred by, or claims asserted against, such Contributor by reason
174 |       of your accepting any such warranty or additional liability.
175 | 
176 |    END OF TERMS AND CONDITIONS
177 | 
178 |    APPENDIX: How to apply the Apache License to your work.
179 | 
180 |       To apply the Apache License to your work, attach the following
181 |       boilerplate notice, with the fields enclosed by brackets "[]"
182 |       replaced with your own identifying information. (Don't include
183 |       the brackets!)  The text should be enclosed in the appropriate
184 |       comment syntax for the file format. We also recommend that a
185 |       file or class name and description of purpose be included on the
186 |       same "printed page" as the copyright notice for easier
187 |       identification within third-party archives.
188 | 
189 |    Copyright [yyyy] [name of copyright owner]
190 | 
191 |    Licensed under the Apache License, Version 2.0 (the "License");
192 |    you may not use this file except in compliance with the License.
193 |    You may obtain a copy of the License at
194 | 
195 |        http://www.apache.org/licenses/LICENSE-2.0
196 | 
197 |    Unless required by applicable law or agreed to in writing, software
198 |    distributed under the License is distributed on an "AS IS" BASIS,
199 |    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200 |    See the License for the specific language governing permissions and
201 |    limitations under the License.
202 | 


--------------------------------------------------------------------------------
/backend/src/agent/graph.py:
--------------------------------------------------------------------------------
  1 | import os
  2 | 
  3 | from dotenv import load_dotenv
  4 | from duckduckgo_search import DDGS
  5 | from langchain_core.messages import AIMessage
  6 | from langchain_core.runnables import RunnableConfig
  7 | from langchain_openai import ChatOpenAI
  8 | from langgraph.graph import END, START, StateGraph
  9 | from langgraph.types import Send
 10 | 
 11 | from agent.configuration import Configuration
 12 | from agent.prompts import (
 13 |     answer_instructions,
 14 |     get_current_date,
 15 |     query_writer_instructions,
 16 |     reflection_instructions,
 17 | )
 18 | from agent.state import (
 19 |     OverallState,
 20 |     QueryGenerationState,
 21 |     ReflectionState,
 22 |     WebSearchState,
 23 | )
 24 | from agent.tools_and_schemas import Reflection, SearchQueryList
 25 | from agent.utils import get_research_topic
 26 | 
 27 | load_dotenv()
 28 | 
 29 | OPENAI_API_BASE = os.getenv("OPENAI_API_BASE", "http://localhost:1234/v1")
 30 | OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "lm-studio")
 31 | 
 32 | 
 33 | # Nodes
 34 | def generate_query(state: OverallState, config: RunnableConfig) -> QueryGenerationState:
 35 |     """LangGraph node that generates a search queries based on the User's question.
 36 | 
 37 |     Uses the configured LLM to create optimized search queries for web research
 38 |     based on the user's question.
 39 | 
 40 |     Args:
 41 |         state: Current graph state containing the User's question
 42 |         config: Configuration for the runnable, including LLM provider settings
 43 | 
 44 |     Returns:
 45 |         Dictionary with state update, including search_query key containing the generated query
 46 |     """
 47 |     configurable = Configuration.from_runnable_config(config)
 48 | 
 49 |     # check for custom initial search query count
 50 |     if state.get("initial_search_query_count") is None:
 51 |         state["initial_search_query_count"] = configurable.number_of_initial_queries
 52 | 
 53 |     llm = ChatOpenAI(
 54 |         model=configurable.query_generator_model,
 55 |         temperature=1.0,
 56 |         max_retries=2,
 57 |         base_url=OPENAI_API_BASE,
 58 |         api_key=OPENAI_API_KEY,
 59 |     )
 60 |     structured_llm = llm.with_structured_output(SearchQueryList)
 61 | 
 62 |     # Format the prompt
 63 |     current_date = get_current_date()
 64 |     formatted_prompt = query_writer_instructions.format(
 65 |         current_date=current_date,
 66 |         research_topic=get_research_topic(state["messages"]),
 67 |         number_queries=state["initial_search_query_count"],
 68 |     )
 69 |     # Generate the search queries
 70 |     result = structured_llm.invoke(formatted_prompt)
 71 |     return {"query_list": result.query}
 72 | 
 73 | 
 74 | def continue_to_web_research(state: QueryGenerationState):
 75 |     """LangGraph node that sends the search queries to the web research node.
 76 | 
 77 |     This is used to spawn n number of web research nodes, one for each search query.
 78 |     """
 79 |     return [
 80 |         Send("web_research", {"search_query": search_query, "id": int(idx)})
 81 |         for idx, search_query in enumerate(state["query_list"])
 82 |     ]
 83 | 
 84 | import requests
 85 | from bs4 import BeautifulSoup
 86 | import urllib.parse
 87 | 
 88 | import requests
 89 | 
 90 | import requests
 91 | import os
 92 | 
 93 | DEFAULT_SEARCH_ENGINE_TIMEOUT = 100
 94 | REFERENCE_COUNT = 5
 95 | GOOGLE_SEARCH_ENDPOINT = "https://customsearch.googleapis.com/customsearch/v1"
 96 | 
 97 | def search_with_google(query: str, subscription_key: str, cx: str):
 98 |     """
 99 |     Search with google and return the contexts.
100 |     """
101 |     params = {
102 |         "key": subscription_key,
103 |         "cx": cx,
104 |         "q": query,
105 |         "num": REFERENCE_COUNT,
106 |     }
107 |     response = requests.get(
108 |         GOOGLE_SEARCH_ENDPOINT, params=params, timeout=DEFAULT_SEARCH_ENGINE_TIMEOUT
109 |     )
110 |     if not response.ok:
111 |         print(f"{response.status_code} {response.text}")
112 |         raise HTTPException(response.status_code, "Search engine error.")
113 |     json_content = response.json()
114 |     try:
115 |         contexts = json_content["items"][:REFERENCE_COUNT]
116 |     except KeyError:
117 |         print(f"Error encountered: {json_content}")
118 |         return []
119 |     return contexts
120 | 
121 | 
122 | 
123 | def web_research(state: WebSearchState, config: RunnableConfig) -> OverallState:
124 |     """LangGraph node that performs web research using DuckDuckGo.
125 | 
126 |     Executes a web search using the DuckDuckGo Search API and formats the
127 |     results for later processing.
128 | 
129 |     Args:
130 |         state: Current graph state containing the search query and research loop count
131 |         config: Configuration for the runnable, including search API settings
132 | 
133 |     Returns:
134 |         Dictionary with state update, including sources_gathered, research_loop_count, and web_research_results
135 |     """
136 |     # Configure
137 |     configurable = Configuration.from_runnable_config(config)
138 |     query = state["search_query"]
139 | 
140 |     search_api_key = os.environ["GOOGLE_SEARCH_API_KEY"]
141 |     # with DDGS() as ddgs:
142 |     #     print(query, "xxx")
143 |     #     results = list(ddgs.text(query, max_results=5))
144 | 
145 |     results = search_with_google(query, search_api_key, 
146 |     os.environ["GOOGLE_SEARCH_CX"])
147 | 
148 |     formatted_lines = []
149 |     sources_gathered = []
150 |     for res in results:
151 |         # print(res, "qqq")
152 |         title = res.get("title", "")
153 |         # href = res.get("href", "")
154 |         # body = res.get("body", "")
155 |         href = res.get("link", "")
156 |         body = res.get("snippet", "")
157 |         formatted_lines.append(f"{title}: {body} ({href})")
158 |         sources_gathered.append({"label": title, "short_url": href, "value": href})
159 | 
160 |     modified_text = "\n".join(formatted_lines)
161 | 
162 |     return {
163 |         "sources_gathered": sources_gathered,
164 |         "search_query": [query],
165 |         "web_research_result": [modified_text],
166 |     }
167 | 
168 | 
169 | def reflection(state: OverallState, config: RunnableConfig) -> ReflectionState:
170 |     """LangGraph node that identifies knowledge gaps and generates potential follow-up queries.
171 | 
172 |     Analyzes the current summary to identify areas for further research and generates
173 |     potential follow-up queries. Uses structured output to extract
174 |     the follow-up query in JSON format.
175 | 
176 |     Args:
177 |         state: Current graph state containing the running summary and research topic
178 |         config: Configuration for the runnable, including LLM provider settings
179 | 
180 |     Returns:
181 |         Dictionary with state update, including search_query key containing the generated follow-up query
182 |     """
183 |     configurable = Configuration.from_runnable_config(config)
184 |     # Increment the research loop count and get the reasoning model
185 |     state["research_loop_count"] = state.get("research_loop_count", 0) + 1
186 |     reasoning_model = state.get("reasoning_model") or configurable.reasoning_model
187 | 
188 |     # Format the prompt
189 |     current_date = get_current_date()
190 |     formatted_prompt = reflection_instructions.format(
191 |         current_date=current_date,
192 |         research_topic=get_research_topic(state["messages"]),
193 |         summaries="\n\n---\n\n".join(state["web_research_result"]),
194 |     )
195 |     llm = ChatOpenAI(
196 |         model=reasoning_model,
197 |         temperature=1.0,
198 |         max_retries=2,
199 |         base_url=OPENAI_API_BASE,
200 |         api_key=OPENAI_API_KEY,
201 |     )
202 |     result = llm.with_structured_output(Reflection).invoke(formatted_prompt)
203 | 
204 |     return {
205 |         "is_sufficient": result.is_sufficient,
206 |         "knowledge_gap": result.knowledge_gap,
207 |         "follow_up_queries": result.follow_up_queries,
208 |         "research_loop_count": state["research_loop_count"],
209 |         "number_of_ran_queries": len(state["search_query"]),
210 |     }
211 | 
212 | 
213 | def evaluate_research(
214 |     state: ReflectionState,
215 |     config: RunnableConfig,
216 | ) -> OverallState:
217 |     """LangGraph routing function that determines the next step in the research flow.
218 | 
219 |     Controls the research loop by deciding whether to continue gathering information
220 |     or to finalize the summary based on the configured maximum number of research loops.
221 | 
222 |     Args:
223 |         state: Current graph state containing the research loop count
224 |         config: Configuration for the runnable, including max_research_loops setting
225 | 
226 |     Returns:
227 |         String literal indicating the next node to visit ("web_research" or "finalize_summary")
228 |     """
229 |     configurable = Configuration.from_runnable_config(config)
230 |     max_research_loops = (
231 |         state.get("max_research_loops")
232 |         if state.get("max_research_loops") is not None
233 |         else configurable.max_research_loops
234 |     )
235 |     if state["is_sufficient"] or state["research_loop_count"] >= max_research_loops:
236 |         return "finalize_answer"
237 |     else:
238 |         return [
239 |             Send(
240 |                 "web_research",
241 |                 {
242 |                     "search_query": follow_up_query,
243 |                     "id": state["number_of_ran_queries"] + int(idx),
244 |                 },
245 |             )
246 |             for idx, follow_up_query in enumerate(state["follow_up_queries"])
247 |         ]
248 | 
249 | 
250 | def finalize_answer(state: OverallState, config: RunnableConfig):
251 |     """LangGraph node that finalizes the research summary.
252 | 
253 |     Prepares the final output by deduplicating and formatting sources, then
254 |     combining them with the running summary to create a well-structured
255 |     research report with proper citations.
256 | 
257 |     Args:
258 |         state: Current graph state containing the running summary and sources gathered
259 | 
260 |     Returns:
261 |         Dictionary with state update, including running_summary key containing the formatted final summary with sources
262 |     """
263 |     configurable = Configuration.from_runnable_config(config)
264 |     reasoning_model = state.get("reasoning_model") or configurable.reasoning_model
265 | 
266 |     # Format the prompt
267 |     current_date = get_current_date()
268 |     formatted_prompt = answer_instructions.format(
269 |         current_date=current_date,
270 |         research_topic=get_research_topic(state["messages"]),
271 |         summaries="\n---\n\n".join(state["web_research_result"]),
272 |     )
273 | 
274 |     llm = ChatOpenAI(
275 |         model=reasoning_model,
276 |         temperature=0,
277 |         max_retries=2,
278 |         base_url=OPENAI_API_BASE,
279 |         api_key=OPENAI_API_KEY,
280 |     )
281 |     result = llm.invoke(formatted_prompt)
282 | 
283 |     return {
284 |         "messages": [AIMessage(content=result.content)],
285 |         "sources_gathered": state["sources_gathered"],
286 |     }
287 | 
288 | 
289 | # Create our Agent Graph
290 | builder = StateGraph(OverallState, config_schema=Configuration)
291 | 
292 | # Define the nodes we will cycle between
293 | builder.add_node("generate_query", generate_query)
294 | builder.add_node("web_research", web_research)
295 | builder.add_node("reflection", reflection)
296 | builder.add_node("finalize_answer", finalize_answer)
297 | 
298 | # Set the entrypoint as `generate_query`
299 | # This means that this node is the first one called
300 | builder.add_edge(START, "generate_query")
301 | # Add conditional edge to continue with search queries in a parallel branch
302 | builder.add_conditional_edges(
303 |     "generate_query", continue_to_web_research, ["web_research"]
304 | )
305 | # Reflect on the web research
306 | builder.add_edge("web_research", "reflection")
307 | # Evaluate the research
308 | builder.add_conditional_edges(
309 |     "reflection", evaluate_research, ["web_research", "finalize_answer"]
310 | )
311 | # Finalize the answer
312 | builder.add_edge("finalize_answer", END)
313 | 
314 | graph = builder.compile(name="pro-search-agent")
315 | 


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