Skip to content

Commit 6f9b5cd

Browse files
tysonthomas9tysonthomas9
authored
Merge pull request #16 from tysonthomas9/feature/enhancements
- Improve LLM SDK Integration - Added tracing with Langfuse - Added VectorDB support for RAG - Added more LLM providers
2 parents ad14051 + 6452464 commit 6f9b5cd

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

49 files changed

+8867
-2275
lines changed

‎.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,3 +57,4 @@ test/perf/.generated
5757

5858
# Dependencies
5959
node_modules/
60+
**/.idea/

‎config/gni/devtools_grd_files.gni

Lines changed: 19 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -604,36 +604,50 @@ grd_files_release_sources = [
604604
"front_end/panels/ai_chat/ui/HelpDialog.js",
605605
"front_end/panels/ai_chat/ui/PromptEditDialog.js",
606606
"front_end/panels/ai_chat/ui/SettingsDialog.js",
607+
"front_end/panels/ai_chat/ui/EvaluationDialog.js",
607608
"front_end/panels/ai_chat/core/AgentService.js",
608609
"front_end/panels/ai_chat/core/State.js",
609610
"front_end/panels/ai_chat/core/Graph.js",
610611
"front_end/panels/ai_chat/core/Types.js",
611612
"front_end/panels/ai_chat/core/Constants.js",
612613
"front_end/panels/ai_chat/core/ConfigurableGraph.js",
613614
"front_end/panels/ai_chat/core/GraphConfigs.js",
614-
"front_end/panels/ai_chat/core/OpenAIClient.js",
615-
"front_end/panels/ai_chat/core/LiteLLMClient.js",
616-
"front_end/panels/ai_chat/core/UnifiedLLMClient.js",
617615
"front_end/panels/ai_chat/core/BaseOrchestratorAgent.js",
618616
"front_end/panels/ai_chat/core/PageInfoManager.js",
619617
"front_end/panels/ai_chat/core/AgentNodes.js",
620-
"front_end/panels/ai_chat/core/ChatOpenAI.js",
621-
"front_end/panels/ai_chat/core/ChatLiteLLM.js",
622618
"front_end/panels/ai_chat/core/GraphHelpers.js",
623619
"front_end/panels/ai_chat/core/StateGraph.js",
620+
"front_end/panels/ai_chat/core/Logger.js",
621+
"front_end/panels/ai_chat/LLM/LLMTypes.js",
622+
"front_end/panels/ai_chat/LLM/LLMProvider.js",
623+
"front_end/panels/ai_chat/LLM/LLMProviderRegistry.js",
624+
"front_end/panels/ai_chat/LLM/LLMErrorHandler.js",
625+
"front_end/panels/ai_chat/LLM/LLMResponseParser.js",
626+
"front_end/panels/ai_chat/LLM/OpenAIProvider.js",
627+
"front_end/panels/ai_chat/LLM/LiteLLMProvider.js",
628+
"front_end/panels/ai_chat/LLM/GroqProvider.js",
629+
"front_end/panels/ai_chat/LLM/OpenRouterProvider.js",
630+
"front_end/panels/ai_chat/LLM/LLMClient.js",
624631
"front_end/panels/ai_chat/tools/Tools.js",
625632
"front_end/panels/ai_chat/tools/CombinedExtractionTool.js",
626633
"front_end/panels/ai_chat/tools/CritiqueTool.js",
627634
"front_end/panels/ai_chat/tools/FetcherTool.js",
628635
"front_end/panels/ai_chat/tools/FinalizeWithCritiqueTool.js",
629636
"front_end/panels/ai_chat/tools/HTMLToMarkdownTool.js",
630637
"front_end/panels/ai_chat/tools/SchemaBasedExtractorTool.js",
638+
"front_end/panels/ai_chat/tools/StreamlinedSchemaExtractorTool.js",
631639
"front_end/panels/ai_chat/tools/VisitHistoryManager.js",
632640
"front_end/panels/ai_chat/tools/FullPageAccessibilityTreeToMarkdownTool.js",
641+
"front_end/panels/ai_chat/tools/VectorDBClient.js",
642+
"front_end/panels/ai_chat/tools/BookmarkStoreTool.js",
643+
"front_end/panels/ai_chat/tools/DocumentSearchTool.js",
633644
"front_end/panels/ai_chat/common/utils.js",
634645
"front_end/panels/ai_chat/common/log.js",
635646
"front_end/panels/ai_chat/common/context.js",
636647
"front_end/panels/ai_chat/common/page.js",
648+
"front_end/panels/ai_chat/tracing/TracingProvider.js",
649+
"front_end/panels/ai_chat/tracing/LangfuseProvider.js",
650+
"front_end/panels/ai_chat/tracing/TracingConfig.js",
637651
"front_end/panels/ai_chat/ai_chat-meta.js",
638652
"front_end/panels/ai_chat/ai_chat.js",
639653
"front_end/panels/ai_chat/ai_chat_impl.js",
@@ -656,7 +670,6 @@ grd_files_release_sources = [
656670
"front_end/panels/ai_chat/evaluation/utils/PromptTemplates.js",
657671
"front_end/panels/ai_chat/evaluation/utils/ResponseParsingUtils.js",
658672
"front_end/panels/ai_chat/evaluation/utils/SanitizationUtils.js",
659-
"front_end/panels/ai_chat/ui/EvaluationDialog.js",
660673
"front_end/panels/animation/animation-meta.js",
661674
"front_end/panels/animation/animation.js",
662675
"front_end/panels/application/application-meta.js",

‎front_end/panels/ai_chat/BUILD.gn

Lines changed: 34 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -29,18 +29,23 @@ devtools_module("ai_chat") {
2929
"core/AgentService.ts",
3030
"core/Constants.ts",
3131
"core/GraphConfigs.ts",
32-
"core/OpenAIClient.ts",
33-
"core/LiteLLMClient.ts",
34-
"core/UnifiedLLMClient.ts",
3532
"core/ConfigurableGraph.ts",
3633
"core/BaseOrchestratorAgent.ts",
3734
"core/PageInfoManager.ts",
3835
"core/AgentNodes.ts",
39-
"core/ChatOpenAI.ts",
40-
"core/ChatLiteLLM.ts",
4136
"core/GraphHelpers.ts",
4237
"core/StateGraph.ts",
4338
"core/Logger.ts",
39+
"LLM/LLMTypes.ts",
40+
"LLM/LLMProvider.ts",
41+
"LLM/LLMProviderRegistry.ts",
42+
"LLM/LLMErrorHandler.ts",
43+
"LLM/LLMResponseParser.ts",
44+
"LLM/OpenAIProvider.ts",
45+
"LLM/LiteLLMProvider.ts",
46+
"LLM/GroqProvider.ts",
47+
"LLM/OpenRouterProvider.ts",
48+
"LLM/LLMClient.ts",
4449
"tools/Tools.ts",
4550
"tools/CritiqueTool.ts",
4651
"tools/FetcherTool.ts",
@@ -51,6 +56,9 @@ devtools_module("ai_chat") {
5156
"tools/StreamlinedSchemaExtractorTool.ts",
5257
"tools/CombinedExtractionTool.ts",
5358
"tools/FullPageAccessibilityTreeToMarkdownTool.ts",
59+
"tools/VectorDBClient.ts",
60+
"tools/BookmarkStoreTool.ts",
61+
"tools/DocumentSearchTool.ts",
5462
"agent_framework/ConfigurableAgentTool.ts",
5563
"agent_framework/AgentRunner.ts",
5664
"agent_framework/implementation/ConfiguredAgents.ts",
@@ -74,6 +82,9 @@ devtools_module("ai_chat") {
7482
"common/log.ts",
7583
"common/context.ts",
7684
"common/page.ts",
85+
"tracing/TracingProvider.ts",
86+
"tracing/LangfuseProvider.ts",
87+
"tracing/TracingConfig.ts",
7788
]
7889

7990
deps = [
@@ -107,26 +118,36 @@ _ai_chat_sources = [
107118
"core/AgentService.ts",
108119
"core/Constants.ts",
109120
"core/GraphConfigs.ts",
110-
"core/OpenAIClient.ts",
111-
"core/LiteLLMClient.ts",
112-
"core/UnifiedLLMClient.ts",
113121
"core/ConfigurableGraph.ts",
114122
"core/BaseOrchestratorAgent.ts",
115123
"core/PageInfoManager.ts",
116124
"core/AgentNodes.ts",
117-
"core/ChatOpenAI.ts",
118-
"core/ChatLiteLLM.ts",
119125
"core/GraphHelpers.ts",
120126
"core/StateGraph.ts",
127+
"core/Logger.ts",
128+
"LLM/LLMTypes.ts",
129+
"LLM/LLMProvider.ts",
130+
"LLM/LLMProviderRegistry.ts",
131+
"LLM/LLMErrorHandler.ts",
132+
"LLM/LLMResponseParser.ts",
133+
"LLM/OpenAIProvider.ts",
134+
"LLM/LiteLLMProvider.ts",
135+
"LLM/GroqProvider.ts",
136+
"LLM/OpenRouterProvider.ts",
137+
"LLM/LLMClient.ts",
121138
"tools/Tools.ts",
122139
"tools/CritiqueTool.ts",
123140
"tools/FetcherTool.ts",
124141
"tools/FinalizeWithCritiqueTool.ts",
125142
"tools/VisitHistoryManager.ts",
126143
"tools/HTMLToMarkdownTool.ts",
127144
"tools/SchemaBasedExtractorTool.ts",
145+
"tools/StreamlinedSchemaExtractorTool.ts",
128146
"tools/CombinedExtractionTool.ts",
129147
"tools/FullPageAccessibilityTreeToMarkdownTool.ts",
148+
"tools/VectorDBClient.ts",
149+
"tools/BookmarkStoreTool.ts",
150+
"tools/DocumentSearchTool.ts",
130151
"agent_framework/ConfigurableAgentTool.ts",
131152
"agent_framework/AgentRunner.ts",
132153
"agent_framework/implementation/ConfiguredAgents.ts",
@@ -150,6 +171,9 @@ _ai_chat_sources = [
150171
"common/log.ts",
151172
"common/context.ts",
152173
"common/page.ts",
174+
"tracing/TracingProvider.ts",
175+
"tracing/LangfuseProvider.ts",
176+
"tracing/TracingConfig.ts",
153177
]
154178

155179
# Construct the expected JS output paths for the metadata

0 commit comments

Comments
 (0)