agent-orcha 0.0.4 → 0.0.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +197 -221
- package/dist/lib/agents/agent-executor.d.ts +13 -4
- package/dist/lib/agents/agent-executor.d.ts.map +1 -1
- package/dist/lib/agents/agent-executor.js +307 -73
- package/dist/lib/agents/agent-executor.js.map +1 -1
- package/dist/lib/agents/agent-loader.d.ts +1 -1
- package/dist/lib/agents/agent-loader.d.ts.map +1 -1
- package/dist/lib/agents/agent-loader.js +10 -3
- package/dist/lib/agents/agent-loader.js.map +1 -1
- package/dist/lib/agents/index.d.ts +5 -5
- package/dist/lib/agents/index.js +4 -4
- package/dist/lib/agents/react-loop.d.ts +30 -0
- package/dist/lib/agents/react-loop.d.ts.map +1 -0
- package/dist/lib/agents/react-loop.js +155 -0
- package/dist/lib/agents/react-loop.js.map +1 -0
- package/dist/lib/agents/structured-output-wrapper.d.ts +3 -3
- package/dist/lib/agents/structured-output-wrapper.d.ts.map +1 -1
- package/dist/lib/agents/structured-output-wrapper.js +2 -2
- package/dist/lib/agents/structured-output-wrapper.js.map +1 -1
- package/dist/lib/agents/types.d.ts +132 -8
- package/dist/lib/agents/types.d.ts.map +1 -1
- package/dist/lib/agents/types.js +16 -2
- package/dist/lib/agents/types.js.map +1 -1
- package/dist/lib/functions/function-loader.d.ts +1 -1
- package/dist/lib/functions/function-loader.js +13 -5
- package/dist/lib/functions/function-loader.js.map +1 -1
- package/dist/lib/functions/index.d.ts +2 -2
- package/dist/lib/functions/index.js +1 -1
- package/dist/lib/functions/simple-function-wrapper.d.ts +2 -11
- package/dist/lib/functions/simple-function-wrapper.d.ts.map +1 -1
- package/dist/lib/functions/simple-function-wrapper.js +3 -3
- package/dist/lib/functions/simple-function-wrapper.js.map +1 -1
- package/dist/lib/index.d.ts +29 -17
- package/dist/lib/index.d.ts.map +1 -1
- package/dist/lib/index.js +21 -9
- package/dist/lib/index.js.map +1 -1
- package/dist/lib/integrations/collabnook.d.ts +56 -0
- package/dist/lib/integrations/collabnook.d.ts.map +1 -0
- package/dist/lib/integrations/collabnook.js +307 -0
- package/dist/lib/integrations/collabnook.js.map +1 -0
- package/dist/lib/integrations/integration-manager.d.ts +14 -0
- package/dist/lib/integrations/integration-manager.d.ts.map +1 -0
- package/dist/lib/integrations/integration-manager.js +109 -0
- package/dist/lib/integrations/integration-manager.js.map +1 -0
- package/dist/lib/integrations/types.d.ts +49 -0
- package/dist/lib/integrations/types.d.ts.map +1 -0
- package/dist/lib/integrations/types.js +14 -0
- package/dist/lib/integrations/types.js.map +1 -0
- package/dist/lib/knowledge/direct-mapper.d.ts +21 -0
- package/dist/lib/knowledge/direct-mapper.d.ts.map +1 -0
- package/dist/lib/knowledge/direct-mapper.js +134 -0
- package/dist/lib/knowledge/direct-mapper.js.map +1 -0
- package/dist/lib/knowledge/index.d.ts +9 -8
- package/dist/lib/knowledge/index.d.ts.map +1 -1
- package/dist/lib/knowledge/index.js +9 -7
- package/dist/lib/knowledge/index.js.map +1 -1
- package/dist/lib/knowledge/knowledge-store-metadata.d.ts +40 -0
- package/dist/lib/knowledge/knowledge-store-metadata.d.ts.map +1 -0
- package/dist/lib/knowledge/knowledge-store-metadata.js +93 -0
- package/dist/lib/knowledge/knowledge-store-metadata.js.map +1 -0
- package/dist/lib/knowledge/knowledge-store.d.ts +43 -0
- package/dist/lib/knowledge/knowledge-store.d.ts.map +1 -0
- package/dist/lib/knowledge/knowledge-store.js +603 -0
- package/dist/lib/knowledge/knowledge-store.js.map +1 -0
- package/dist/lib/knowledge/loaders/database-loader.d.ts +5 -5
- package/dist/lib/knowledge/loaders/database-loader.d.ts.map +1 -1
- package/dist/lib/knowledge/loaders/database-loader.js +21 -10
- package/dist/lib/knowledge/loaders/database-loader.js.map +1 -1
- package/dist/lib/knowledge/loaders/file-loaders.d.ts +39 -0
- package/dist/lib/knowledge/loaders/file-loaders.d.ts.map +1 -0
- package/dist/lib/knowledge/loaders/file-loaders.js +131 -0
- package/dist/lib/knowledge/loaders/file-loaders.js.map +1 -0
- package/dist/lib/knowledge/loaders/index.d.ts +3 -3
- package/dist/lib/knowledge/loaders/index.d.ts.map +1 -1
- package/dist/lib/knowledge/loaders/index.js +3 -3
- package/dist/lib/knowledge/loaders/index.js.map +1 -1
- package/dist/lib/knowledge/loaders/web-loader.d.ts +2 -2
- package/dist/lib/knowledge/loaders/web-loader.d.ts.map +1 -1
- package/dist/lib/knowledge/loaders/web-loader.js +22 -21
- package/dist/lib/knowledge/loaders/web-loader.js.map +1 -1
- package/dist/lib/knowledge/sqlite-store.d.ts +107 -0
- package/dist/lib/knowledge/sqlite-store.d.ts.map +1 -0
- package/dist/lib/knowledge/sqlite-store.js +327 -0
- package/dist/lib/knowledge/sqlite-store.js.map +1 -0
- package/dist/lib/knowledge/types.d.ts +72 -1432
- package/dist/lib/knowledge/types.d.ts.map +1 -1
- package/dist/lib/knowledge/types.js +36 -57
- package/dist/lib/knowledge/types.js.map +1 -1
- package/dist/lib/knowledge/utils/connection-pool.d.ts +3 -2
- package/dist/lib/knowledge/utils/connection-pool.d.ts.map +1 -1
- package/dist/lib/knowledge/utils/connection-pool.js +20 -10
- package/dist/lib/knowledge/utils/connection-pool.js.map +1 -1
- package/dist/lib/knowledge/utils/index.d.ts +2 -2
- package/dist/lib/knowledge/utils/index.js +2 -2
- package/dist/lib/llm/index.d.ts +5 -5
- package/dist/lib/llm/index.d.ts.map +1 -1
- package/dist/lib/llm/index.js +3 -3
- package/dist/lib/llm/index.js.map +1 -1
- package/dist/lib/llm/llm-call-logger.d.ts +37 -0
- package/dist/lib/llm/llm-call-logger.d.ts.map +1 -0
- package/dist/lib/llm/llm-call-logger.js +105 -0
- package/dist/lib/llm/llm-call-logger.js.map +1 -0
- package/dist/lib/llm/llm-config.d.ts +22 -16
- package/dist/lib/llm/llm-config.d.ts.map +1 -1
- package/dist/lib/llm/llm-config.js +19 -3
- package/dist/lib/llm/llm-config.js.map +1 -1
- package/dist/lib/llm/llm-factory.d.ts +3 -16
- package/dist/lib/llm/llm-factory.d.ts.map +1 -1
- package/dist/lib/llm/llm-factory.js +30 -48
- package/dist/lib/llm/llm-factory.js.map +1 -1
- package/dist/lib/llm/provider-detector.d.ts +1 -1
- package/dist/lib/llm/providers/anthropic-chat-model.d.ts +27 -0
- package/dist/lib/llm/providers/anthropic-chat-model.d.ts.map +1 -0
- package/dist/lib/llm/providers/anthropic-chat-model.js +186 -0
- package/dist/lib/llm/providers/anthropic-chat-model.js.map +1 -0
- package/dist/lib/llm/providers/gemini-chat-model.d.ts +29 -0
- package/dist/lib/llm/providers/gemini-chat-model.d.ts.map +1 -0
- package/dist/lib/llm/providers/gemini-chat-model.js +241 -0
- package/dist/lib/llm/providers/gemini-chat-model.js.map +1 -0
- package/dist/lib/llm/providers/gemini-embeddings.d.ts +14 -0
- package/dist/lib/llm/providers/gemini-embeddings.d.ts.map +1 -0
- package/dist/lib/llm/providers/gemini-embeddings.js +20 -0
- package/dist/lib/llm/providers/gemini-embeddings.js.map +1 -0
- package/dist/lib/llm/providers/openai-chat-model.d.ts +30 -0
- package/dist/lib/llm/providers/openai-chat-model.d.ts.map +1 -0
- package/dist/lib/llm/providers/openai-chat-model.js +196 -0
- package/dist/lib/llm/providers/openai-chat-model.js.map +1 -0
- package/dist/lib/llm/providers/openai-embeddings.d.ts +17 -0
- package/dist/lib/llm/providers/openai-embeddings.d.ts.map +1 -0
- package/dist/lib/llm/providers/openai-embeddings.js +35 -0
- package/dist/lib/llm/providers/openai-embeddings.js.map +1 -0
- package/dist/lib/llm/types.d.ts +2 -2
- package/dist/lib/llm/types.js +1 -1
- package/dist/lib/logger.d.ts +7 -0
- package/dist/lib/logger.d.ts.map +1 -1
- package/dist/lib/logger.js +43 -4
- package/dist/lib/logger.js.map +1 -1
- package/dist/lib/mcp/index.d.ts +3 -3
- package/dist/lib/mcp/index.js +2 -2
- package/dist/lib/mcp/mcp-client.d.ts +3 -2
- package/dist/lib/mcp/mcp-client.d.ts.map +1 -1
- package/dist/lib/mcp/mcp-client.js +5 -2
- package/dist/lib/mcp/mcp-client.js.map +1 -1
- package/dist/lib/mcp/types.d.ts +33 -23
- package/dist/lib/mcp/types.d.ts.map +1 -1
- package/dist/lib/mcp/types.js +11 -2
- package/dist/lib/mcp/types.js.map +1 -1
- package/dist/lib/memory/conversation-store.d.ts +2 -2
- package/dist/lib/memory/conversation-store.d.ts.map +1 -1
- package/dist/lib/memory/conversation-store.js +1 -1
- package/dist/lib/memory/index.d.ts +3 -2
- package/dist/lib/memory/index.d.ts.map +1 -1
- package/dist/lib/memory/index.js +3 -2
- package/dist/lib/memory/index.js.map +1 -1
- package/dist/lib/memory/memory-manager.d.ts +9 -0
- package/dist/lib/memory/memory-manager.d.ts.map +1 -0
- package/dist/lib/memory/memory-manager.js +34 -0
- package/dist/lib/memory/memory-manager.js.map +1 -0
- package/dist/lib/memory/types.d.ts +1 -1
- package/dist/lib/memory/types.d.ts.map +1 -1
- package/dist/lib/orchestrator.d.ts +78 -18
- package/dist/lib/orchestrator.d.ts.map +1 -1
- package/dist/lib/orchestrator.js +247 -55
- package/dist/lib/orchestrator.js.map +1 -1
- package/dist/lib/sandbox/html-to-markdown.d.ts +2 -0
- package/dist/lib/sandbox/html-to-markdown.d.ts.map +1 -0
- package/dist/lib/sandbox/html-to-markdown.js +64 -0
- package/dist/lib/sandbox/html-to-markdown.js.map +1 -0
- package/dist/lib/sandbox/index.d.ts +7 -0
- package/dist/lib/sandbox/index.d.ts.map +1 -0
- package/dist/lib/sandbox/index.js +6 -0
- package/dist/lib/sandbox/index.js.map +1 -0
- package/dist/lib/sandbox/sandbox-exec.d.ts +5 -0
- package/dist/lib/sandbox/sandbox-exec.d.ts.map +1 -0
- package/dist/lib/sandbox/sandbox-exec.js +35 -0
- package/dist/lib/sandbox/sandbox-exec.js.map +1 -0
- package/dist/lib/sandbox/sandbox-web.d.ts +5 -0
- package/dist/lib/sandbox/sandbox-web.d.ts.map +1 -0
- package/dist/lib/sandbox/sandbox-web.js +211 -0
- package/dist/lib/sandbox/sandbox-web.js.map +1 -0
- package/dist/lib/sandbox/types.d.ts +21 -0
- package/dist/lib/sandbox/types.d.ts.map +1 -0
- package/dist/lib/sandbox/types.js +7 -0
- package/dist/lib/sandbox/types.js.map +1 -0
- package/dist/lib/sandbox/vm-executor.d.ts +12 -0
- package/dist/lib/sandbox/vm-executor.d.ts.map +1 -0
- package/dist/lib/sandbox/vm-executor.js +101 -0
- package/dist/lib/sandbox/vm-executor.js.map +1 -0
- package/dist/lib/skills/index.d.ts +4 -0
- package/dist/lib/skills/index.d.ts.map +1 -0
- package/dist/lib/skills/index.js +3 -0
- package/dist/lib/skills/index.js.map +1 -0
- package/dist/lib/skills/skill-loader.d.ts +19 -0
- package/dist/lib/skills/skill-loader.d.ts.map +1 -0
- package/dist/lib/skills/skill-loader.js +105 -0
- package/dist/lib/skills/skill-loader.js.map +1 -0
- package/dist/lib/skills/types.d.ts +17 -0
- package/dist/lib/skills/types.d.ts.map +1 -0
- package/dist/lib/skills/types.js +6 -0
- package/dist/lib/skills/types.js.map +1 -0
- package/dist/lib/tasks/index.d.ts +4 -0
- package/dist/lib/tasks/index.d.ts.map +1 -0
- package/dist/lib/tasks/index.js +4 -0
- package/dist/lib/tasks/index.js.map +1 -0
- package/dist/lib/tasks/task-manager.d.ts +25 -0
- package/dist/lib/tasks/task-manager.d.ts.map +1 -0
- package/dist/lib/tasks/task-manager.js +189 -0
- package/dist/lib/tasks/task-manager.js.map +1 -0
- package/dist/lib/tasks/task-store.d.ts +20 -0
- package/dist/lib/tasks/task-store.d.ts.map +1 -0
- package/dist/lib/tasks/task-store.js +102 -0
- package/dist/lib/tasks/task-store.js.map +1 -0
- package/dist/lib/tasks/types.d.ts +38 -0
- package/dist/lib/tasks/types.d.ts.map +1 -0
- package/dist/lib/tasks/types.js +2 -0
- package/dist/lib/tasks/types.js.map +1 -0
- package/dist/lib/templates/resource-templates.d.ts +6 -0
- package/dist/lib/templates/resource-templates.d.ts.map +1 -0
- package/dist/lib/templates/resource-templates.js +146 -0
- package/dist/lib/templates/resource-templates.js.map +1 -0
- package/dist/lib/tools/agent-tool-wrapper.d.ts +5 -5
- package/dist/lib/tools/agent-tool-wrapper.js +3 -3
- package/dist/lib/tools/agent-tool-wrapper.js.map +1 -1
- package/dist/lib/tools/built-in/ask-user.tool.d.ts +1 -1
- package/dist/lib/tools/built-in/ask-user.tool.d.ts.map +1 -1
- package/dist/lib/tools/built-in/ask-user.tool.js +2 -2
- package/dist/lib/tools/built-in/ask-user.tool.js.map +1 -1
- package/dist/lib/tools/built-in/index.d.ts +9 -2
- package/dist/lib/tools/built-in/index.d.ts.map +1 -1
- package/dist/lib/tools/built-in/index.js +9 -2
- package/dist/lib/tools/built-in/index.js.map +1 -1
- package/dist/lib/tools/built-in/knowledge-entity-lookup.tool.d.ts +9 -0
- package/dist/lib/tools/built-in/knowledge-entity-lookup.tool.d.ts.map +1 -0
- package/dist/lib/tools/built-in/knowledge-entity-lookup.tool.js +102 -0
- package/dist/lib/tools/built-in/knowledge-entity-lookup.tool.js.map +1 -0
- package/dist/lib/tools/built-in/knowledge-graph-schema.tool.d.ts +9 -0
- package/dist/lib/tools/built-in/knowledge-graph-schema.tool.d.ts.map +1 -0
- package/dist/lib/tools/built-in/knowledge-graph-schema.tool.js +96 -0
- package/dist/lib/tools/built-in/knowledge-graph-schema.tool.js.map +1 -0
- package/dist/lib/tools/built-in/knowledge-search.tool.d.ts +2 -2
- package/dist/lib/tools/built-in/knowledge-search.tool.d.ts.map +1 -1
- package/dist/lib/tools/built-in/knowledge-search.tool.js +1 -1
- package/dist/lib/tools/built-in/knowledge-search.tool.js.map +1 -1
- package/dist/lib/tools/built-in/knowledge-sql.tool.d.ts +7 -0
- package/dist/lib/tools/built-in/knowledge-sql.tool.d.ts.map +1 -0
- package/dist/lib/tools/built-in/knowledge-sql.tool.js +127 -0
- package/dist/lib/tools/built-in/knowledge-sql.tool.js.map +1 -0
- package/dist/lib/tools/built-in/knowledge-tools-factory.d.ts +17 -0
- package/dist/lib/tools/built-in/knowledge-tools-factory.d.ts.map +1 -0
- package/dist/lib/tools/built-in/knowledge-tools-factory.js +54 -0
- package/dist/lib/tools/built-in/knowledge-tools-factory.js.map +1 -0
- package/dist/lib/tools/built-in/knowledge-traverse.tool.d.ts +9 -0
- package/dist/lib/tools/built-in/knowledge-traverse.tool.d.ts.map +1 -0
- package/dist/lib/tools/built-in/knowledge-traverse.tool.js +91 -0
- package/dist/lib/tools/built-in/knowledge-traverse.tool.js.map +1 -0
- package/dist/lib/tools/built-in/memory-save.tool.d.ts +4 -0
- package/dist/lib/tools/built-in/memory-save.tool.d.ts.map +1 -0
- package/dist/lib/tools/built-in/memory-save.tool.js +21 -0
- package/dist/lib/tools/built-in/memory-save.tool.js.map +1 -0
- package/dist/lib/tools/built-in/query-validators.d.ts +18 -0
- package/dist/lib/tools/built-in/query-validators.d.ts.map +1 -0
- package/dist/lib/tools/built-in/query-validators.js +87 -0
- package/dist/lib/tools/built-in/query-validators.js.map +1 -0
- package/dist/lib/tools/index.d.ts +4 -4
- package/dist/lib/tools/index.js +4 -4
- package/dist/lib/tools/tool-discovery.d.ts +11 -33
- package/dist/lib/tools/tool-discovery.d.ts.map +1 -1
- package/dist/lib/tools/tool-discovery.js +16 -43
- package/dist/lib/tools/tool-discovery.js.map +1 -1
- package/dist/lib/tools/tool-registry.d.ts +10 -18
- package/dist/lib/tools/tool-registry.d.ts.map +1 -1
- package/dist/lib/tools/tool-registry.js +37 -23
- package/dist/lib/tools/tool-registry.js.map +1 -1
- package/dist/lib/tools/workspace/workspace-tools.d.ts +60 -0
- package/dist/lib/tools/workspace/workspace-tools.d.ts.map +1 -0
- package/dist/lib/tools/workspace/workspace-tools.js +140 -0
- package/dist/lib/tools/workspace/workspace-tools.js.map +1 -0
- package/dist/lib/triggers/cron-trigger.d.ts +12 -0
- package/dist/lib/triggers/cron-trigger.d.ts.map +1 -0
- package/dist/lib/triggers/cron-trigger.js +45 -0
- package/dist/lib/triggers/cron-trigger.js.map +1 -0
- package/dist/lib/triggers/index.d.ts +6 -0
- package/dist/lib/triggers/index.d.ts.map +1 -0
- package/dist/lib/triggers/index.js +5 -0
- package/dist/lib/triggers/index.js.map +1 -0
- package/dist/lib/triggers/trigger-manager.d.ts +11 -0
- package/dist/lib/triggers/trigger-manager.d.ts.map +1 -0
- package/dist/lib/triggers/trigger-manager.js +51 -0
- package/dist/lib/triggers/trigger-manager.js.map +1 -0
- package/dist/lib/triggers/types.d.ts +57 -0
- package/dist/lib/triggers/types.d.ts.map +1 -0
- package/dist/lib/triggers/types.js +16 -0
- package/dist/lib/triggers/types.js.map +1 -0
- package/dist/lib/triggers/webhook-trigger.d.ts +12 -0
- package/dist/lib/triggers/webhook-trigger.d.ts.map +1 -0
- package/dist/lib/triggers/webhook-trigger.js +34 -0
- package/dist/lib/triggers/webhook-trigger.js.map +1 -0
- package/dist/lib/types/llm-types.d.ts +54 -0
- package/dist/lib/types/llm-types.d.ts.map +1 -0
- package/dist/lib/types/llm-types.js +22 -0
- package/dist/lib/types/llm-types.js.map +1 -0
- package/dist/lib/types/text-splitters.d.ts +27 -0
- package/dist/lib/types/text-splitters.d.ts.map +1 -0
- package/dist/lib/types/text-splitters.js +114 -0
- package/dist/lib/types/text-splitters.js.map +1 -0
- package/dist/lib/types/tool-factory.d.ts +9 -0
- package/dist/lib/types/tool-factory.d.ts.map +1 -0
- package/dist/lib/types/tool-factory.js +12 -0
- package/dist/lib/types/tool-factory.js.map +1 -0
- package/dist/lib/utils/file-utils.d.ts +11 -0
- package/dist/lib/utils/file-utils.d.ts.map +1 -0
- package/dist/lib/utils/file-utils.js +86 -0
- package/dist/lib/utils/file-utils.js.map +1 -0
- package/dist/lib/workflows/index.d.ts +6 -6
- package/dist/lib/workflows/index.d.ts.map +1 -1
- package/dist/lib/workflows/index.js +5 -5
- package/dist/lib/workflows/index.js.map +1 -1
- package/dist/lib/workflows/interrupt-manager.d.ts +1 -1
- package/dist/lib/workflows/interrupt-manager.js +1 -1
- package/dist/lib/workflows/react-workflow-executor.d.ts +26 -0
- package/dist/lib/workflows/react-workflow-executor.d.ts.map +1 -0
- package/dist/lib/workflows/react-workflow-executor.js +333 -0
- package/dist/lib/workflows/react-workflow-executor.js.map +1 -0
- package/dist/lib/workflows/types.d.ts +96 -97
- package/dist/lib/workflows/types.d.ts.map +1 -1
- package/dist/lib/workflows/types.js +5 -5
- package/dist/lib/workflows/types.js.map +1 -1
- package/dist/lib/workflows/workflow-executor.d.ts +3 -3
- package/dist/lib/workflows/workflow-executor.js +2 -2
- package/dist/lib/workflows/workflow-executor.js.map +1 -1
- package/dist/lib/workflows/workflow-loader.d.ts +1 -1
- package/dist/lib/workflows/workflow-loader.d.ts.map +1 -1
- package/dist/lib/workflows/workflow-loader.js +10 -3
- package/dist/lib/workflows/workflow-loader.js.map +1 -1
- package/dist/public/index.html +32 -0
- package/dist/public/src/components/AgentsView.js +298 -44
- package/dist/public/src/components/AppRoot.js +124 -4
- package/dist/public/src/components/GraphView.js +336 -0
- package/dist/public/src/components/IdeView.js +462 -7
- package/dist/public/src/components/KnowledgeView.js +443 -66
- package/dist/public/src/components/MonitorView.js +388 -0
- package/dist/public/src/components/NavBar.js +3 -0
- package/dist/public/src/components/SkillsView.js +137 -0
- package/dist/public/src/components/WorkflowsView.js +287 -114
- package/dist/public/src/services/ApiService.js +165 -25
- package/dist/public/src/store.js +3 -2
- package/dist/public/src/utils/markdown.js +1 -0
- package/dist/src/cli/commands/init.js +2 -2
- package/dist/src/cli/commands/init.js.map +1 -1
- package/dist/src/cli/commands/start.d.ts +0 -1
- package/dist/src/cli/commands/start.d.ts.map +1 -1
- package/dist/src/cli/commands/start.js +23 -14
- package/dist/src/cli/commands/start.js.map +1 -1
- package/dist/src/cli/index.js +3 -3
- package/dist/src/index.d.ts +1 -1
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/index.js +23 -9
- package/dist/src/index.js.map +1 -1
- package/dist/src/middleware/auth.d.ts +3 -0
- package/dist/src/middleware/auth.d.ts.map +1 -0
- package/dist/src/middleware/auth.js +107 -0
- package/dist/src/middleware/auth.js.map +1 -0
- package/dist/src/routes/agents.route.d.ts.map +1 -1
- package/dist/src/routes/agents.route.js +31 -2
- package/dist/src/routes/agents.route.js.map +1 -1
- package/dist/src/routes/files.route.d.ts.map +1 -1
- package/dist/src/routes/files.route.js +128 -87
- package/dist/src/routes/files.route.js.map +1 -1
- package/dist/src/routes/functions.route.js +1 -1
- package/dist/src/routes/graph.route.d.ts +3 -0
- package/dist/src/routes/graph.route.d.ts.map +1 -0
- package/dist/src/routes/graph.route.js +173 -0
- package/dist/src/routes/graph.route.js.map +1 -0
- package/dist/src/routes/knowledge.route.d.ts.map +1 -1
- package/dist/src/routes/knowledge.route.js +172 -69
- package/dist/src/routes/knowledge.route.js.map +1 -1
- package/dist/src/routes/llm.route.d.ts.map +1 -1
- package/dist/src/routes/llm.route.js +48 -7
- package/dist/src/routes/llm.route.js.map +1 -1
- package/dist/src/routes/mcp.route.js +1 -1
- package/dist/src/routes/skills.route.d.ts +3 -0
- package/dist/src/routes/skills.route.d.ts.map +1 -0
- package/dist/src/routes/skills.route.js +23 -0
- package/dist/src/routes/skills.route.js.map +1 -0
- package/dist/src/routes/tasks.route.d.ts +3 -0
- package/dist/src/routes/tasks.route.d.ts.map +1 -0
- package/dist/src/routes/tasks.route.js +135 -0
- package/dist/src/routes/tasks.route.js.map +1 -0
- package/dist/src/routes/workflows.route.d.ts.map +1 -1
- package/dist/src/routes/workflows.route.js +9 -1
- package/dist/src/routes/workflows.route.js.map +1 -1
- package/dist/src/server.d.ts +1 -1
- package/dist/src/server.d.ts.map +1 -1
- package/dist/src/server.js +21 -8
- package/dist/src/server.js.map +1 -1
- package/dist/templates/.env.example +4 -19
- package/dist/templates/README.md +18 -10
- package/dist/templates/agents/architect.agent.yaml +49 -0
- package/dist/templates/agents/chatbot.agent.yaml +58 -0
- package/dist/templates/agents/knowledge-broker.agent.yaml +39 -0
- package/dist/templates/agents/sandbox.agent.yaml +56 -0
- package/dist/templates/knowledge/example.knowledge.yaml +0 -3
- package/dist/templates/knowledge/org-chart/personnel.csv +21 -0
- package/dist/templates/knowledge/org-chart.knowledge.yaml +49 -0
- package/dist/templates/knowledge/pet-store/pet-store.db +0 -0
- package/dist/templates/knowledge/pet-store.knowledge.yaml +78 -0
- package/dist/templates/knowledge/web-docs.knowledge.yaml +20 -0
- package/dist/templates/llm.json +2 -2
- package/dist/templates/mcp.json +3 -0
- package/dist/templates/skills/orcha-builder/SKILL.md +339 -0
- package/dist/templates/skills/sandbox/SKILL.md +63 -0
- package/dist/templates/workflows/example.workflow.yaml +25 -30
- package/dist/templates/workflows/react-example.workflow.yaml +53 -0
- package/package.json +17 -20
- package/dist/lib/knowledge/graph-rag/community-detector.d.ts +0 -16
- package/dist/lib/knowledge/graph-rag/community-detector.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/community-detector.js +0 -81
- package/dist/lib/knowledge/graph-rag/community-detector.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/community-summarizer.d.ts +0 -17
- package/dist/lib/knowledge/graph-rag/community-summarizer.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/community-summarizer.js +0 -87
- package/dist/lib/knowledge/graph-rag/community-summarizer.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/entity-extractor.d.ts +0 -36
- package/dist/lib/knowledge/graph-rag/entity-extractor.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/entity-extractor.js +0 -192
- package/dist/lib/knowledge/graph-rag/entity-extractor.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/extraction-cache.d.ts +0 -30
- package/dist/lib/knowledge/graph-rag/extraction-cache.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/extraction-cache.js +0 -88
- package/dist/lib/knowledge/graph-rag/extraction-cache.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/global-search.d.ts +0 -19
- package/dist/lib/knowledge/graph-rag/global-search.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/global-search.js +0 -96
- package/dist/lib/knowledge/graph-rag/global-search.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/graph-rag-factory.d.ts +0 -24
- package/dist/lib/knowledge/graph-rag/graph-rag-factory.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/graph-rag-factory.js +0 -239
- package/dist/lib/knowledge/graph-rag/graph-rag-factory.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/index.d.ts +0 -14
- package/dist/lib/knowledge/graph-rag/index.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/index.js +0 -12
- package/dist/lib/knowledge/graph-rag/index.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/local-search.d.ts +0 -20
- package/dist/lib/knowledge/graph-rag/local-search.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/local-search.js +0 -110
- package/dist/lib/knowledge/graph-rag/local-search.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/memory-graph-store.d.ts +0 -31
- package/dist/lib/knowledge/graph-rag/memory-graph-store.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/memory-graph-store.js +0 -165
- package/dist/lib/knowledge/graph-rag/memory-graph-store.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/neo4j-graph-store.d.ts +0 -38
- package/dist/lib/knowledge/graph-rag/neo4j-graph-store.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/neo4j-graph-store.js +0 -190
- package/dist/lib/knowledge/graph-rag/neo4j-graph-store.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/search-mode-detector.d.ts +0 -11
- package/dist/lib/knowledge/graph-rag/search-mode-detector.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/search-mode-detector.js +0 -50
- package/dist/lib/knowledge/graph-rag/search-mode-detector.js.map +0 -1
- package/dist/lib/knowledge/graph-rag/types.d.ts +0 -368
- package/dist/lib/knowledge/graph-rag/types.d.ts.map +0 -1
- package/dist/lib/knowledge/graph-rag/types.js +0 -48
- package/dist/lib/knowledge/graph-rag/types.js.map +0 -1
- package/dist/lib/knowledge/knowledge-store-factory.d.ts +0 -16
- package/dist/lib/knowledge/knowledge-store-factory.d.ts.map +0 -1
- package/dist/lib/knowledge/knowledge-store-factory.js +0 -376
- package/dist/lib/knowledge/knowledge-store-factory.js.map +0 -1
- package/dist/lib/knowledge/knowledge-store-manager.d.ts +0 -18
- package/dist/lib/knowledge/knowledge-store-manager.d.ts.map +0 -1
- package/dist/lib/knowledge/knowledge-store-manager.js +0 -98
- package/dist/lib/knowledge/knowledge-store-manager.js.map +0 -1
- package/dist/lib/knowledge/loaders/s3-loader.d.ts +0 -17
- package/dist/lib/knowledge/loaders/s3-loader.d.ts.map +0 -1
- package/dist/lib/knowledge/loaders/s3-loader.js +0 -185
- package/dist/lib/knowledge/loaders/s3-loader.js.map +0 -1
- package/dist/lib/workflows/langgraph-executor.d.ts +0 -51
- package/dist/lib/workflows/langgraph-executor.d.ts.map +0 -1
- package/dist/lib/workflows/langgraph-executor.js +0 -297
- package/dist/lib/workflows/langgraph-executor.js.map +0 -1
- package/dist/templates/agents/call-center-analyst-simple.agent.yaml +0 -36
- package/dist/templates/agents/math.agent.yaml +0 -28
- package/dist/templates/agents/sentiment-structured.agent.yaml +0 -42
- package/dist/templates/llm.md +0 -1195
- package/dist/templates/workflows/langgraph-example.workflow.yaml +0 -84
|
@@ -0,0 +1,241 @@
|
|
|
1
|
+
import { GoogleGenerativeAI, SchemaType } from '@google/generative-ai';
|
|
2
|
+
import { zodToJsonSchema } from 'zod-to-json-schema';
|
|
3
|
+
export class GeminiChatModel {
|
|
4
|
+
genAI;
|
|
5
|
+
modelName;
|
|
6
|
+
temperature;
|
|
7
|
+
maxTokens;
|
|
8
|
+
boundTools;
|
|
9
|
+
structuredSchema;
|
|
10
|
+
constructor(options) {
|
|
11
|
+
this.genAI = new GoogleGenerativeAI(options.apiKey ?? process.env.GOOGLE_API_KEY ?? '');
|
|
12
|
+
this.modelName = options.modelName;
|
|
13
|
+
this.temperature = options.temperature;
|
|
14
|
+
this.maxTokens = options.maxTokens;
|
|
15
|
+
}
|
|
16
|
+
extractSystemAndContents(messages) {
|
|
17
|
+
let systemInstruction;
|
|
18
|
+
const contents = [];
|
|
19
|
+
for (const msg of messages) {
|
|
20
|
+
if (msg.role === 'system') {
|
|
21
|
+
systemInstruction = systemInstruction
|
|
22
|
+
? `${systemInstruction}\n\n${msg.content}`
|
|
23
|
+
: msg.content;
|
|
24
|
+
continue;
|
|
25
|
+
}
|
|
26
|
+
if (msg.role === 'human') {
|
|
27
|
+
contents.push({ role: 'user', parts: [{ text: msg.content }] });
|
|
28
|
+
}
|
|
29
|
+
else if (msg.role === 'ai') {
|
|
30
|
+
const parts = [];
|
|
31
|
+
if (msg.content) {
|
|
32
|
+
parts.push({ text: msg.content });
|
|
33
|
+
}
|
|
34
|
+
if (msg.tool_calls?.length) {
|
|
35
|
+
for (const tc of msg.tool_calls) {
|
|
36
|
+
parts.push({
|
|
37
|
+
functionCall: { name: tc.name, args: tc.args },
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
contents.push({ role: 'model', parts });
|
|
42
|
+
}
|
|
43
|
+
else if (msg.role === 'tool') {
|
|
44
|
+
contents.push({
|
|
45
|
+
role: 'function',
|
|
46
|
+
parts: [
|
|
47
|
+
{
|
|
48
|
+
functionResponse: {
|
|
49
|
+
name: msg.name,
|
|
50
|
+
response: { result: msg.content },
|
|
51
|
+
},
|
|
52
|
+
},
|
|
53
|
+
],
|
|
54
|
+
});
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
return { systemInstruction, contents };
|
|
58
|
+
}
|
|
59
|
+
toGeminiTools() {
|
|
60
|
+
if (!this.boundTools?.length)
|
|
61
|
+
return undefined;
|
|
62
|
+
const declarations = this.boundTools.map((t) => {
|
|
63
|
+
const jsonSchema = zodToJsonSchema(t.schema);
|
|
64
|
+
return {
|
|
65
|
+
name: t.name,
|
|
66
|
+
description: t.description,
|
|
67
|
+
parameters: this.convertToGeminiSchema(jsonSchema),
|
|
68
|
+
};
|
|
69
|
+
});
|
|
70
|
+
return [{ functionDeclarations: declarations }];
|
|
71
|
+
}
|
|
72
|
+
convertToGeminiSchema(jsonSchema) {
|
|
73
|
+
const schemaType = jsonSchema.type;
|
|
74
|
+
const properties = jsonSchema.properties;
|
|
75
|
+
const required = jsonSchema.required;
|
|
76
|
+
const result = {
|
|
77
|
+
type: this.mapSchemaType(schemaType),
|
|
78
|
+
properties: {},
|
|
79
|
+
};
|
|
80
|
+
if (properties) {
|
|
81
|
+
const geminiProps = {};
|
|
82
|
+
for (const [key, prop] of Object.entries(properties)) {
|
|
83
|
+
geminiProps[key] = this.convertPropertyToGemini(prop);
|
|
84
|
+
}
|
|
85
|
+
result.properties = geminiProps;
|
|
86
|
+
}
|
|
87
|
+
if (required) {
|
|
88
|
+
result.required = required;
|
|
89
|
+
}
|
|
90
|
+
return result;
|
|
91
|
+
}
|
|
92
|
+
convertPropertyToGemini(prop) {
|
|
93
|
+
const type = prop.type;
|
|
94
|
+
const result = {
|
|
95
|
+
type: this.mapSchemaType(type),
|
|
96
|
+
};
|
|
97
|
+
if (prop.description) {
|
|
98
|
+
result.description = prop.description;
|
|
99
|
+
}
|
|
100
|
+
if (type === 'array') {
|
|
101
|
+
const items = prop.items;
|
|
102
|
+
result.items = items
|
|
103
|
+
? this.convertPropertyToGemini(items)
|
|
104
|
+
: { type: SchemaType.STRING };
|
|
105
|
+
}
|
|
106
|
+
if (type === 'object' && prop.properties) {
|
|
107
|
+
const nested = prop.properties;
|
|
108
|
+
const geminiProps = {};
|
|
109
|
+
for (const [key, nestedProp] of Object.entries(nested)) {
|
|
110
|
+
geminiProps[key] = this.convertPropertyToGemini(nestedProp);
|
|
111
|
+
}
|
|
112
|
+
result.properties = geminiProps;
|
|
113
|
+
if (prop.required) {
|
|
114
|
+
result.required = prop.required;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
return result;
|
|
118
|
+
}
|
|
119
|
+
mapSchemaType(type) {
|
|
120
|
+
switch (type) {
|
|
121
|
+
case 'string': return SchemaType.STRING;
|
|
122
|
+
case 'number': return SchemaType.NUMBER;
|
|
123
|
+
case 'integer': return SchemaType.INTEGER;
|
|
124
|
+
case 'boolean': return SchemaType.BOOLEAN;
|
|
125
|
+
case 'array': return SchemaType.ARRAY;
|
|
126
|
+
case 'object': return SchemaType.OBJECT;
|
|
127
|
+
default: return SchemaType.STRING;
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
async invoke(messages) {
|
|
131
|
+
const { systemInstruction, contents } = this.extractSystemAndContents(messages);
|
|
132
|
+
const model = this.genAI.getGenerativeModel({
|
|
133
|
+
model: this.modelName,
|
|
134
|
+
...(systemInstruction ? { systemInstruction } : {}),
|
|
135
|
+
generationConfig: {
|
|
136
|
+
...(this.temperature !== undefined ? { temperature: this.temperature } : {}),
|
|
137
|
+
...(this.maxTokens ? { maxOutputTokens: this.maxTokens } : {}),
|
|
138
|
+
},
|
|
139
|
+
...(this.toGeminiTools() ? { tools: this.toGeminiTools() } : {}),
|
|
140
|
+
});
|
|
141
|
+
const result = await model.generateContent({ contents });
|
|
142
|
+
const response = result.response;
|
|
143
|
+
const candidate = response.candidates?.[0];
|
|
144
|
+
let content = '';
|
|
145
|
+
const toolCalls = [];
|
|
146
|
+
if (candidate?.content?.parts) {
|
|
147
|
+
for (const part of candidate.content.parts) {
|
|
148
|
+
if (part.text) {
|
|
149
|
+
content += part.text;
|
|
150
|
+
}
|
|
151
|
+
if (part.functionCall) {
|
|
152
|
+
toolCalls.push({
|
|
153
|
+
id: `call_${Math.random().toString(36).substring(7)}`,
|
|
154
|
+
name: part.functionCall.name,
|
|
155
|
+
args: (part.functionCall.args ?? {}),
|
|
156
|
+
});
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
const usage = response.usageMetadata;
|
|
161
|
+
return {
|
|
162
|
+
content,
|
|
163
|
+
tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
164
|
+
usage_metadata: usage
|
|
165
|
+
? {
|
|
166
|
+
input_tokens: usage.promptTokenCount ?? 0,
|
|
167
|
+
output_tokens: usage.candidatesTokenCount ?? 0,
|
|
168
|
+
total_tokens: usage.totalTokenCount ?? 0,
|
|
169
|
+
}
|
|
170
|
+
: undefined,
|
|
171
|
+
};
|
|
172
|
+
}
|
|
173
|
+
async *stream(messages, _options) {
|
|
174
|
+
const { systemInstruction, contents } = this.extractSystemAndContents(messages);
|
|
175
|
+
const model = this.genAI.getGenerativeModel({
|
|
176
|
+
model: this.modelName,
|
|
177
|
+
...(systemInstruction ? { systemInstruction } : {}),
|
|
178
|
+
generationConfig: {
|
|
179
|
+
...(this.temperature !== undefined ? { temperature: this.temperature } : {}),
|
|
180
|
+
...(this.maxTokens ? { maxOutputTokens: this.maxTokens } : {}),
|
|
181
|
+
},
|
|
182
|
+
...(this.toGeminiTools() ? { tools: this.toGeminiTools() } : {}),
|
|
183
|
+
});
|
|
184
|
+
const result = await model.generateContentStream({ contents });
|
|
185
|
+
for await (const chunk of result.stream) {
|
|
186
|
+
const candidate = chunk.candidates?.[0];
|
|
187
|
+
let content = '';
|
|
188
|
+
const toolCalls = [];
|
|
189
|
+
if (candidate?.content?.parts) {
|
|
190
|
+
for (const part of candidate.content.parts) {
|
|
191
|
+
if (part.text) {
|
|
192
|
+
content += part.text;
|
|
193
|
+
}
|
|
194
|
+
if (part.functionCall) {
|
|
195
|
+
toolCalls.push({
|
|
196
|
+
id: `call_${Math.random().toString(36).substring(7)}`,
|
|
197
|
+
name: part.functionCall.name,
|
|
198
|
+
args: (part.functionCall.args ?? {}),
|
|
199
|
+
});
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
yield {
|
|
204
|
+
content,
|
|
205
|
+
tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
206
|
+
usage_metadata: chunk.usageMetadata
|
|
207
|
+
? {
|
|
208
|
+
input_tokens: chunk.usageMetadata.promptTokenCount ?? 0,
|
|
209
|
+
output_tokens: chunk.usageMetadata.candidatesTokenCount ?? 0,
|
|
210
|
+
total_tokens: chunk.usageMetadata.totalTokenCount ?? 0,
|
|
211
|
+
}
|
|
212
|
+
: undefined,
|
|
213
|
+
};
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
bindTools(tools) {
|
|
217
|
+
const bound = new GeminiChatModel({
|
|
218
|
+
apiKey: '',
|
|
219
|
+
modelName: this.modelName,
|
|
220
|
+
temperature: this.temperature,
|
|
221
|
+
maxTokens: this.maxTokens,
|
|
222
|
+
});
|
|
223
|
+
bound.genAI = this.genAI;
|
|
224
|
+
bound.boundTools = tools;
|
|
225
|
+
bound.structuredSchema = this.structuredSchema;
|
|
226
|
+
return bound;
|
|
227
|
+
}
|
|
228
|
+
withStructuredOutput(schema) {
|
|
229
|
+
const wrapped = new GeminiChatModel({
|
|
230
|
+
apiKey: '',
|
|
231
|
+
modelName: this.modelName,
|
|
232
|
+
temperature: this.temperature,
|
|
233
|
+
maxTokens: this.maxTokens,
|
|
234
|
+
});
|
|
235
|
+
wrapped.genAI = this.genAI;
|
|
236
|
+
wrapped.structuredSchema = schema;
|
|
237
|
+
wrapped.boundTools = this.boundTools;
|
|
238
|
+
return wrapped;
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
//# sourceMappingURL=gemini-chat-model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-chat-model.js","sourceRoot":"","sources":["../../../../lib/llm/providers/gemini-chat-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAqF,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAC1J,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAgBrD,MAAM,OAAO,eAAe;IAClB,KAAK,CAAqB;IAC1B,SAAS,CAAS;IAClB,WAAW,CAAU;IACrB,SAAS,CAAU;IACnB,UAAU,CAAoB;IAC9B,gBAAgB,CAA2B;IAEnD,YAAY,OAA+B;QACzC,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;QACxF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACrC,CAAC;IAEO,wBAAwB,CAC9B,QAAuB;QAEvB,IAAI,iBAAqC,CAAC;QAC1C,MAAM,QAAQ,GAAc,EAAE,CAAC;QAE/B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,iBAAiB,GAAG,iBAAiB;oBACnC,CAAC,CAAC,GAAG,iBAAiB,OAAO,GAAG,CAAC,OAAO,EAAE;oBAC1C,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;gBAChB,SAAS;YACX,CAAC;YAED,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACzB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAAW,EAAE,CAAC;gBACzB,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBAChB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBACpC,CAAC;gBACD,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;oBAC3B,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;wBAChC,KAAK,CAAC,IAAI,CAAC;4BACT,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE;yBAC/C,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1C,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC/B,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,UAAU;oBAChB,KAAK,EAAE;wBACL;4BACE,gBAAgB,EAAE;gCAChB,IAAI,EAAE,GAAG,CAAC,IAAK;gCACf,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE;6BAClC;yBACF;qBACF;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC;IACzC,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAC/C,MAAM,YAAY,GAA0B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACpE,MAAM,UAAU,GAAG,eAAe,CAAC,CAAC,CAAC,MAAM,CAA4B,CAAC;YACxE,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC;aACnD,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,EAAE,oBAAoB,EAAE,YAAY,EAAE,CAAC,CAAC;IAClD,CAAC;IAEO,qBAAqB,CAAC,UAAmC;QAC/D,MAAM,UAAU,GAAG,UAAU,CAAC,IAAc,CAAC;QAC7C,MAAM,UAAU,GAAG,UAAU,CAAC,UAAiE,CAAC;QAChG,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAgC,CAAC;QAE7D,MAAM,MAAM,GAA8B;YACxC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC;YACpC,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,WAAW,GAAwB,EAAE,CAAC;YAC5C,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;gBACrD,WAAW,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;YACxD,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,WAAW,CAAC;QAClC,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC7B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,uBAAuB,CAAC,IAA6B;QAC3D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAc,CAAC;QACjC,MAAM,MAAM,GAA4B;YACtC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;SAC/B,CAAC;QAEF,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACxC,CAAC;QAED,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,IAAI,CAAC,KAA4C,CAAC;YAChE,MAAM,CAAC,KAAK,GAAG,KAAK;gBAClB,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC;gBACrC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC;QAClC,CAAC;QAED,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAqD,CAAC;YAC1E,MAAM,WAAW,GAA4B,EAAE,CAAC;YAChD,KAAK,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvD,WAAW,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;YAC9D,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,WAAW,CAAC;YAChC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;YACxC,KAAK,QAAQ,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;YACxC,KAAK,SAAS,CAAC,CAAC,OAAO,UAAU,CAAC,OAAO,CAAC;YAC1C,KAAK,SAAS,CAAC,CAAC,OAAO,UAAU,CAAC,OAAO,CAAC;YAC1C,KAAK,OAAO,CAAC,CAAC,OAAO,UAAU,CAAC,KAAK,CAAC;YACtC,KAAK,QAAQ,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;YACxC,OAAO,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;QACpC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAuB;QAClC,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,gBAAgB,EAAE;gBAChB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC/D;YACD,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACjE,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3C,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACd,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC;gBACvB,CAAC;gBACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,SAAS,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,QAAQ,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;wBACrD,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;wBAC5B,IAAI,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAA4B;qBAChE,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC;QAErC,OAAO;YACL,OAAO;YACP,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACxD,cAAc,EAAE,KAAK;gBACnB,CAAC,CAAC;oBACE,YAAY,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;oBACzC,aAAa,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC;oBAC9C,YAAY,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;iBACzC;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,QAAmC;QAEnC,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,gBAAgB,EAAE;gBAChB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC/D;YACD,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACjE,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE/D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;YACxC,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,MAAM,SAAS,GAAe,EAAE,CAAC;YAEjC,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC3C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACd,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC;oBACvB,CAAC;oBACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;wBACtB,SAAS,CAAC,IAAI,CAAC;4BACb,EAAE,EAAE,QAAQ,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;4BACrD,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;4BAC5B,IAAI,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAA4B;yBAChE,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM;gBACJ,OAAO;gBACP,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACxD,cAAc,EAAE,KAAK,CAAC,aAAa;oBACjC,CAAC,CAAC;wBACE,YAAY,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;wBACvD,aAAa,EAAE,KAAK,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;wBAC5D,YAAY,EAAE,KAAK,CAAC,aAAa,CAAC,eAAe,IAAI,CAAC;qBACvD;oBACH,CAAC,CAAC,SAAS;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAuB;QAC/B,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC;YAChC,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC;QACzB,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC/C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oBAAoB,CAAC,MAA+B;QAClD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC;YAClC,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAC3B,OAAO,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAClC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACrC,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { Embeddings } from '../../types/llm-types.ts';
|
|
2
|
+
interface GeminiEmbeddingsOptions {
|
|
3
|
+
apiKey?: string;
|
|
4
|
+
modelName: string;
|
|
5
|
+
}
|
|
6
|
+
export declare class GeminiEmbeddingsProvider implements Embeddings {
|
|
7
|
+
private genAI;
|
|
8
|
+
private modelName;
|
|
9
|
+
constructor(options: GeminiEmbeddingsOptions);
|
|
10
|
+
embedQuery(text: string): Promise<number[]>;
|
|
11
|
+
embedDocuments(texts: string[]): Promise<number[][]>;
|
|
12
|
+
}
|
|
13
|
+
export {};
|
|
14
|
+
//# sourceMappingURL=gemini-embeddings.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-embeddings.d.ts","sourceRoot":"","sources":["../../../../lib/llm/providers/gemini-embeddings.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAE3D,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,wBAAyB,YAAW,UAAU;IACzD,OAAO,CAAC,KAAK,CAAqB;IAClC,OAAO,CAAC,SAAS,CAAS;gBAEd,OAAO,EAAE,uBAAuB;IAKtC,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAM3C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;CAO3D"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { GoogleGenerativeAI } from '@google/generative-ai';
|
|
2
|
+
export class GeminiEmbeddingsProvider {
|
|
3
|
+
genAI;
|
|
4
|
+
modelName;
|
|
5
|
+
constructor(options) {
|
|
6
|
+
this.genAI = new GoogleGenerativeAI(options.apiKey ?? process.env.GOOGLE_API_KEY ?? '');
|
|
7
|
+
this.modelName = options.modelName;
|
|
8
|
+
}
|
|
9
|
+
async embedQuery(text) {
|
|
10
|
+
const model = this.genAI.getGenerativeModel({ model: this.modelName });
|
|
11
|
+
const result = await model.embedContent(text);
|
|
12
|
+
return result.embedding.values;
|
|
13
|
+
}
|
|
14
|
+
async embedDocuments(texts) {
|
|
15
|
+
const model = this.genAI.getGenerativeModel({ model: this.modelName });
|
|
16
|
+
const results = await Promise.all(texts.map((text) => model.embedContent(text)));
|
|
17
|
+
return results.map((r) => r.embedding.values);
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=gemini-embeddings.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-embeddings.js","sourceRoot":"","sources":["../../../../lib/llm/providers/gemini-embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAQ3D,MAAM,OAAO,wBAAwB;IAC3B,KAAK,CAAqB;IAC1B,SAAS,CAAS;IAE1B,YAAY,OAAgC;QAC1C,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;QACxF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC9C,OAAO,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAe;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAC9C,CAAC;QACF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;CACF"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import type { ChatModel, ChatModelResponse, BaseMessage, StructuredTool } from '../../types/llm-types.ts';
|
|
2
|
+
interface OpenAIChatModelOptions {
|
|
3
|
+
apiKey?: string;
|
|
4
|
+
modelName: string;
|
|
5
|
+
temperature?: number;
|
|
6
|
+
maxTokens?: number;
|
|
7
|
+
baseURL?: string;
|
|
8
|
+
streamUsage?: boolean;
|
|
9
|
+
}
|
|
10
|
+
export declare class OpenAIChatModel implements ChatModel {
|
|
11
|
+
private client;
|
|
12
|
+
private modelName;
|
|
13
|
+
private temperature?;
|
|
14
|
+
private maxTokens?;
|
|
15
|
+
private streamUsage;
|
|
16
|
+
private boundTools?;
|
|
17
|
+
private structuredSchema?;
|
|
18
|
+
constructor(options: OpenAIChatModelOptions);
|
|
19
|
+
private toOpenAIMessages;
|
|
20
|
+
private toOpenAITools;
|
|
21
|
+
private parseToolCalls;
|
|
22
|
+
invoke(messages: BaseMessage[]): Promise<ChatModelResponse>;
|
|
23
|
+
stream(messages: BaseMessage[], options?: {
|
|
24
|
+
signal?: AbortSignal;
|
|
25
|
+
}): AsyncIterable<ChatModelResponse>;
|
|
26
|
+
bindTools(tools: StructuredTool[]): ChatModel;
|
|
27
|
+
withStructuredOutput(schema: Record<string, unknown>): ChatModel;
|
|
28
|
+
}
|
|
29
|
+
export {};
|
|
30
|
+
//# sourceMappingURL=openai-chat-model.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-chat-model.d.ts","sourceRoot":"","sources":["../../../../lib/llm/providers/openai-chat-model.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,SAAS,EACT,iBAAiB,EACjB,WAAW,EACX,cAAc,EAEf,MAAM,0BAA0B,CAAC;AAElC,UAAU,sBAAsB;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,qBAAa,eAAgB,YAAW,SAAS;IAC/C,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAC,CAAS;IAC3B,OAAO,CAAC,WAAW,CAAU;IAC7B,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,gBAAgB,CAAC,CAA0B;gBAEvC,OAAO,EAAE,sBAAsB;IAW3C,OAAO,CAAC,gBAAgB;IAiCxB,OAAO,CAAC,aAAa;IAYrB,OAAO,CAAC,cAAc;IAchB,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAqC1D,MAAM,CACX,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,aAAa,CAAC,iBAAiB,CAAC;IAgEnC,SAAS,CAAC,KAAK,EAAE,cAAc,EAAE,GAAG,SAAS;IAe7C,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS;CAajE"}
|
|
@@ -0,0 +1,196 @@
|
|
|
1
|
+
import OpenAI from 'openai';
|
|
2
|
+
import { zodToJsonSchema } from 'zod-to-json-schema';
|
|
3
|
+
export class OpenAIChatModel {
|
|
4
|
+
client;
|
|
5
|
+
modelName;
|
|
6
|
+
temperature;
|
|
7
|
+
maxTokens;
|
|
8
|
+
streamUsage;
|
|
9
|
+
boundTools;
|
|
10
|
+
structuredSchema;
|
|
11
|
+
constructor(options) {
|
|
12
|
+
this.client = new OpenAI({
|
|
13
|
+
apiKey: options.apiKey ?? process.env.OPENAI_API_KEY ?? 'not-set',
|
|
14
|
+
...(options.baseURL ? { baseURL: options.baseURL } : {}),
|
|
15
|
+
});
|
|
16
|
+
this.modelName = options.modelName;
|
|
17
|
+
this.temperature = options.temperature;
|
|
18
|
+
this.maxTokens = options.maxTokens;
|
|
19
|
+
this.streamUsage = options.streamUsage ?? true;
|
|
20
|
+
}
|
|
21
|
+
toOpenAIMessages(messages) {
|
|
22
|
+
return messages.map((msg) => {
|
|
23
|
+
switch (msg.role) {
|
|
24
|
+
case 'system':
|
|
25
|
+
return { role: 'system', content: msg.content };
|
|
26
|
+
case 'human':
|
|
27
|
+
return { role: 'user', content: msg.content };
|
|
28
|
+
case 'ai': {
|
|
29
|
+
const aiMsg = {
|
|
30
|
+
role: 'assistant',
|
|
31
|
+
content: msg.content || null,
|
|
32
|
+
};
|
|
33
|
+
if (msg.tool_calls?.length) {
|
|
34
|
+
aiMsg.tool_calls = msg.tool_calls.map((tc) => ({
|
|
35
|
+
id: tc.id,
|
|
36
|
+
type: 'function',
|
|
37
|
+
function: { name: tc.name, arguments: JSON.stringify(tc.args) },
|
|
38
|
+
}));
|
|
39
|
+
}
|
|
40
|
+
return aiMsg;
|
|
41
|
+
}
|
|
42
|
+
case 'tool':
|
|
43
|
+
return {
|
|
44
|
+
role: 'tool',
|
|
45
|
+
tool_call_id: msg.tool_call_id,
|
|
46
|
+
content: msg.content,
|
|
47
|
+
};
|
|
48
|
+
default:
|
|
49
|
+
return { role: 'user', content: msg.content };
|
|
50
|
+
}
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
toOpenAITools() {
|
|
54
|
+
if (!this.boundTools?.length)
|
|
55
|
+
return undefined;
|
|
56
|
+
return this.boundTools.map((t) => ({
|
|
57
|
+
type: 'function',
|
|
58
|
+
function: {
|
|
59
|
+
name: t.name,
|
|
60
|
+
description: t.description,
|
|
61
|
+
parameters: zodToJsonSchema(t.schema),
|
|
62
|
+
},
|
|
63
|
+
}));
|
|
64
|
+
}
|
|
65
|
+
parseToolCalls(choices) {
|
|
66
|
+
const toolCalls = choices[0]?.message?.tool_calls;
|
|
67
|
+
if (!toolCalls?.length)
|
|
68
|
+
return undefined;
|
|
69
|
+
return toolCalls
|
|
70
|
+
.filter((tc) => tc.type === 'function')
|
|
71
|
+
.map((tc) => ({
|
|
72
|
+
id: tc.id,
|
|
73
|
+
name: tc.function.name,
|
|
74
|
+
args: JSON.parse(tc.function.arguments || '{}'),
|
|
75
|
+
}));
|
|
76
|
+
}
|
|
77
|
+
async invoke(messages) {
|
|
78
|
+
const params = {
|
|
79
|
+
model: this.modelName,
|
|
80
|
+
messages: this.toOpenAIMessages(messages),
|
|
81
|
+
...(this.temperature !== undefined ? { temperature: this.temperature } : {}),
|
|
82
|
+
...(this.maxTokens ? { max_tokens: this.maxTokens } : {}),
|
|
83
|
+
...(this.toOpenAITools() ? { tools: this.toOpenAITools() } : {}),
|
|
84
|
+
...(this.structuredSchema
|
|
85
|
+
? {
|
|
86
|
+
response_format: {
|
|
87
|
+
type: 'json_schema',
|
|
88
|
+
json_schema: {
|
|
89
|
+
name: 'structured_output',
|
|
90
|
+
schema: this.structuredSchema,
|
|
91
|
+
strict: true,
|
|
92
|
+
},
|
|
93
|
+
},
|
|
94
|
+
}
|
|
95
|
+
: {}),
|
|
96
|
+
};
|
|
97
|
+
const response = await this.client.chat.completions.create(params);
|
|
98
|
+
const choice = response.choices[0];
|
|
99
|
+
return {
|
|
100
|
+
content: choice.message.content ?? '',
|
|
101
|
+
tool_calls: this.parseToolCalls(response.choices),
|
|
102
|
+
usage_metadata: response.usage
|
|
103
|
+
? {
|
|
104
|
+
input_tokens: response.usage.prompt_tokens,
|
|
105
|
+
output_tokens: response.usage.completion_tokens,
|
|
106
|
+
total_tokens: response.usage.total_tokens,
|
|
107
|
+
}
|
|
108
|
+
: undefined,
|
|
109
|
+
};
|
|
110
|
+
}
|
|
111
|
+
async *stream(messages, options) {
|
|
112
|
+
const params = {
|
|
113
|
+
model: this.modelName,
|
|
114
|
+
messages: this.toOpenAIMessages(messages),
|
|
115
|
+
stream: true,
|
|
116
|
+
stream_options: this.streamUsage ? { include_usage: true } : undefined,
|
|
117
|
+
...(this.temperature !== undefined ? { temperature: this.temperature } : {}),
|
|
118
|
+
...(this.maxTokens ? { max_tokens: this.maxTokens } : {}),
|
|
119
|
+
...(this.toOpenAITools() ? { tools: this.toOpenAITools() } : {}),
|
|
120
|
+
};
|
|
121
|
+
const stream = await this.client.chat.completions.create(params, {
|
|
122
|
+
signal: options?.signal,
|
|
123
|
+
});
|
|
124
|
+
let accumulatedToolCalls = new Map();
|
|
125
|
+
for await (const chunk of stream) {
|
|
126
|
+
const delta = chunk.choices[0]?.delta;
|
|
127
|
+
// Yield content chunks
|
|
128
|
+
if (delta?.content) {
|
|
129
|
+
yield { content: delta.content };
|
|
130
|
+
}
|
|
131
|
+
// Accumulate tool call deltas
|
|
132
|
+
if (delta?.tool_calls) {
|
|
133
|
+
for (const tc of delta.tool_calls) {
|
|
134
|
+
const existing = accumulatedToolCalls.get(tc.index);
|
|
135
|
+
if (existing) {
|
|
136
|
+
existing.args += tc.function?.arguments ?? '';
|
|
137
|
+
}
|
|
138
|
+
else {
|
|
139
|
+
accumulatedToolCalls.set(tc.index, {
|
|
140
|
+
id: tc.id ?? '',
|
|
141
|
+
name: tc.function?.name ?? '',
|
|
142
|
+
args: tc.function?.arguments ?? '',
|
|
143
|
+
});
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
// Yield usage from final chunk
|
|
148
|
+
if (chunk.usage) {
|
|
149
|
+
const toolCalls = accumulatedToolCalls.size > 0
|
|
150
|
+
? Array.from(accumulatedToolCalls.values()).map((tc) => ({
|
|
151
|
+
id: tc.id,
|
|
152
|
+
name: tc.name,
|
|
153
|
+
args: JSON.parse(tc.args || '{}'),
|
|
154
|
+
}))
|
|
155
|
+
: undefined;
|
|
156
|
+
yield {
|
|
157
|
+
content: '',
|
|
158
|
+
tool_calls: toolCalls,
|
|
159
|
+
usage_metadata: {
|
|
160
|
+
input_tokens: chunk.usage.prompt_tokens,
|
|
161
|
+
output_tokens: chunk.usage.completion_tokens,
|
|
162
|
+
total_tokens: chunk.usage.total_tokens,
|
|
163
|
+
},
|
|
164
|
+
};
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
bindTools(tools) {
|
|
169
|
+
const bound = new OpenAIChatModel({
|
|
170
|
+
apiKey: this.client.apiKey,
|
|
171
|
+
modelName: this.modelName,
|
|
172
|
+
temperature: this.temperature,
|
|
173
|
+
maxTokens: this.maxTokens,
|
|
174
|
+
streamUsage: this.streamUsage,
|
|
175
|
+
});
|
|
176
|
+
bound.boundTools = tools;
|
|
177
|
+
bound.structuredSchema = this.structuredSchema;
|
|
178
|
+
// Share the same client instance
|
|
179
|
+
bound.client = this.client;
|
|
180
|
+
return bound;
|
|
181
|
+
}
|
|
182
|
+
withStructuredOutput(schema) {
|
|
183
|
+
const wrapped = new OpenAIChatModel({
|
|
184
|
+
apiKey: this.client.apiKey,
|
|
185
|
+
modelName: this.modelName,
|
|
186
|
+
temperature: this.temperature,
|
|
187
|
+
maxTokens: this.maxTokens,
|
|
188
|
+
streamUsage: this.streamUsage,
|
|
189
|
+
});
|
|
190
|
+
wrapped.structuredSchema = schema;
|
|
191
|
+
wrapped.boundTools = this.boundTools;
|
|
192
|
+
wrapped.client = this.client;
|
|
193
|
+
return wrapped;
|
|
194
|
+
}
|
|
195
|
+
}
|
|
196
|
+
//# sourceMappingURL=openai-chat-model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-chat-model.js","sourceRoot":"","sources":["../../../../lib/llm/providers/openai-chat-model.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAkBrD,MAAM,OAAO,eAAe;IAClB,MAAM,CAAS;IACf,SAAS,CAAS;IAClB,WAAW,CAAU;IACrB,SAAS,CAAU;IACnB,WAAW,CAAU;IACrB,UAAU,CAAoB;IAC9B,gBAAgB,CAA2B;IAEnD,YAAY,OAA+B;QACzC,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,SAAS;YACjE,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzD,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC;IACjD,CAAC;IAEO,gBAAgB,CAAC,QAAuB;QAC9C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAqC,EAAE;YAC7D,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,QAAQ;oBACX,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;gBAClD,KAAK,OAAO;oBACV,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChD,KAAK,IAAI,CAAC,CAAC,CAAC;oBACV,MAAM,KAAK,GAA+C;wBACxD,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,IAAI;qBAC7B,CAAC;oBACF,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;wBAC3B,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;4BAC7C,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,UAAmB;4BACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE;yBAChE,CAAC,CAAC,CAAC;oBACN,CAAC;oBACD,OAAO,KAAK,CAAC;gBACf,CAAC;gBACD,KAAK,MAAM;oBACT,OAAO;wBACL,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,GAAG,CAAC,YAAa;wBAC/B,OAAO,EAAE,GAAG,CAAC,OAAO;qBACrB,CAAC;gBACJ;oBACE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;YAClD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAC/C,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACjC,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC,MAAM,CAA4B;aACjE;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,cAAc,CACpB,OAAuC;QAEvC,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC;QAClD,IAAI,CAAC,SAAS,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QACzC,OAAO,SAAS;aACb,MAAM,CAAC,CAAC,EAAE,EAAuE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,UAAU,CAAC;aAC3G,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACZ,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;YACtB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC;SAChD,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAuB;QAClC,MAAM,MAAM,GAAkD;YAC5D,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC;YACzC,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzD,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChE,GAAG,CAAC,IAAI,CAAC,gBAAgB;gBACvB,CAAC,CAAC;oBACE,eAAe,EAAE;wBACf,IAAI,EAAE,aAAa;wBACnB,WAAW,EAAE;4BACX,IAAI,EAAE,mBAAmB;4BACzB,MAAM,EAAE,IAAI,CAAC,gBAAgB;4BAC7B,MAAM,EAAE,IAAI;yBACb;qBACF;iBACF;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC;QAEpC,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE;YACrC,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC;YACjD,cAAc,EAAE,QAAQ,CAAC,KAAK;gBAC5B,CAAC,CAAC;oBACE,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC1C,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB;oBAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;iBAC1C;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,OAAkC;QAElC,MAAM,MAAM,GAA+C;YACzD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC;YACzC,MAAM,EAAE,IAAI;YACZ,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;YACtE,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzD,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACjE,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE;YAC/D,MAAM,EAAE,OAAO,EAAE,MAAM;SACxB,CAAC,CAAC;QAEH,IAAI,oBAAoB,GAA4D,IAAI,GAAG,EAAE,CAAC;QAE9F,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;YAEtC,uBAAuB;YACvB,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;gBACnB,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;YACnC,CAAC;YAED,8BAA8B;YAC9B,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;gBACtB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBAClC,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;oBACpD,IAAI,QAAQ,EAAE,CAAC;wBACb,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC;oBAChD,CAAC;yBAAM,CAAC;wBACN,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE;4BACjC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE;4BACf,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;4BAC7B,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE;yBACnC,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM,SAAS,GAA2B,oBAAoB,CAAC,IAAI,GAAG,CAAC;oBACrE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBACrD,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,EAAE,CAAC,IAAI;wBACb,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC;qBAClC,CAAC,CAAC;oBACL,CAAC,CAAC,SAAS,CAAC;gBAEd,MAAM;oBACJ,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE,SAAS;oBACrB,cAAc,EAAE;wBACd,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,aAAa;wBACvC,aAAa,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB;wBAC5C,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;qBACvC;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAuB;QAC/B,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC;YAChC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC,CAAC;QACH,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC;QACzB,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC/C,iCAAiC;QACjC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oBAAoB,CAAC,MAA+B;QAClD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC;YAClC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC,CAAC;QACH,OAAO,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAClC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACrC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7B,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { Embeddings } from '../../types/llm-types.ts';
|
|
2
|
+
interface OpenAIEmbeddingsOptions {
|
|
3
|
+
apiKey?: string;
|
|
4
|
+
modelName: string;
|
|
5
|
+
baseURL?: string;
|
|
6
|
+
dimensions?: number;
|
|
7
|
+
}
|
|
8
|
+
export declare class OpenAIEmbeddingsProvider implements Embeddings {
|
|
9
|
+
private client;
|
|
10
|
+
private modelName;
|
|
11
|
+
private dimensions?;
|
|
12
|
+
constructor(options: OpenAIEmbeddingsOptions);
|
|
13
|
+
embedQuery(text: string): Promise<number[]>;
|
|
14
|
+
embedDocuments(texts: string[]): Promise<number[][]>;
|
|
15
|
+
}
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=openai-embeddings.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-embeddings.d.ts","sourceRoot":"","sources":["../../../../lib/llm/providers/openai-embeddings.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAE3D,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,wBAAyB,YAAW,UAAU;IACzD,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,UAAU,CAAC,CAAS;gBAEhB,OAAO,EAAE,uBAAuB;IAStC,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAU3C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;CAW3D"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import OpenAI from 'openai';
|
|
2
|
+
export class OpenAIEmbeddingsProvider {
|
|
3
|
+
client;
|
|
4
|
+
modelName;
|
|
5
|
+
dimensions;
|
|
6
|
+
constructor(options) {
|
|
7
|
+
this.client = new OpenAI({
|
|
8
|
+
apiKey: options.apiKey,
|
|
9
|
+
...(options.baseURL ? { baseURL: options.baseURL } : {}),
|
|
10
|
+
});
|
|
11
|
+
this.modelName = options.modelName;
|
|
12
|
+
this.dimensions = options.dimensions;
|
|
13
|
+
}
|
|
14
|
+
async embedQuery(text) {
|
|
15
|
+
const response = await this.client.embeddings.create({
|
|
16
|
+
model: this.modelName,
|
|
17
|
+
input: text,
|
|
18
|
+
encoding_format: 'float',
|
|
19
|
+
...(this.dimensions ? { dimensions: this.dimensions } : {}),
|
|
20
|
+
});
|
|
21
|
+
return response.data[0].embedding;
|
|
22
|
+
}
|
|
23
|
+
async embedDocuments(texts) {
|
|
24
|
+
const response = await this.client.embeddings.create({
|
|
25
|
+
model: this.modelName,
|
|
26
|
+
input: texts,
|
|
27
|
+
encoding_format: 'float',
|
|
28
|
+
...(this.dimensions ? { dimensions: this.dimensions } : {}),
|
|
29
|
+
});
|
|
30
|
+
return response.data
|
|
31
|
+
.sort((a, b) => a.index - b.index)
|
|
32
|
+
.map((d) => d.embedding);
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=openai-embeddings.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-embeddings.js","sourceRoot":"","sources":["../../../../lib/llm/providers/openai-embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAU5B,MAAM,OAAO,wBAAwB;IAC3B,MAAM,CAAS;IACf,SAAS,CAAS;IAClB,UAAU,CAAU;IAE5B,YAAY,OAAgC;QAC1C,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzD,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YACnD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,KAAK,EAAE,IAAI;YACX,eAAe,EAAE,OAAO;YACxB,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5D,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAe;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YACnD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,KAAK,EAAE,KAAK;YACZ,eAAe,EAAE,OAAO;YACxB,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5D,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI;aACjB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;CACF"}
|
package/dist/lib/llm/types.d.ts
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
|
-
export type { ModelConfig, EmbeddingModelConfig, LLMJsonConfig, } from './llm-config.
|
|
3
|
-
export { ModelConfigSchema, EmbeddingModelConfigSchema, LLMJsonConfigSchema, } from './llm-config.
|
|
2
|
+
export type { ModelConfig, EmbeddingModelConfig, LLMJsonConfig, } from './llm-config.ts';
|
|
3
|
+
export { ModelConfigSchema, EmbeddingModelConfigSchema, LLMJsonConfigSchema, } from './llm-config.ts';
|
|
4
4
|
export declare const AgentLLMRefSchema: z.ZodUnion<[z.ZodString, z.ZodObject<{
|
|
5
5
|
name: z.ZodString;
|
|
6
6
|
temperature: z.ZodOptional<z.ZodNumber>;
|
package/dist/lib/llm/types.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
|
-
export { ModelConfigSchema, EmbeddingModelConfigSchema, LLMJsonConfigSchema, } from
|
|
2
|
+
export { ModelConfigSchema, EmbeddingModelConfigSchema, LLMJsonConfigSchema, } from "./llm-config.js";
|
|
3
3
|
// Schema for agent LLM reference (can be string or object with overrides)
|
|
4
4
|
export const AgentLLMRefSchema = z.union([
|
|
5
5
|
z.string(), // Just the config name: "default"
|
package/dist/lib/logger.d.ts
CHANGED
|
@@ -1,4 +1,11 @@
|
|
|
1
1
|
import pino from 'pino';
|
|
2
|
+
export interface LogEntry {
|
|
3
|
+
level: string;
|
|
4
|
+
message: string;
|
|
5
|
+
timestamp: string;
|
|
6
|
+
component?: string;
|
|
7
|
+
}
|
|
8
|
+
export declare function getRecentLogs(limit?: number): LogEntry[];
|
|
2
9
|
/**
|
|
3
10
|
* Shared Pino configuration that can be used by both the standalone logger and Fastify
|
|
4
11
|
*/
|
package/dist/lib/logger.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../lib/logger.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB;;GAEG;AACH,wBAAgB,aAAa;;;;;;;;;;;EAkB5B;AAUD,eAAO,MAAM,UAAU,6BAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../lib/logger.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,MAAM,WAAW,QAAQ;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AA6BD,wBAAgB,aAAa,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,QAAQ,EAAE,CAExD;AAED;;GAEG;AACH,wBAAgB,aAAa;;;;;;;;;;;EAkB5B;AAUD,eAAO,MAAM,UAAU,6BAAqB,CAAC;AA4E7C,eAAO,MAAM,MAAM;oBA3DC,MAAM,WAAW,OAAO,EAAE;oBAO1B,MAAM,WAAW,OAAO,EAAE;qBAQzB,MAAM,WAAW,OAAO,EAAE;qBAa1B,MAAM,WAAW,OAAO,EAAE;qBAO1B,MAAM,WAAW,OAAO,EAAE;qBAO1B,MAAM,WAAW,OAAO,EAAE;CAiBM,CAAC;AAEtD;;;;;;;;GAQG;AACH,wBAAgB,YAAY,CAAC,SAAS,EAAE,MAAM;oBAtE1B,MAAM,WAAW,OAAO,EAAE;oBAO1B,MAAM,WAAW,OAAO,EAAE;qBAQzB,MAAM,WAAW,OAAO,EAAE;qBAa1B,MAAM,WAAW,OAAO,EAAE;qBAO1B,MAAM,WAAW,OAAO,EAAE;qBAO1B,MAAM,WAAW,OAAO,EAAE;EA+B9C"}
|