@llm4ts/core 0.1.0
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/LICENSE +21 -0
- package/README.md +47 -0
- package/dist/AgentSession.d.ts +66 -0
- package/dist/AgentSession.d.ts.map +1 -0
- package/dist/AgentSession.js +53 -0
- package/dist/AgentSession.js.map +1 -0
- package/dist/Capability.d.ts +101 -0
- package/dist/Capability.d.ts.map +1 -0
- package/dist/Capability.js +210 -0
- package/dist/Capability.js.map +1 -0
- package/dist/Connector.d.ts +69 -0
- package/dist/Connector.d.ts.map +1 -0
- package/dist/Connector.js +86 -0
- package/dist/Connector.js.map +1 -0
- package/dist/ConnectorConfig.d.ts +70 -0
- package/dist/ConnectorConfig.d.ts.map +1 -0
- package/dist/ConnectorConfig.js +110 -0
- package/dist/ConnectorConfig.js.map +1 -0
- package/dist/ConnectorRegistry.d.ts +28 -0
- package/dist/ConnectorRegistry.d.ts.map +1 -0
- package/dist/ConnectorRegistry.js +87 -0
- package/dist/ConnectorRegistry.js.map +1 -0
- package/dist/ContextManagement.d.ts +78 -0
- package/dist/ContextManagement.d.ts.map +1 -0
- package/dist/ContextManagement.js +221 -0
- package/dist/ContextManagement.js.map +1 -0
- package/dist/Conversation.d.ts +18 -0
- package/dist/Conversation.d.ts.map +1 -0
- package/dist/Conversation.js +17 -0
- package/dist/Conversation.js.map +1 -0
- package/dist/Errors.d.ts +64 -0
- package/dist/Errors.d.ts.map +1 -0
- package/dist/Errors.js +76 -0
- package/dist/Errors.js.map +1 -0
- package/dist/HttpClient.d.ts +43 -0
- package/dist/HttpClient.d.ts.map +1 -0
- package/dist/HttpClient.js +99 -0
- package/dist/HttpClient.js.map +1 -0
- package/dist/LlmService.d.ts +24 -0
- package/dist/LlmService.d.ts.map +1 -0
- package/dist/LlmService.js +4 -0
- package/dist/LlmService.js.map +1 -0
- package/dist/Models.d.ts +138 -0
- package/dist/Models.d.ts.map +1 -0
- package/dist/Models.js +209 -0
- package/dist/Models.js.map +1 -0
- package/dist/ProcessExecutor.d.ts +57 -0
- package/dist/ProcessExecutor.d.ts.map +1 -0
- package/dist/ProcessExecutor.js +71 -0
- package/dist/ProcessExecutor.js.map +1 -0
- package/dist/RateLimiter.d.ts +44 -0
- package/dist/RateLimiter.d.ts.map +1 -0
- package/dist/RateLimiter.js +155 -0
- package/dist/RateLimiter.js.map +1 -0
- package/dist/Streaming.d.ts +17 -0
- package/dist/Streaming.d.ts.map +1 -0
- package/dist/Streaming.js +65 -0
- package/dist/Streaming.js.map +1 -0
- package/dist/StructuredOutput.d.ts +10 -0
- package/dist/StructuredOutput.d.ts.map +1 -0
- package/dist/StructuredOutput.js +97 -0
- package/dist/StructuredOutput.js.map +1 -0
- package/dist/TemporaryFiles.d.ts +20 -0
- package/dist/TemporaryFiles.d.ts.map +1 -0
- package/dist/TemporaryFiles.js +18 -0
- package/dist/TemporaryFiles.js.map +1 -0
- package/dist/UsageLimits.d.ts +5 -0
- package/dist/UsageLimits.d.ts.map +1 -0
- package/dist/UsageLimits.js +187 -0
- package/dist/UsageLimits.js.map +1 -0
- package/dist/eval/Checks.d.ts +6 -0
- package/dist/eval/Checks.d.ts.map +1 -0
- package/dist/eval/Checks.js +39 -0
- package/dist/eval/Checks.js.map +1 -0
- package/dist/eval/Eval.d.ts +57 -0
- package/dist/eval/Eval.d.ts.map +1 -0
- package/dist/eval/Eval.js +91 -0
- package/dist/eval/Eval.js.map +1 -0
- package/dist/eval/EvalSuite.d.ts +21 -0
- package/dist/eval/EvalSuite.d.ts.map +1 -0
- package/dist/eval/EvalSuite.js +30 -0
- package/dist/eval/EvalSuite.js.map +1 -0
- package/dist/eval/Evaluator.d.ts +10 -0
- package/dist/eval/Evaluator.d.ts.map +1 -0
- package/dist/eval/Evaluator.js +14 -0
- package/dist/eval/Evaluator.js.map +1 -0
- package/dist/eval/Judge.d.ts +15 -0
- package/dist/eval/Judge.d.ts.map +1 -0
- package/dist/eval/Judge.js +73 -0
- package/dist/eval/Judge.js.map +1 -0
- package/dist/observability/MeteredLlmService.d.ts +16 -0
- package/dist/observability/MeteredLlmService.d.ts.map +1 -0
- package/dist/observability/MeteredLlmService.js +66 -0
- package/dist/observability/MeteredLlmService.js.map +1 -0
- package/dist/observability/Metrics.d.ts +71 -0
- package/dist/observability/Metrics.d.ts.map +1 -0
- package/dist/observability/Metrics.js +147 -0
- package/dist/observability/Metrics.js.map +1 -0
- package/dist/observability/Redaction.d.ts +8 -0
- package/dist/observability/Redaction.d.ts.map +1 -0
- package/dist/observability/Redaction.js +38 -0
- package/dist/observability/Redaction.js.map +1 -0
- package/dist/observability/StreamRecorder.d.ts +9 -0
- package/dist/observability/StreamRecorder.d.ts.map +1 -0
- package/dist/observability/StreamRecorder.js +10 -0
- package/dist/observability/StreamRecorder.js.map +1 -0
- package/dist/observability/StructuredLogger.d.ts +31 -0
- package/dist/observability/StructuredLogger.d.ts.map +1 -0
- package/dist/observability/StructuredLogger.js +34 -0
- package/dist/observability/StructuredLogger.js.map +1 -0
- package/dist/observability/Tracing.d.ts +38 -0
- package/dist/observability/Tracing.d.ts.map +1 -0
- package/dist/observability/Tracing.js +75 -0
- package/dist/observability/Tracing.js.map +1 -0
- package/dist/providers/AnthropicModels.d.ts +95 -0
- package/dist/providers/AnthropicModels.d.ts.map +1 -0
- package/dist/providers/AnthropicModels.js +83 -0
- package/dist/providers/AnthropicModels.js.map +1 -0
- package/dist/providers/AnthropicProvider.d.ts +13 -0
- package/dist/providers/AnthropicProvider.d.ts.map +1 -0
- package/dist/providers/AnthropicProvider.js +177 -0
- package/dist/providers/AnthropicProvider.js.map +1 -0
- package/dist/providers/AntigravityConnector.d.ts +7 -0
- package/dist/providers/AntigravityConnector.d.ts.map +1 -0
- package/dist/providers/AntigravityConnector.js +52 -0
- package/dist/providers/AntigravityConnector.js.map +1 -0
- package/dist/providers/ClaudeAgentSession.d.ts +11 -0
- package/dist/providers/ClaudeAgentSession.d.ts.map +1 -0
- package/dist/providers/ClaudeAgentSession.js +158 -0
- package/dist/providers/ClaudeAgentSession.js.map +1 -0
- package/dist/providers/ClaudeCliConnector.d.ts +9 -0
- package/dist/providers/ClaudeCliConnector.d.ts.map +1 -0
- package/dist/providers/ClaudeCliConnector.js +123 -0
- package/dist/providers/ClaudeCliConnector.js.map +1 -0
- package/dist/providers/CliSupport.d.ts +21 -0
- package/dist/providers/CliSupport.d.ts.map +1 -0
- package/dist/providers/CliSupport.js +62 -0
- package/dist/providers/CliSupport.js.map +1 -0
- package/dist/providers/CodexConnector.d.ts +12 -0
- package/dist/providers/CodexConnector.d.ts.map +1 -0
- package/dist/providers/CodexConnector.js +175 -0
- package/dist/providers/CodexConnector.js.map +1 -0
- package/dist/providers/ConnectorFactories.d.ts +15 -0
- package/dist/providers/ConnectorFactories.d.ts.map +1 -0
- package/dist/providers/ConnectorFactories.js +74 -0
- package/dist/providers/ConnectorFactories.js.map +1 -0
- package/dist/providers/CopilotConnector.d.ts +6 -0
- package/dist/providers/CopilotConnector.d.ts.map +1 -0
- package/dist/providers/CopilotConnector.js +35 -0
- package/dist/providers/CopilotConnector.js.map +1 -0
- package/dist/providers/CursorConnector.d.ts +8 -0
- package/dist/providers/CursorConnector.d.ts.map +1 -0
- package/dist/providers/CursorConnector.js +82 -0
- package/dist/providers/CursorConnector.js.map +1 -0
- package/dist/providers/GeminiApiProvider.d.ts +9 -0
- package/dist/providers/GeminiApiProvider.d.ts.map +1 -0
- package/dist/providers/GeminiApiProvider.js +181 -0
- package/dist/providers/GeminiApiProvider.js.map +1 -0
- package/dist/providers/GeminiCliProvider.d.ts +106 -0
- package/dist/providers/GeminiCliProvider.d.ts.map +1 -0
- package/dist/providers/GeminiCliProvider.js +596 -0
- package/dist/providers/GeminiCliProvider.js.map +1 -0
- package/dist/providers/GeminiModels.d.ts +92 -0
- package/dist/providers/GeminiModels.d.ts.map +1 -0
- package/dist/providers/GeminiModels.js +80 -0
- package/dist/providers/GeminiModels.js.map +1 -0
- package/dist/providers/GrokCliConnector.d.ts +8 -0
- package/dist/providers/GrokCliConnector.d.ts.map +1 -0
- package/dist/providers/GrokCliConnector.js +82 -0
- package/dist/providers/GrokCliConnector.js.map +1 -0
- package/dist/providers/LmStudioModels.d.ts +114 -0
- package/dist/providers/LmStudioModels.d.ts.map +1 -0
- package/dist/providers/LmStudioModels.js +102 -0
- package/dist/providers/LmStudioModels.js.map +1 -0
- package/dist/providers/LmStudioProvider.d.ts +13 -0
- package/dist/providers/LmStudioProvider.d.ts.map +1 -0
- package/dist/providers/LmStudioProvider.js +158 -0
- package/dist/providers/LmStudioProvider.js.map +1 -0
- package/dist/providers/MockProvider.d.ts +6 -0
- package/dist/providers/MockProvider.d.ts.map +1 -0
- package/dist/providers/MockProvider.js +271 -0
- package/dist/providers/MockProvider.js.map +1 -0
- package/dist/providers/OllamaModels.d.ts +68 -0
- package/dist/providers/OllamaModels.d.ts.map +1 -0
- package/dist/providers/OllamaModels.js +59 -0
- package/dist/providers/OllamaModels.js.map +1 -0
- package/dist/providers/OllamaProvider.d.ts +7 -0
- package/dist/providers/OllamaProvider.d.ts.map +1 -0
- package/dist/providers/OllamaProvider.js +134 -0
- package/dist/providers/OllamaProvider.js.map +1 -0
- package/dist/providers/OpenAIModels.d.ts +139 -0
- package/dist/providers/OpenAIModels.d.ts.map +1 -0
- package/dist/providers/OpenAIModels.js +123 -0
- package/dist/providers/OpenAIModels.js.map +1 -0
- package/dist/providers/OpenAIProvider.d.ts +9 -0
- package/dist/providers/OpenAIProvider.d.ts.map +1 -0
- package/dist/providers/OpenAIProvider.js +185 -0
- package/dist/providers/OpenAIProvider.js.map +1 -0
- package/dist/providers/OpenCodeCliConnector.d.ts +8 -0
- package/dist/providers/OpenCodeCliConnector.d.ts.map +1 -0
- package/dist/providers/OpenCodeCliConnector.js +93 -0
- package/dist/providers/OpenCodeCliConnector.js.map +1 -0
- package/dist/providers/OpenCodeModels.d.ts +61 -0
- package/dist/providers/OpenCodeModels.d.ts.map +1 -0
- package/dist/providers/OpenCodeModels.js +53 -0
- package/dist/providers/OpenCodeModels.js.map +1 -0
- package/dist/providers/OpenCodeProvider.d.ts +10 -0
- package/dist/providers/OpenCodeProvider.d.ts.map +1 -0
- package/dist/providers/OpenCodeProvider.js +154 -0
- package/dist/providers/OpenCodeProvider.js.map +1 -0
- package/dist/providers/PiConnector.d.ts +8 -0
- package/dist/providers/PiConnector.d.ts.map +1 -0
- package/dist/providers/PiConnector.js +99 -0
- package/dist/providers/PiConnector.js.map +1 -0
- package/dist/tools/Tool.d.ts +83 -0
- package/dist/tools/Tool.d.ts.map +1 -0
- package/dist/tools/Tool.js +278 -0
- package/dist/tools/Tool.js.map +1 -0
- package/dist/tools/ToolCallingExecutor.d.ts +18 -0
- package/dist/tools/ToolCallingExecutor.d.ts.map +1 -0
- package/dist/tools/ToolCallingExecutor.js +64 -0
- package/dist/tools/ToolCallingExecutor.js.map +1 -0
- package/dist/tools/ToolRegistry.d.ts +45 -0
- package/dist/tools/ToolRegistry.d.ts.map +1 -0
- package/dist/tools/ToolRegistry.js +133 -0
- package/dist/tools/ToolRegistry.js.map +1 -0
- package/package.json +105 -0
- package/src/.gitkeep +1 -0
- package/src/AgentSession.ts +69 -0
- package/src/Capability.ts +240 -0
- package/src/Connector.ts +220 -0
- package/src/ConnectorConfig.ts +146 -0
- package/src/ConnectorRegistry.ts +184 -0
- package/src/ContextManagement.ts +372 -0
- package/src/Conversation.ts +21 -0
- package/src/Errors.ts +87 -0
- package/src/HttpClient.ts +202 -0
- package/src/LlmService.ts +45 -0
- package/src/Models.ts +250 -0
- package/src/ProcessExecutor.ts +157 -0
- package/src/RateLimiter.ts +238 -0
- package/src/Streaming.ts +164 -0
- package/src/StructuredOutput.ts +138 -0
- package/src/TemporaryFiles.ts +51 -0
- package/src/UsageLimits.ts +242 -0
- package/src/eval/Checks.ts +68 -0
- package/src/eval/Eval.ts +112 -0
- package/src/eval/EvalSuite.ts +63 -0
- package/src/eval/Evaluator.ts +31 -0
- package/src/eval/Judge.ts +92 -0
- package/src/observability/MeteredLlmService.ts +156 -0
- package/src/observability/Metrics.ts +195 -0
- package/src/observability/Redaction.ts +55 -0
- package/src/observability/StreamRecorder.ts +32 -0
- package/src/observability/StructuredLogger.ts +62 -0
- package/src/observability/Tracing.ts +141 -0
- package/src/providers/AnthropicModels.ts +96 -0
- package/src/providers/AnthropicProvider.ts +320 -0
- package/src/providers/AntigravityConnector.ts +77 -0
- package/src/providers/ClaudeAgentSession.ts +226 -0
- package/src/providers/ClaudeCliConnector.ts +201 -0
- package/src/providers/CliSupport.ts +100 -0
- package/src/providers/CodexConnector.ts +251 -0
- package/src/providers/ConnectorFactories.ts +142 -0
- package/src/providers/CopilotConnector.ts +47 -0
- package/src/providers/CursorConnector.ts +120 -0
- package/src/providers/GeminiApiProvider.ts +337 -0
- package/src/providers/GeminiCliProvider.ts +793 -0
- package/src/providers/GeminiModels.ts +97 -0
- package/src/providers/GrokCliConnector.ts +117 -0
- package/src/providers/LmStudioModels.ts +117 -0
- package/src/providers/LmStudioProvider.ts +260 -0
- package/src/providers/MockProvider.ts +322 -0
- package/src/providers/OllamaModels.ts +64 -0
- package/src/providers/OllamaProvider.ts +236 -0
- package/src/providers/OpenAIModels.ts +152 -0
- package/src/providers/OpenAIProvider.ts +331 -0
- package/src/providers/OpenCodeCliConnector.ts +135 -0
- package/src/providers/OpenCodeModels.ts +60 -0
- package/src/providers/OpenCodeProvider.ts +278 -0
- package/src/providers/PiConnector.ts +142 -0
- package/src/tools/Tool.ts +379 -0
- package/src/tools/ToolCallingExecutor.ts +106 -0
- package/src/tools/ToolRegistry.ts +224 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AnthropicModels.js","sourceRoot":"","sources":["../../src/providers/AnthropicModels.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAA;AAEzC,MAAM,OAAO,gBAAiB,SAAQ,MAAM,CAAC,KAAK,CAAmB,kBAAkB,CAAC,CAAC;IACvF,IAAI,EAAE,MAAM,CAAC,MAAM;IACnB,OAAO,EAAE,MAAM,CAAC,MAAM;CACvB,CAAC;CAAG;AAEL,MAAM,OAAO,gBAAiB,SAAQ,MAAM,CAAC,KAAK,CAAmB,kBAAkB,CAAC,CAAC;IACvF,KAAK,EAAE,MAAM,CAAC,MAAM;IACpB,UAAU,EAAE,MAAM,CAAC,GAAG;IACtB,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC;IACxC,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IAC9C,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACzC,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC;CAC3C,CAAC;CAAG;AAEL,MAAM,OAAO,qBAAsB,SAAQ,MAAM,CAAC,KAAK,CACrD,uBAAuB,CACxB,CAAC;IACA,IAAI,EAAE,MAAM,CAAC,MAAM;IACnB,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;CACvD,CAAC;CAAG;AAEL,MAAM,OAAO,cAAe,SAAQ,MAAM,CAAC,KAAK,CAAiB,gBAAgB,CAAC,CAAC;IACjF,YAAY,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC;IAC5C,aAAa,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC;CAC9C,CAAC;CAAG;AAEL,MAAM,OAAO,iBAAkB,SAAQ,MAAM,CAAC,KAAK,CAAoB,mBAAmB,CAAC,CAAC;IAC1F,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACrC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC;IAC5C,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC;IACzC,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;CAC9D,CAAC;CAAG;AAEL,MAAM,OAAO,wBAAyB,SAAQ,MAAM,CAAC,KAAK,CACxD,0BAA0B,CAC3B,CAAC;IACA,IAAI,EAAE,MAAM,CAAC,MAAM;IACnB,UAAU,EAAE,UAAU;IACtB,QAAQ,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;CAC1D,CAAC;CAAG;AAEL,MAAM,OAAO,aAAc,SAAQ,MAAM,CAAC,KAAK,CAAgB,eAAe,CAAC,CAAC;IAC9E,IAAI,EAAE,MAAM,CAAC,MAAM;IACnB,WAAW,EAAE,MAAM,CAAC,MAAM;IAC1B,YAAY,EAAE,wBAAwB;CACvC,CAAC;CAAG;AAEL,MAAM,OAAO,yBAA0B,SAAQ,MAAM,CAAC,KAAK,CACzD,2BAA2B,CAC5B,CAAC;IACA,IAAI,EAAE,MAAM,CAAC,MAAM;IACnB,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACtD,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACrC,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACvC,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC;CACvC,CAAC;CAAG;AAEL,MAAM,OAAO,yBAA0B,SAAQ,MAAM,CAAC,KAAK,CACzD,2BAA2B,CAC5B,CAAC;IACA,KAAK,EAAE,MAAM,CAAC,MAAM;IACpB,UAAU,EAAE,MAAM,CAAC,GAAG;IACtB,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC;IACxC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC;IAClC,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IAC9C,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;CAC1C,CAAC;CAAG;AAEL,MAAM,OAAO,0BAA2B,SAAQ,MAAM,CAAC,KAAK,CAC1D,4BAA4B,CAC7B,CAAC;IACA,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACrC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC;IAChD,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC;IACzC,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;CAC9D,CAAC;CAAG;AAEL,MAAM,OAAO,yBAA0B,SAAQ,MAAM,CAAC,KAAK,CACzD,2BAA2B,CAC5B,CAAC;IACA,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;IACvC,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACtD,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;CAC9D,CAAC;CAAG;AAEL,MAAM,OAAO,oBAAqB,SAAQ,MAAM,CAAC,KAAK,CACpD,sBAAsB,CACvB,CAAC;IACA,IAAI,EAAE,MAAM,CAAC,MAAM;IACnB,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,yBAAyB,CAAC;CACrD,CAAC;CAAG"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { type ApiConnectorShape } from "../Connector.ts";
|
|
2
|
+
import type { HttpClientShape } from "../HttpClient.ts";
|
|
3
|
+
import { type LlmConfig, type Message, type ToolDefinition } from "../Models.ts";
|
|
4
|
+
import { AnthropicMessage, AnthropicRequest, AnthropicRequestWithTools } from "./AnthropicModels.ts";
|
|
5
|
+
export interface AnthropicHistory {
|
|
6
|
+
readonly messages: ReadonlyArray<AnthropicMessage>;
|
|
7
|
+
readonly system: string | undefined;
|
|
8
|
+
}
|
|
9
|
+
export declare const anthropicHistory: (messages: ReadonlyArray<Message>) => AnthropicHistory;
|
|
10
|
+
export declare const buildAnthropicRequest: (config: LlmConfig, messages: ReadonlyArray<AnthropicMessage>, stream: boolean, system?: string) => AnthropicRequest;
|
|
11
|
+
export declare const buildAnthropicToolRequest: (config: LlmConfig, prompt: string, tools: ReadonlyArray<ToolDefinition>) => AnthropicRequestWithTools;
|
|
12
|
+
export declare const makeAnthropicProvider: (config: LlmConfig, httpClient: HttpClientShape) => ApiConnectorShape;
|
|
13
|
+
//# sourceMappingURL=AnthropicProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AnthropicProvider.d.ts","sourceRoot":"","sources":["../../src/providers/AnthropicProvider.ts"],"names":[],"mappings":"AAIA,OAAO,EAAoB,KAAK,iBAAiB,EAAE,MAAM,iBAAiB,CAAA;AAE1E,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAA;AAEvD,OAAO,EAML,KAAK,SAAS,EACd,KAAK,OAAO,EACZ,KAAK,cAAc,EACpB,MAAM,cAAc,CAAA;AAIrB,OAAO,EACL,gBAAgB,EAChB,gBAAgB,EAChB,yBAAyB,EAM1B,MAAM,sBAAsB,CAAA;AAuC7B,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAA;IAClD,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAA;CACpC;AAED,eAAO,MAAM,gBAAgB,GAAI,UAAU,aAAa,CAAC,OAAO,CAAC,KAAG,gBAUlE,CAAA;AAEF,eAAO,MAAM,qBAAqB,GAChC,QAAQ,SAAS,EACjB,UAAU,aAAa,CAAC,gBAAgB,CAAC,EACzC,QAAQ,OAAO,EACf,SAAS,MAAM,KACd,gBAQC,CAAA;AAEJ,eAAO,MAAM,yBAAyB,GACpC,QAAQ,SAAS,EACjB,QAAQ,MAAM,EACd,OAAO,aAAa,CAAC,cAAc,CAAC,KACnC,yBAqBC,CAAA;AAmBJ,eAAO,MAAM,qBAAqB,GAChC,QAAQ,SAAS,EACjB,YAAY,eAAe,KAC1B,iBA2KF,CAAA"}
|
|
@@ -0,0 +1,177 @@
|
|
|
1
|
+
import * as Effect from "effect/Effect";
|
|
2
|
+
import * as Redacted from "effect/Redacted";
|
|
3
|
+
import * as Schema from "effect/Schema";
|
|
4
|
+
import * as Stream from "effect/Stream";
|
|
5
|
+
import { makeApiConnector } from "../Connector.js";
|
|
6
|
+
import { AuthenticationError, ConfigError, ParseError } from "../Errors.js";
|
|
7
|
+
import { ConnectorIds, LlmChunk, ToolCall, ToolCallResponse } from "../Models.js";
|
|
8
|
+
import { collect } from "../Streaming.js";
|
|
9
|
+
import { parseFromText } from "../StructuredOutput.js";
|
|
10
|
+
import { AnthropicMessage, AnthropicRequest, AnthropicRequestWithTools, AnthropicResponse, AnthropicResponseWithTools, AnthropicStreamChunk, AnthropicTool, AnthropicToolInputSchema } from "./AnthropicModels.js";
|
|
11
|
+
const normalizedBaseUrl = (baseUrl) => baseUrl.replace(/\/+$/, "");
|
|
12
|
+
const decodeResponse = (raw) => Schema.decodeUnknownEffect(Schema.fromJsonString(AnthropicResponse))(raw).pipe(Effect.mapError((error) => ParseError.make({
|
|
13
|
+
message: `Failed to decode Anthropic response: ${String(error)}`,
|
|
14
|
+
raw
|
|
15
|
+
})));
|
|
16
|
+
const decodeToolResponse = (raw) => Schema.decodeUnknownEffect(Schema.fromJsonString(AnthropicResponseWithTools))(raw).pipe(Effect.mapError((error) => ParseError.make({
|
|
17
|
+
message: `Failed to decode Anthropic tool response: ${String(error)}`,
|
|
18
|
+
raw
|
|
19
|
+
})));
|
|
20
|
+
const decodeStreamChunk = (raw) => Schema.decodeUnknownEffect(Schema.fromJsonString(AnthropicStreamChunk))(raw).pipe(Effect.mapError((error) => ParseError.make({
|
|
21
|
+
message: `Failed to parse Anthropic stream chunk: ${String(error)}`,
|
|
22
|
+
raw
|
|
23
|
+
})));
|
|
24
|
+
export const anthropicHistory = (messages) => ({
|
|
25
|
+
system: messages.find((message) => message.role === "System")?.content,
|
|
26
|
+
messages: messages
|
|
27
|
+
.filter((message) => message.role !== "System")
|
|
28
|
+
.map((message) => AnthropicMessage.make({
|
|
29
|
+
role: message.role === "Assistant" ? "assistant" : "user",
|
|
30
|
+
content: message.content
|
|
31
|
+
}))
|
|
32
|
+
});
|
|
33
|
+
export const buildAnthropicRequest = (config, messages, stream, system) => AnthropicRequest.make({
|
|
34
|
+
model: config.model,
|
|
35
|
+
max_tokens: config.maxTokens ?? 4096,
|
|
36
|
+
messages,
|
|
37
|
+
stream,
|
|
38
|
+
...(config.temperature === undefined ? {} : { temperature: config.temperature }),
|
|
39
|
+
...(system === undefined ? {} : { system })
|
|
40
|
+
});
|
|
41
|
+
export const buildAnthropicToolRequest = (config, prompt, tools) => AnthropicRequestWithTools.make({
|
|
42
|
+
model: config.model,
|
|
43
|
+
max_tokens: config.maxTokens ?? 4096,
|
|
44
|
+
messages: [
|
|
45
|
+
AnthropicMessage.make({
|
|
46
|
+
role: "user",
|
|
47
|
+
content: prompt
|
|
48
|
+
})
|
|
49
|
+
],
|
|
50
|
+
tools: tools.map((tool) => AnthropicTool.make({
|
|
51
|
+
name: tool.name,
|
|
52
|
+
description: tool.description,
|
|
53
|
+
input_schema: AnthropicToolInputSchema.make({
|
|
54
|
+
type: "object",
|
|
55
|
+
properties: tool.parameters
|
|
56
|
+
})
|
|
57
|
+
})),
|
|
58
|
+
...(config.temperature === undefined ? {} : { temperature: config.temperature })
|
|
59
|
+
});
|
|
60
|
+
const contentFromResponse = (response, raw) => {
|
|
61
|
+
const content = response.content[0]?.text?.trim();
|
|
62
|
+
return content === undefined || content === null || content.length === 0
|
|
63
|
+
? Effect.fail(ParseError.make({
|
|
64
|
+
message: "Anthropic response missing content[0].text",
|
|
65
|
+
raw
|
|
66
|
+
}))
|
|
67
|
+
: Effect.succeed(content);
|
|
68
|
+
};
|
|
69
|
+
const jsonString = (value) => JSON.stringify(value) ?? "{}";
|
|
70
|
+
export const makeAnthropicProvider = (config, httpClient) => {
|
|
71
|
+
const requiredConfig = config.baseUrl === undefined
|
|
72
|
+
? Effect.fail(ConfigError.make({
|
|
73
|
+
message: "Missing baseUrl for Anthropic provider"
|
|
74
|
+
}))
|
|
75
|
+
: config.apiKey === undefined
|
|
76
|
+
? Effect.fail(AuthenticationError.make({
|
|
77
|
+
message: "Missing API key for Anthropic provider (set ANTHROPIC_API_KEY or pass apiKey in the connector config)"
|
|
78
|
+
}))
|
|
79
|
+
: Effect.succeed({
|
|
80
|
+
baseUrl: normalizedBaseUrl(config.baseUrl),
|
|
81
|
+
apiKey: Redacted.value(config.apiKey)
|
|
82
|
+
});
|
|
83
|
+
const authHeaders = (apiKey) => ({
|
|
84
|
+
"x-api-key": apiKey,
|
|
85
|
+
"anthropic-version": "2023-06-01"
|
|
86
|
+
});
|
|
87
|
+
const executeRequest = (messages, system) => Effect.gen(function* () {
|
|
88
|
+
const { apiKey, baseUrl } = yield* requiredConfig;
|
|
89
|
+
const request = buildAnthropicRequest(config, messages, false, system);
|
|
90
|
+
const raw = yield* httpClient.postJson(`${baseUrl}/messages`, JSON.stringify(request), authHeaders(apiKey), config.timeout);
|
|
91
|
+
const response = yield* decodeResponse(raw);
|
|
92
|
+
return [response, raw];
|
|
93
|
+
});
|
|
94
|
+
const executeStreamRequest = (messages, system) => Stream.unwrap(Effect.map(requiredConfig, ({ apiKey, baseUrl }) => {
|
|
95
|
+
const request = buildAnthropicRequest(config, messages, true, system);
|
|
96
|
+
return httpClient
|
|
97
|
+
.postJsonStreamSSE(`${baseUrl}/messages`, JSON.stringify(request), authHeaders(apiKey), config.timeout)
|
|
98
|
+
.pipe(Stream.mapEffect(decodeStreamChunk), Stream.flatMap((chunk) => {
|
|
99
|
+
if (chunk.type === "content_block_delta") {
|
|
100
|
+
const delta = chunk.delta?.text ?? "";
|
|
101
|
+
return delta.length === 0
|
|
102
|
+
? Stream.empty
|
|
103
|
+
: Stream.succeed(LlmChunk.make({
|
|
104
|
+
delta,
|
|
105
|
+
metadata: {
|
|
106
|
+
provider: "anthropic",
|
|
107
|
+
model: config.model
|
|
108
|
+
}
|
|
109
|
+
}));
|
|
110
|
+
}
|
|
111
|
+
const finishReason = chunk.type === "message_delta" ? (chunk.delta?.stop_reason ?? undefined) : undefined;
|
|
112
|
+
return finishReason === undefined
|
|
113
|
+
? Stream.empty
|
|
114
|
+
: Stream.succeed(LlmChunk.make({
|
|
115
|
+
delta: "",
|
|
116
|
+
finishReason,
|
|
117
|
+
metadata: {
|
|
118
|
+
provider: "anthropic",
|
|
119
|
+
model: config.model
|
|
120
|
+
}
|
|
121
|
+
}));
|
|
122
|
+
}));
|
|
123
|
+
}));
|
|
124
|
+
const executeStructuredWithUsage = (prompt, schema, jsonSchema) => Effect.gen(function* () {
|
|
125
|
+
const [response, raw] = yield* executeRequest([
|
|
126
|
+
AnthropicMessage.make({
|
|
127
|
+
role: "user",
|
|
128
|
+
content: `${prompt}\n\n` + "Please respond with valid JSON matching the provided schema."
|
|
129
|
+
})
|
|
130
|
+
]);
|
|
131
|
+
const content = yield* contentFromResponse(response, raw);
|
|
132
|
+
const value = yield* parseFromText(content, schema, jsonSchema);
|
|
133
|
+
const result = [value, undefined, undefined];
|
|
134
|
+
return result;
|
|
135
|
+
});
|
|
136
|
+
const executeWithTools = (prompt, tools) => Effect.gen(function* () {
|
|
137
|
+
const { apiKey, baseUrl } = yield* requiredConfig;
|
|
138
|
+
const request = buildAnthropicToolRequest(config, prompt, tools);
|
|
139
|
+
const raw = yield* httpClient.postJson(`${baseUrl}/messages`, JSON.stringify(request), authHeaders(apiKey), config.timeout);
|
|
140
|
+
const response = yield* decodeToolResponse(raw);
|
|
141
|
+
const toolCalls = response.content
|
|
142
|
+
.filter((block) => block.type === "tool_use")
|
|
143
|
+
.map((block) => ToolCall.make({
|
|
144
|
+
id: block.id ?? "",
|
|
145
|
+
name: block.name ?? "",
|
|
146
|
+
arguments: block.input === undefined ? "{}" : jsonString(block.input)
|
|
147
|
+
}));
|
|
148
|
+
const content = response.content.find((block) => block.type === "text" && block.text !== undefined && block.text !== null)?.text;
|
|
149
|
+
return ToolCallResponse.make({
|
|
150
|
+
toolCalls,
|
|
151
|
+
finishReason: response.stop_reason ?? "stop",
|
|
152
|
+
...(content === undefined || content === null ? {} : { content })
|
|
153
|
+
});
|
|
154
|
+
});
|
|
155
|
+
const executeStream = (prompt) => executeStreamRequest([
|
|
156
|
+
AnthropicMessage.make({
|
|
157
|
+
role: "user",
|
|
158
|
+
content: prompt
|
|
159
|
+
})
|
|
160
|
+
]);
|
|
161
|
+
const isAvailable = collect(executeStream("health check")).pipe(Effect.match({
|
|
162
|
+
onFailure: () => false,
|
|
163
|
+
onSuccess: () => true
|
|
164
|
+
}));
|
|
165
|
+
return makeApiConnector({
|
|
166
|
+
id: ConnectorIds.Anthropic,
|
|
167
|
+
executeStream,
|
|
168
|
+
executeStreamWithHistory: (messages) => {
|
|
169
|
+
const history = anthropicHistory(messages);
|
|
170
|
+
return executeStreamRequest(history.messages, history.system);
|
|
171
|
+
},
|
|
172
|
+
executeWithTools,
|
|
173
|
+
executeStructuredWithUsage,
|
|
174
|
+
isAvailable
|
|
175
|
+
});
|
|
176
|
+
};
|
|
177
|
+
//# sourceMappingURL=AnthropicProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AnthropicProvider.js","sourceRoot":"","sources":["../../src/providers/AnthropicProvider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,QAAQ,MAAM,iBAAiB,CAAA;AAC3C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,EAAE,gBAAgB,EAA0B,MAAM,iBAAiB,CAAA;AAC1E,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,UAAU,EAAiB,MAAM,cAAc,CAAA;AAG1F,OAAO,EACL,YAAY,EACZ,QAAQ,EACR,QAAQ,EACR,gBAAgB,EAKjB,MAAM,cAAc,CAAA;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAA;AAEtD,OAAO,EACL,gBAAgB,EAChB,gBAAgB,EAChB,yBAAyB,EACzB,iBAAiB,EACjB,0BAA0B,EAC1B,oBAAoB,EACpB,aAAa,EACb,wBAAwB,EACzB,MAAM,sBAAsB,CAAA;AAO7B,MAAM,iBAAiB,GAAG,CAAC,OAAe,EAAU,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;AAElF,MAAM,cAAc,GAAG,CAAC,GAAW,EAAgD,EAAE,CACnF,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAC5E,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACxB,UAAU,CAAC,IAAI,CAAC;IACd,OAAO,EAAE,wCAAwC,MAAM,CAAC,KAAK,CAAC,EAAE;IAChE,GAAG;CACJ,CAAC,CACH,CACF,CAAA;AAEH,MAAM,kBAAkB,GAAG,CAAC,GAAW,EAAyD,EAAE,CAChG,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,0BAA0B,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CACrF,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACxB,UAAU,CAAC,IAAI,CAAC;IACd,OAAO,EAAE,6CAA6C,MAAM,CAAC,KAAK,CAAC,EAAE;IACrE,GAAG;CACJ,CAAC,CACH,CACF,CAAA;AAEH,MAAM,iBAAiB,GAAG,CAAC,GAAW,EAAmD,EAAE,CACzF,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAC/E,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACxB,UAAU,CAAC,IAAI,CAAC;IACd,OAAO,EAAE,2CAA2C,MAAM,CAAC,KAAK,CAAC,EAAE;IACnE,GAAG;CACJ,CAAC,CACH,CACF,CAAA;AAOH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,QAAgC,EAAoB,EAAE,CAAC,CAAC;IACvF,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,EAAE,OAAO;IACtE,QAAQ,EAAE,QAAQ;SACf,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;SAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CACf,gBAAgB,CAAC,IAAI,CAAC;QACpB,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;QACzD,OAAO,EAAE,OAAO,CAAC,OAAO;KACzB,CAAC,CACH;CACJ,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,MAAiB,EACjB,QAAyC,EACzC,MAAe,EACf,MAAe,EACG,EAAE,CACpB,gBAAgB,CAAC,IAAI,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC,KAAK;IACnB,UAAU,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;IACpC,QAAQ;IACR,MAAM;IACN,GAAG,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC;IAChF,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5C,CAAC,CAAA;AAEJ,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,MAAiB,EACjB,MAAc,EACd,KAAoC,EACT,EAAE,CAC7B,yBAAyB,CAAC,IAAI,CAAC;IAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;IACnB,UAAU,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;IACpC,QAAQ,EAAE;QACR,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM;SAChB,CAAC;KACH;IACD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACxB,aAAa,CAAC,IAAI,CAAC;QACjB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,YAAY,EAAE,wBAAwB,CAAC,IAAI,CAAC;YAC1C,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;KACH,CAAC,CACH;IACD,GAAG,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC;CACjF,CAAC,CAAA;AAEJ,MAAM,mBAAmB,GAAG,CAC1B,QAA2B,EAC3B,GAAW,EACwB,EAAE;IACrC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAA;IACjD,OAAO,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QACtE,CAAC,CAAC,MAAM,CAAC,IAAI,CACT,UAAU,CAAC,IAAI,CAAC;YACd,OAAO,EAAE,4CAA4C;YACrD,GAAG;SACJ,CAAC,CACH;QACH,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;AAC7B,CAAC,CAAA;AAED,MAAM,UAAU,GAAG,CAAC,KAA8B,EAAU,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,IAAI,CAAA;AAE5F,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,MAAiB,EACjB,UAA2B,EACR,EAAE;IACrB,MAAM,cAAc,GAClB,MAAM,CAAC,OAAO,KAAK,SAAS;QAC1B,CAAC,CAAC,MAAM,CAAC,IAAI,CACT,WAAW,CAAC,IAAI,CAAC;YACf,OAAO,EAAE,wCAAwC;SAClD,CAAC,CACH;QACH,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS;YAC3B,CAAC,CAAC,MAAM,CAAC,IAAI,CACT,mBAAmB,CAAC,IAAI,CAAC;gBACvB,OAAO,EACL,uGAAuG;aAC1G,CAAC,CACH;YACH,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;gBACb,OAAO,EAAE,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC;gBAC1C,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;aACtC,CAAC,CAAA;IAEV,MAAM,WAAW,GAAG,CAAC,MAAc,EAAoC,EAAE,CAAC,CAAC;QACzE,WAAW,EAAE,MAAM;QACnB,mBAAmB,EAAE,YAAY;KAClC,CAAC,CAAA;IAEF,MAAM,cAAc,GAAG,CACrB,QAAyC,EACzC,MAAe,EAC+D,EAAE,CAChF,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC,cAAc,CAAA;QACjD,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;QACtE,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,QAAQ,CACpC,GAAG,OAAO,WAAW,EACrB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,WAAW,CAAC,MAAM,CAAC,EACnB,MAAM,CAAC,OAAO,CACf,CAAA;QACD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,CAAA;QAC3C,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAA;IACxB,CAAC,CAAC,CAAA;IAEJ,MAAM,oBAAoB,GAAG,CAC3B,QAAyC,EACzC,MAAe,EACoB,EAAE,CACrC,MAAM,CAAC,MAAM,CACX,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;QACjD,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAA;QACrE,OAAO,UAAU;aACd,iBAAiB,CAChB,GAAG,OAAO,WAAW,EACrB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,WAAW,CAAC,MAAM,CAAC,EACnB,MAAM,CAAC,OAAO,CACf;aACA,IAAI,CACH,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,EACnC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACvB,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,IAAI,EAAE,CAAA;gBACrC,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;oBACvB,CAAC,CAAC,MAAM,CAAC,KAAK;oBACd,CAAC,CAAC,MAAM,CAAC,OAAO,CACZ,QAAQ,CAAC,IAAI,CAAC;wBACZ,KAAK;wBACL,QAAQ,EAAE;4BACR,QAAQ,EAAE,WAAW;4BACrB,KAAK,EAAE,MAAM,CAAC,KAAK;yBACpB;qBACF,CAAC,CACH,CAAA;YACP,CAAC;YACD,MAAM,YAAY,GAChB,KAAK,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YACtF,OAAO,YAAY,KAAK,SAAS;gBAC/B,CAAC,CAAC,MAAM,CAAC,KAAK;gBACd,CAAC,CAAC,MAAM,CAAC,OAAO,CACZ,QAAQ,CAAC,IAAI,CAAC;oBACZ,KAAK,EAAE,EAAE;oBACT,YAAY;oBACZ,QAAQ,EAAE;wBACR,QAAQ,EAAE,WAAW;wBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;qBACpB;iBACF,CAAC,CACH,CAAA;QACP,CAAC,CAAC,CACH,CAAA;IACL,CAAC,CAAC,CACH,CAAA;IAEH,MAAM,0BAA0B,GAAG,CACjC,MAAc,EACd,MAA4C,EAC5C,UAAsB,EAC4B,EAAE,CACpD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,cAAc,CAAC;YAC5C,gBAAgB,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,GAAG,MAAM,MAAM,GAAG,8DAA8D;aAC1F,CAAC;SACH,CAAC,CAAA;QACF,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,mBAAmB,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAA;QACzD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,CAAA;QAC/D,MAAM,MAAM,GAAwB,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAA;QACjE,OAAO,MAAM,CAAA;IACf,CAAC,CAAC,CAAA;IAEJ,MAAM,gBAAgB,GAAG,CACvB,MAAc,EACd,KAAoC,EACO,EAAE,CAC7C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC,cAAc,CAAA;QACjD,MAAM,OAAO,GAAG,yBAAyB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,CAAA;QAChE,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,QAAQ,CACpC,GAAG,OAAO,WAAW,EACrB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,WAAW,CAAC,MAAM,CAAC,EACnB,MAAM,CAAC,OAAO,CACf,CAAA;QACD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAC/C,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO;aAC/B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC;aAC5C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACb,QAAQ,CAAC,IAAI,CAAC;YACZ,EAAE,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE;YAClB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;YACtB,SAAS,EAAE,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC;SACtE,CAAC,CACH,CAAA;QACH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CACnC,CAAC,KAAK,EAAsC,EAAE,CAC5C,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,CAC3E,EAAE,IAAI,CAAA;QAEP,OAAO,gBAAgB,CAAC,IAAI,CAAC;YAC3B,SAAS;YACT,YAAY,EAAE,QAAQ,CAAC,WAAW,IAAI,MAAM;YAC5C,GAAG,CAAC,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC;SAClE,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEJ,MAAM,aAAa,GAAG,CAAC,MAAc,EAAqC,EAAE,CAC1E,oBAAoB,CAAC;QACnB,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM;SAChB,CAAC;KACH,CAAC,CAAA;IAEJ,MAAM,WAAW,GAA2B,OAAO,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CACrF,MAAM,CAAC,KAAK,CAAC;QACX,SAAS,EAAE,GAAG,EAAE,CAAC,KAAK;QACtB,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI;KACtB,CAAC,CACH,CAAA;IAED,OAAO,gBAAgB,CAAC;QACtB,EAAE,EAAE,YAAY,CAAC,SAAS;QAC1B,aAAa;QACb,wBAAwB,EAAE,CAAC,QAAQ,EAAE,EAAE;YACrC,MAAM,OAAO,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAA;YAC1C,OAAO,oBAAoB,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QAC/D,CAAC;QACD,gBAAgB;QAChB,0BAA0B;QAC1B,WAAW;KACZ,CAAC,CAAA;AACJ,CAAC,CAAA"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { type CliConnectorShape } from "../Connector.ts";
|
|
2
|
+
import type { CliConnectorConfig } from "../ConnectorConfig.ts";
|
|
3
|
+
import type { ProcessExecutorShape } from "../ProcessExecutor.ts";
|
|
4
|
+
export declare const antigravityExtraArgs: (config: CliConnectorConfig) => ReadonlyArray<string>;
|
|
5
|
+
export declare const buildAntigravityArgv: (config: CliConnectorConfig, prompt: string) => ReadonlyArray<string>;
|
|
6
|
+
export declare const makeAntigravityConnector: (config: CliConnectorConfig, executor: ProcessExecutorShape) => CliConnectorShape;
|
|
7
|
+
//# sourceMappingURL=AntigravityConnector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AntigravityConnector.d.ts","sourceRoot":"","sources":["../../src/providers/AntigravityConnector.ts"],"names":[],"mappings":"AAEA,OAAO,EAAoB,KAAK,iBAAiB,EAAE,MAAM,iBAAiB,CAAA;AAC1E,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAA;AAG/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAA;AAajE,eAAO,MAAM,oBAAoB,GAAI,QAAQ,kBAAkB,KAAG,aAAa,CAAC,MAAM,CAUrF,CAAA;AAED,eAAO,MAAM,oBAAoB,GAC/B,QAAQ,kBAAkB,EAC1B,QAAQ,MAAM,KACb,aAAa,CAAC,MAAM,CAAgE,CAAA;AAEvF,eAAO,MAAM,wBAAwB,GACnC,QAAQ,kBAAkB,EAC1B,UAAU,oBAAoB,KAC7B,iBAqCF,CAAA"}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
import * as Effect from "effect/Effect";
|
|
2
|
+
import * as Stream from "effect/Stream";
|
|
3
|
+
import { makeCliConnector } from "../Connector.js";
|
|
4
|
+
import { ProviderError } from "../Errors.js";
|
|
5
|
+
import { ConnectorIds, HealthStatus, LlmChunk } from "../Models.js";
|
|
6
|
+
import { optionalModelArgs, sortedFlagArgs } from "./CliSupport.js";
|
|
7
|
+
const healthy = HealthStatus.make({
|
|
8
|
+
availability: "Healthy",
|
|
9
|
+
authStatus: "Valid"
|
|
10
|
+
});
|
|
11
|
+
const unhealthy = HealthStatus.make({
|
|
12
|
+
availability: "Unhealthy",
|
|
13
|
+
authStatus: "Unknown"
|
|
14
|
+
});
|
|
15
|
+
export const antigravityExtraArgs = (config) => {
|
|
16
|
+
const workspaceArgs = config.workingDir === undefined ? [] : ["--add-dir", config.workingDir];
|
|
17
|
+
const effectiveFlags = config.readOnly
|
|
18
|
+
? {
|
|
19
|
+
...config.flags,
|
|
20
|
+
mode: "plan"
|
|
21
|
+
}
|
|
22
|
+
: config.flags;
|
|
23
|
+
return [...workspaceArgs, ...optionalModelArgs(config.model), ...sortedFlagArgs(effectiveFlags)];
|
|
24
|
+
};
|
|
25
|
+
export const buildAntigravityArgv = (config, prompt) => ["agy", ...antigravityExtraArgs(config), "--print", prompt];
|
|
26
|
+
export const makeAntigravityConnector = (config, executor) => {
|
|
27
|
+
const cwd = config.workingDir ?? ".";
|
|
28
|
+
const healthCheck = executor.run(["agy", "--version"], cwd, {}).pipe(Effect.map((result) => (result.exitCode === 0 ? healthy : unhealthy)), Effect.catch(() => Effect.succeed(unhealthy)));
|
|
29
|
+
const complete = Effect.fn("@llm4ts/core/providers/AntigravityConnector.complete")(function* (prompt) {
|
|
30
|
+
const result = yield* executor.run(buildAntigravityArgv(config, prompt), cwd, config.envVars);
|
|
31
|
+
if (result.exitCode !== 0) {
|
|
32
|
+
const detail = [...result.stderr, ...result.stdout].join("\n");
|
|
33
|
+
return yield* ProviderError.make({
|
|
34
|
+
message: `agy exited with code ${result.exitCode}: ${detail}`
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
return result.stdout.join("\n");
|
|
38
|
+
});
|
|
39
|
+
const completeStream = (prompt) => Stream.concat(executor
|
|
40
|
+
.runStreaming(buildAntigravityArgv(config, prompt), cwd, config.envVars)
|
|
41
|
+
.pipe(Stream.map((line) => LlmChunk.make({ delta: line }))), Stream.make(LlmChunk.make({ delta: "", finishReason: "stop" })));
|
|
42
|
+
return makeCliConnector({
|
|
43
|
+
id: ConnectorIds.AntigravityCli,
|
|
44
|
+
interactionSupport: "InteractiveStdin",
|
|
45
|
+
buildArgv: (prompt, _context) => buildAntigravityArgv(config, prompt),
|
|
46
|
+
buildInteractiveArgv: (_context) => ["agy", ...antigravityExtraArgs(config)],
|
|
47
|
+
complete,
|
|
48
|
+
completeStream,
|
|
49
|
+
healthCheck
|
|
50
|
+
});
|
|
51
|
+
};
|
|
52
|
+
//# sourceMappingURL=AntigravityConnector.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AntigravityConnector.js","sourceRoot":"","sources":["../../src/providers/AntigravityConnector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,EAAE,gBAAgB,EAA0B,MAAM,iBAAiB,CAAA;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAA;AAC5C,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAA;AAEnE,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAEnE,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC;IAChC,YAAY,EAAE,SAAS;IACvB,UAAU,EAAE,OAAO;CACpB,CAAC,CAAA;AAEF,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC;IAClC,YAAY,EAAE,WAAW;IACzB,UAAU,EAAE,SAAS;CACtB,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,MAA0B,EAAyB,EAAE;IACxF,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC,CAAA;IAC7F,MAAM,cAAc,GAAG,MAAM,CAAC,QAAQ;QACpC,CAAC,CAAC;YACE,GAAG,MAAM,CAAC,KAAK;YACf,IAAI,EAAE,MAAM;SACb;QACH,CAAC,CAAC,MAAM,CAAC,KAAK,CAAA;IAEhB,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,cAAc,CAAC,cAAc,CAAC,CAAC,CAAA;AAClG,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAClC,MAA0B,EAC1B,MAAc,EACS,EAAE,CAAC,CAAC,KAAK,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAA;AAEvF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACtC,MAA0B,EAC1B,QAA8B,EACX,EAAE;IACrB,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,GAAG,CAAA;IACpC,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,WAAW,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,IAAI,CAClE,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EACrE,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAC9C,CAAA;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,EAAE,CAAC,sDAAsD,CAAC,CAAC,QAAQ,CAAC,EAC1F,MAAc;QAEd,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,OAAO,CAAC,CAAA;QAC7F,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC9D,OAAO,KAAK,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC/B,OAAO,EAAE,wBAAwB,MAAM,CAAC,QAAQ,KAAK,MAAM,EAAE;aAC9D,CAAC,CAAA;QACJ,CAAC;QACD,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,MAAM,cAAc,GAAG,CAAC,MAAc,EAAE,EAAE,CACxC,MAAM,CAAC,MAAM,CACX,QAAQ;SACL,YAAY,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,OAAO,CAAC;SACvE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAC7D,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC,CAChE,CAAA;IAEH,OAAO,gBAAgB,CAAC;QACtB,EAAE,EAAE,YAAY,CAAC,cAAc;QAC/B,kBAAkB,EAAE,kBAAkB;QACtC,SAAS,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC;QACrE,oBAAoB,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAC5E,QAAQ;QACR,cAAc;QACd,WAAW;KACZ,CAAC,CAAA;AACJ,CAAC,CAAA"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import * as Effect from "effect/Effect";
|
|
2
|
+
import type * as Scope from "effect/Scope";
|
|
3
|
+
import { type AgentSessionShape, type SessionEvent } from "../AgentSession.ts";
|
|
4
|
+
import { ProviderError } from "../Errors.ts";
|
|
5
|
+
import type { ProcessExecutorShape } from "../ProcessExecutor.ts";
|
|
6
|
+
export declare const claudeSessionArgv: (model: string | undefined, extraFlags: ReadonlyArray<string>) => ReadonlyArray<string>;
|
|
7
|
+
export declare const claudeUserMessageJson: (text: string) => string;
|
|
8
|
+
export declare const parseClaudeSessionLine: (line: string) => ReadonlyArray<SessionEvent>;
|
|
9
|
+
export declare const claudeSessionInitModel: (line: string) => string | undefined;
|
|
10
|
+
export declare const openClaudeAgentSession: (executor: ProcessExecutorShape, argv: readonly string[], cwd: string, envVars?: Readonly<Record<string, string>> | undefined) => Effect.Effect<AgentSessionShape, ProviderError | import("../Errors.ts").UsageLimitError | import("../Errors.ts").AuthenticationError | import("../Errors.ts").InvalidRequestError | import("../Errors.ts").ParseError | import("../Errors.ts").ToolError | import("../Errors.ts").ConfigError | import("../Errors.ts").RateLimitError | import("../Errors.ts").TimeoutError | import("../Errors.ts").TurnLimitError, Scope.Scope>;
|
|
11
|
+
//# sourceMappingURL=ClaudeAgentSession.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ClaudeAgentSession.d.ts","sourceRoot":"","sources":["../../src/providers/ClaudeAgentSession.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AAIvC,OAAO,KAAK,KAAK,KAAK,MAAM,cAAc,CAAA;AAE1C,OAAO,EAOL,KAAK,iBAAiB,EACtB,KAAK,YAAY,EAClB,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAAE,aAAa,EAAiB,MAAM,cAAc,CAAA;AAE3D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAA;AAYjE,eAAO,MAAM,iBAAiB,GAC5B,OAAO,MAAM,GAAG,SAAS,EACzB,YAAY,aAAa,CAAC,MAAM,CAAC,KAChC,aAAa,CAAC,MAAM,CAUtB,CAAA;AAED,eAAO,MAAM,qBAAqB,GAAI,MAAM,MAAM,KAAG,MAOjD,CAAA;AAsEJ,eAAO,MAAM,sBAAsB,GAAI,MAAM,MAAM,KAAG,aAAa,CAAC,YAAY,CAe/E,CAAA;AAED,eAAO,MAAM,sBAAsB,GAAI,MAAM,MAAM,KAAG,MAAM,GAAG,SAK9D,CAAA;AAED,eAAO,MAAM,sBAAsB,qiBA8ElC,CAAA"}
|
|
@@ -0,0 +1,158 @@
|
|
|
1
|
+
import * as Deferred from "effect/Deferred";
|
|
2
|
+
import * as Effect from "effect/Effect";
|
|
3
|
+
import * as PubSub from "effect/PubSub";
|
|
4
|
+
import * as Queue from "effect/Queue";
|
|
5
|
+
import * as Ref from "effect/Ref";
|
|
6
|
+
import * as Stream from "effect/Stream";
|
|
7
|
+
import { SessionDone, SessionResult, SessionTextDelta, SessionToolResult, SessionToolUse, SessionUsage } from "../AgentSession.js";
|
|
8
|
+
import { ProviderError } from "../Errors.js";
|
|
9
|
+
import { TokenUsage } from "../Models.js";
|
|
10
|
+
import { jsonArray, jsonBooleanField, jsonField, jsonIntField, jsonStringField, jsonText, parseJsonLine } from "./CliSupport.js";
|
|
11
|
+
export const claudeSessionArgv = (model, extraFlags) => [
|
|
12
|
+
"claude",
|
|
13
|
+
"--print",
|
|
14
|
+
"--input-format",
|
|
15
|
+
"stream-json",
|
|
16
|
+
"--output-format",
|
|
17
|
+
"stream-json",
|
|
18
|
+
"--verbose",
|
|
19
|
+
...(model === undefined ? [] : ["--model", model]),
|
|
20
|
+
...extraFlags
|
|
21
|
+
];
|
|
22
|
+
export const claudeUserMessageJson = (text) => JSON.stringify({
|
|
23
|
+
type: "user",
|
|
24
|
+
message: {
|
|
25
|
+
role: "user",
|
|
26
|
+
content: text
|
|
27
|
+
}
|
|
28
|
+
});
|
|
29
|
+
const contentBlocks = (json) => jsonArray(jsonField(jsonField(json, "message"), "content"));
|
|
30
|
+
const assistantEvents = (json) => contentBlocks(json).flatMap((block) => {
|
|
31
|
+
switch (jsonStringField(block, "type")) {
|
|
32
|
+
case "text": {
|
|
33
|
+
const text = jsonStringField(block, "text");
|
|
34
|
+
return text === undefined || text.length === 0 ? [] : [SessionTextDelta.make({ text })];
|
|
35
|
+
}
|
|
36
|
+
case "tool_use":
|
|
37
|
+
return [
|
|
38
|
+
SessionToolUse.make({
|
|
39
|
+
tool: jsonStringField(block, "name") ?? "",
|
|
40
|
+
rawInput: jsonText(jsonField(block, "input") ?? {}),
|
|
41
|
+
id: jsonStringField(block, "id") ?? ""
|
|
42
|
+
})
|
|
43
|
+
];
|
|
44
|
+
default:
|
|
45
|
+
return [];
|
|
46
|
+
}
|
|
47
|
+
});
|
|
48
|
+
const toolResultEvents = (json) => contentBlocks(json).flatMap((block) => {
|
|
49
|
+
if (jsonStringField(block, "type") !== "tool_result") {
|
|
50
|
+
return [];
|
|
51
|
+
}
|
|
52
|
+
const content = jsonField(block, "content");
|
|
53
|
+
return [
|
|
54
|
+
SessionToolResult.make({
|
|
55
|
+
id: jsonStringField(block, "tool_use_id") ?? "",
|
|
56
|
+
status: jsonBooleanField(block, "is_error") === true ? "error" : "success",
|
|
57
|
+
content: content === undefined ? "" : jsonText(content)
|
|
58
|
+
})
|
|
59
|
+
];
|
|
60
|
+
});
|
|
61
|
+
const resultEvents = (json) => {
|
|
62
|
+
const usage = jsonField(json, "usage");
|
|
63
|
+
const usageEvents = usage === undefined
|
|
64
|
+
? []
|
|
65
|
+
: [
|
|
66
|
+
SessionUsage.make({
|
|
67
|
+
usage: TokenUsage.make({
|
|
68
|
+
prompt: jsonIntField(usage, "input_tokens") ?? 0,
|
|
69
|
+
completion: jsonIntField(usage, "output_tokens") ?? 0,
|
|
70
|
+
total: (jsonIntField(usage, "input_tokens") ?? 0) +
|
|
71
|
+
(jsonIntField(usage, "output_tokens") ?? 0),
|
|
72
|
+
...(jsonIntField(usage, "cache_read_input_tokens") === undefined
|
|
73
|
+
? {}
|
|
74
|
+
: {
|
|
75
|
+
cached: jsonIntField(usage, "cache_read_input_tokens") ?? 0
|
|
76
|
+
})
|
|
77
|
+
})
|
|
78
|
+
})
|
|
79
|
+
];
|
|
80
|
+
return [
|
|
81
|
+
...usageEvents,
|
|
82
|
+
SessionDone.make({
|
|
83
|
+
result: jsonStringField(json, "result") ?? ""
|
|
84
|
+
})
|
|
85
|
+
];
|
|
86
|
+
};
|
|
87
|
+
export const parseClaudeSessionLine = (line) => {
|
|
88
|
+
const json = parseJsonLine(line);
|
|
89
|
+
if (json === undefined) {
|
|
90
|
+
return [];
|
|
91
|
+
}
|
|
92
|
+
switch (jsonStringField(json, "type")) {
|
|
93
|
+
case "assistant":
|
|
94
|
+
return assistantEvents(json);
|
|
95
|
+
case "user":
|
|
96
|
+
return toolResultEvents(json);
|
|
97
|
+
case "result":
|
|
98
|
+
return resultEvents(json);
|
|
99
|
+
default:
|
|
100
|
+
return [];
|
|
101
|
+
}
|
|
102
|
+
};
|
|
103
|
+
export const claudeSessionInitModel = (line) => {
|
|
104
|
+
const json = parseJsonLine(line);
|
|
105
|
+
return json !== undefined && jsonStringField(json, "type") === "system"
|
|
106
|
+
? jsonStringField(json, "model")
|
|
107
|
+
: undefined;
|
|
108
|
+
};
|
|
109
|
+
export const openClaudeAgentSession = Effect.fn("@llm4ts/core/providers/ClaudeAgentSession.open")(function* (executor, argv, cwd, envVars = {}) {
|
|
110
|
+
const [stdin, stdout] = yield* executor.runBidirectional(argv, cwd, envVars);
|
|
111
|
+
const events = yield* PubSub.unbounded();
|
|
112
|
+
const done = yield* Deferred.make();
|
|
113
|
+
const model = yield* Ref.make(undefined);
|
|
114
|
+
const text = yield* Ref.make("");
|
|
115
|
+
const usage = yield* Ref.make(undefined);
|
|
116
|
+
const trackEvent = Effect.fn("@llm4ts/core/providers/ClaudeAgentSession.trackEvent")(function* (event) {
|
|
117
|
+
switch (event._tag) {
|
|
118
|
+
case "TextDelta":
|
|
119
|
+
yield* Ref.update(text, (current) => current + event.text);
|
|
120
|
+
break;
|
|
121
|
+
case "Usage":
|
|
122
|
+
yield* Ref.set(usage, event.usage);
|
|
123
|
+
break;
|
|
124
|
+
case "Done": {
|
|
125
|
+
const accumulated = yield* Ref.get(text);
|
|
126
|
+
const finalUsage = yield* Ref.get(usage);
|
|
127
|
+
const finalModel = yield* Ref.get(model);
|
|
128
|
+
yield* Deferred.succeed(done, SessionResult.make({
|
|
129
|
+
text: event.result.length === 0 ? accumulated : event.result,
|
|
130
|
+
...(finalUsage === undefined ? {} : { usage: finalUsage }),
|
|
131
|
+
...(finalModel === undefined ? {} : { model: finalModel })
|
|
132
|
+
}));
|
|
133
|
+
break;
|
|
134
|
+
}
|
|
135
|
+
default:
|
|
136
|
+
break;
|
|
137
|
+
}
|
|
138
|
+
yield* PubSub.publish(events, event);
|
|
139
|
+
});
|
|
140
|
+
yield* stdout.pipe(Stream.mapEffect((line) => {
|
|
141
|
+
const initializedModel = claudeSessionInitModel(line);
|
|
142
|
+
return initializedModel === undefined
|
|
143
|
+
? Effect.succeed(line)
|
|
144
|
+
: Ref.set(model, initializedModel).pipe(Effect.as(line));
|
|
145
|
+
}), Stream.flatMap((line) => Stream.fromIterable(parseClaudeSessionLine(line))), Stream.mapEffect(trackEvent), Stream.runDrain, Effect.catch((error) => Deferred.fail(done, error).pipe(Effect.asVoid)), Effect.ensuring(Deferred.fail(done, ProviderError.make({
|
|
146
|
+
message: "claude session ended before a result"
|
|
147
|
+
})).pipe(Effect.asVoid)), Effect.forkScoped);
|
|
148
|
+
const sendUserMessage = (message) => Queue.offer(stdin, claudeUserMessageJson(message)).pipe(Effect.asVoid);
|
|
149
|
+
return {
|
|
150
|
+
events: Stream.fromPubSub(events),
|
|
151
|
+
sendUserMessage,
|
|
152
|
+
respondToAsk: sendUserMessage,
|
|
153
|
+
respondToApproval: (_id, _approved) => Effect.void,
|
|
154
|
+
awaitResult: Deferred.await(done),
|
|
155
|
+
cancel: Queue.shutdown(stdin)
|
|
156
|
+
};
|
|
157
|
+
});
|
|
158
|
+
//# sourceMappingURL=ClaudeAgentSession.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ClaudeAgentSession.js","sourceRoot":"","sources":["../../src/providers/ClaudeAgentSession.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,iBAAiB,CAAA;AAC3C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,KAAK,MAAM,cAAc,CAAA;AACrC,OAAO,KAAK,GAAG,MAAM,YAAY,CAAA;AAEjC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,EACL,WAAW,EACX,aAAa,EACb,gBAAgB,EAChB,iBAAiB,EACjB,cAAc,EACd,YAAY,EAGb,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAAE,aAAa,EAAiB,MAAM,cAAc,CAAA;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAA;AAEzC,OAAO,EACL,SAAS,EACT,gBAAgB,EAChB,SAAS,EACT,YAAY,EACZ,eAAe,EACf,QAAQ,EACR,aAAa,EAEd,MAAM,iBAAiB,CAAA;AAExB,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,KAAyB,EACzB,UAAiC,EACV,EAAE,CAAC;IAC1B,QAAQ;IACR,SAAS;IACT,gBAAgB;IAChB,aAAa;IACb,iBAAiB;IACjB,aAAa;IACb,WAAW;IACX,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAClD,GAAG,UAAU;CACd,CAAA;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,IAAY,EAAU,EAAE,CAC5D,IAAI,CAAC,SAAS,CAAC;IACb,IAAI,EAAE,MAAM;IACZ,OAAO,EAAE;QACP,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,IAAI;KACd;CACF,CAAC,CAAA;AAEJ,MAAM,aAAa,GAAG,CAAC,IAAe,EAA4B,EAAE,CAClE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC,CAAA;AAE7D,MAAM,eAAe,GAAG,CAAC,IAAe,EAA+B,EAAE,CACvE,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAA+B,EAAE;IACjE,QAAQ,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,IAAI,GAAG,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;YAC3C,OAAO,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QACzF,CAAC;QACD,KAAK,UAAU;YACb,OAAO;gBACL,cAAc,CAAC,IAAI,CAAC;oBAClB,IAAI,EAAE,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE;oBAC1C,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;oBACnD,EAAE,EAAE,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE;iBACvC,CAAC;aACH,CAAA;QACH;YACE,OAAO,EAAE,CAAA;IACb,CAAC;AACH,CAAC,CAAC,CAAA;AAEJ,MAAM,gBAAgB,GAAG,CAAC,IAAe,EAA+B,EAAE,CACxE,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAA+B,EAAE;IACjE,IAAI,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,aAAa,EAAE,CAAC;QACrD,OAAO,EAAE,CAAA;IACX,CAAC;IACD,MAAM,OAAO,GAAG,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,CAAA;IAC3C,OAAO;QACL,iBAAiB,CAAC,IAAI,CAAC;YACrB,EAAE,EAAE,eAAe,CAAC,KAAK,EAAE,aAAa,CAAC,IAAI,EAAE;YAC/C,MAAM,EAAE,gBAAgB,CAAC,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;YAC1E,OAAO,EAAE,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;SACxD,CAAC;KACH,CAAA;AACH,CAAC,CAAC,CAAA;AAEJ,MAAM,YAAY,GAAG,CAAC,IAAe,EAA+B,EAAE;IACpE,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;IACtC,MAAM,WAAW,GACf,KAAK,KAAK,SAAS;QACjB,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC;YACE,YAAY,CAAC,IAAI,CAAC;gBAChB,KAAK,EAAE,UAAU,CAAC,IAAI,CAAC;oBACrB,MAAM,EAAE,YAAY,CAAC,KAAK,EAAE,cAAc,CAAC,IAAI,CAAC;oBAChD,UAAU,EAAE,YAAY,CAAC,KAAK,EAAE,eAAe,CAAC,IAAI,CAAC;oBACrD,KAAK,EACH,CAAC,YAAY,CAAC,KAAK,EAAE,cAAc,CAAC,IAAI,CAAC,CAAC;wBAC1C,CAAC,YAAY,CAAC,KAAK,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;oBAC7C,GAAG,CAAC,YAAY,CAAC,KAAK,EAAE,yBAAyB,CAAC,KAAK,SAAS;wBAC9D,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC;4BACE,MAAM,EAAE,YAAY,CAAC,KAAK,EAAE,yBAAyB,CAAC,IAAI,CAAC;yBAC5D,CAAC;iBACP,CAAC;aACH,CAAC;SACH,CAAA;IAEP,OAAO;QACL,GAAG,WAAW;QACd,WAAW,CAAC,IAAI,CAAC;YACf,MAAM,EAAE,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE;SAC9C,CAAC;KACH,CAAA;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,IAAY,EAA+B,EAAE;IAClF,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAA;IAChC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,EAAE,CAAA;IACX,CAAC;IACD,QAAQ,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC;QACtC,KAAK,WAAW;YACd,OAAO,eAAe,CAAC,IAAI,CAAC,CAAA;QAC9B,KAAK,MAAM;YACT,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAA;QAC/B,KAAK,QAAQ;YACX,OAAO,YAAY,CAAC,IAAI,CAAC,CAAA;QAC3B;YACE,OAAO,EAAE,CAAA;IACb,CAAC;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,IAAY,EAAsB,EAAE;IACzE,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAA;IAChC,OAAO,IAAI,KAAK,SAAS,IAAI,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,QAAQ;QACrE,CAAC,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC;QAChC,CAAC,CAAC,SAAS,CAAA;AACf,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,MAAM,CAAC,EAAE,CAAC,gDAAgD,CAAC,CAC/F,QAAQ,CAAC,EACP,QAA8B,EAC9B,IAA2B,EAC3B,GAAW,EACX,UAA4C,EAAE;IAE9C,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC5E,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,SAAS,EAAgB,CAAA;IACtD,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,EAA2B,CAAA;IAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAqB,SAAS,CAAC,CAAA;IAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAChC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAyB,SAAS,CAAC,CAAA;IAEhE,MAAM,UAAU,GAAG,MAAM,CAAC,EAAE,CAAC,sDAAsD,CAAC,CAAC,QAAQ,CAAC,EAC5F,KAAmB;QAEnB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,WAAW;gBACd,KAAK,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAA;gBAC1D,MAAK;YACP,KAAK,OAAO;gBACV,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;gBAClC,MAAK;YACP,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBACxC,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;gBACxC,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;gBACxC,KAAK,CAAC,CAAC,QAAQ,CAAC,OAAO,CACrB,IAAI,EACJ,aAAa,CAAC,IAAI,CAAC;oBACjB,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM;oBAC5D,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;oBAC1D,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;iBAC3D,CAAC,CACH,CAAA;gBACD,MAAK;YACP,CAAC;YACD;gBACE,MAAK;QACT,CAAC;QACD,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAChB,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE;QACxB,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAA;QACrD,OAAO,gBAAgB,KAAK,SAAS;YACnC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;YACtB,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAA;IAC5D,CAAC,CAAC,EACF,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,EAC3E,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,EAC5B,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EACvE,MAAM,CAAC,QAAQ,CACb,QAAQ,CAAC,IAAI,CACX,IAAI,EACJ,aAAa,CAAC,IAAI,CAAC;QACjB,OAAO,EAAE,sCAAsC;KAChD,CAAC,CACH,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CACtB,EACD,MAAM,CAAC,UAAU,CAClB,CAAA;IAED,MAAM,eAAe,GAAG,CAAC,OAAe,EAAiC,EAAE,CACzE,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAExE,OAAO;QACL,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;QACjC,eAAe;QACf,YAAY,EAAE,eAAe;QAC7B,iBAAiB,EAAE,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI;QAClD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;QACjC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;KAC9B,CAAA;AACH,CAAC,CACF,CAAA"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { type CliConnectorShape } from "../Connector.ts";
|
|
2
|
+
import type { CliConnectorConfig } from "../ConnectorConfig.ts";
|
|
3
|
+
import { LlmChunk } from "../Models.ts";
|
|
4
|
+
import type { ProcessExecutorShape } from "../ProcessExecutor.ts";
|
|
5
|
+
export declare const claudeCliExtraArgs: (config: CliConnectorConfig) => ReadonlyArray<string>;
|
|
6
|
+
export declare const claudeCliInitModel: (line: string) => string | undefined;
|
|
7
|
+
export declare const parseClaudeCliStreamLine: (line: string) => ReadonlyArray<LlmChunk>;
|
|
8
|
+
export declare const makeClaudeCliConnector: (config: CliConnectorConfig, executor: ProcessExecutorShape) => CliConnectorShape;
|
|
9
|
+
//# sourceMappingURL=ClaudeCliConnector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ClaudeCliConnector.d.ts","sourceRoot":"","sources":["../../src/providers/ClaudeCliConnector.ts"],"names":[],"mappings":"AAIA,OAAO,EAAoB,KAAK,iBAAiB,EAAE,MAAM,iBAAiB,CAAA;AAC1E,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAA;AAG/D,OAAO,EAGL,QAAQ,EAGT,MAAM,cAAc,CAAA;AACrB,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAA;AAgBjE,eAAO,MAAM,kBAAkB,GAAI,QAAQ,kBAAkB,KAAG,aAAa,CAAC,MAAM,CASnF,CAAA;AAED,eAAO,MAAM,kBAAkB,GAAI,MAAM,MAAM,KAAG,MAAM,GAAG,SAK1D,CAAA;AAED,eAAO,MAAM,wBAAwB,GAAI,MAAM,MAAM,KAAG,aAAa,CAAC,QAAQ,CA+C7E,CAAA;AAED,eAAO,MAAM,sBAAsB,GACjC,QAAQ,kBAAkB,EAC1B,UAAU,oBAAoB,KAC7B,iBAmGF,CAAA"}
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
import * as Effect from "effect/Effect";
|
|
2
|
+
import * as Ref from "effect/Ref";
|
|
3
|
+
import * as Stream from "effect/Stream";
|
|
4
|
+
import { makeCliConnector } from "../Connector.js";
|
|
5
|
+
import { ConnectorCapabilities, ConnectorIds, LlmChunk, TokenUsage } from "../Models.js";
|
|
6
|
+
import { collect } from "../Streaming.js";
|
|
7
|
+
import { parseFromText, withSchemaHint } from "../StructuredOutput.js";
|
|
8
|
+
import { failClassifiedCliError, jsonArray, jsonField, jsonIntField, jsonStringField, optionalModelArgs, parseJsonLine, sortedFlagArgs, toolEventChunk, usageEventChunk } from "./CliSupport.js";
|
|
9
|
+
export const claudeCliExtraArgs = (config) => {
|
|
10
|
+
const effectiveFlags = config.readOnly
|
|
11
|
+
? {
|
|
12
|
+
...config.flags,
|
|
13
|
+
"permission-mode": "default",
|
|
14
|
+
"disallowed-tools": "Write,Edit,NotebookEdit"
|
|
15
|
+
}
|
|
16
|
+
: config.flags;
|
|
17
|
+
return [...optionalModelArgs(config.model), ...sortedFlagArgs(effectiveFlags)];
|
|
18
|
+
};
|
|
19
|
+
export const claudeCliInitModel = (line) => {
|
|
20
|
+
const json = parseJsonLine(line);
|
|
21
|
+
return json !== undefined && jsonStringField(json, "type") === "system"
|
|
22
|
+
? jsonStringField(json, "model")
|
|
23
|
+
: undefined;
|
|
24
|
+
};
|
|
25
|
+
export const parseClaudeCliStreamLine = (line) => {
|
|
26
|
+
const json = parseJsonLine(line);
|
|
27
|
+
if (json === undefined) {
|
|
28
|
+
return [];
|
|
29
|
+
}
|
|
30
|
+
switch (jsonStringField(json, "type")) {
|
|
31
|
+
case "assistant":
|
|
32
|
+
return jsonArray(jsonField(jsonField(json, "message"), "content")).flatMap((block) => {
|
|
33
|
+
switch (jsonStringField(block, "type")) {
|
|
34
|
+
case "text": {
|
|
35
|
+
const text = jsonStringField(block, "text");
|
|
36
|
+
return text === undefined || text.length === 0 ? [] : [LlmChunk.make({ delta: text })];
|
|
37
|
+
}
|
|
38
|
+
case "tool_use":
|
|
39
|
+
return [
|
|
40
|
+
toolEventChunk(jsonStringField(block, "name") ?? "", jsonField(block, "input"))
|
|
41
|
+
];
|
|
42
|
+
default:
|
|
43
|
+
return [];
|
|
44
|
+
}
|
|
45
|
+
});
|
|
46
|
+
case "result": {
|
|
47
|
+
const usage = jsonField(json, "usage");
|
|
48
|
+
if (usage === undefined) {
|
|
49
|
+
return [];
|
|
50
|
+
}
|
|
51
|
+
const prompt = jsonIntField(usage, "input_tokens") ?? 0;
|
|
52
|
+
const completion = jsonIntField(usage, "output_tokens") ?? 0;
|
|
53
|
+
const cached = jsonIntField(usage, "cache_read_input_tokens");
|
|
54
|
+
return [
|
|
55
|
+
usageEventChunk(undefined, TokenUsage.make({
|
|
56
|
+
prompt,
|
|
57
|
+
completion,
|
|
58
|
+
total: prompt + completion,
|
|
59
|
+
...(cached === undefined ? {} : { cached })
|
|
60
|
+
}))
|
|
61
|
+
];
|
|
62
|
+
}
|
|
63
|
+
default:
|
|
64
|
+
return [];
|
|
65
|
+
}
|
|
66
|
+
};
|
|
67
|
+
export const makeClaudeCliConnector = (config, executor) => {
|
|
68
|
+
const cwd = config.workingDir ?? ".";
|
|
69
|
+
const extraArgs = claudeCliExtraArgs(config);
|
|
70
|
+
const complete = Effect.fn("@llm4ts/core/providers/ClaudeCliConnector.complete")(function* (prompt) {
|
|
71
|
+
const result = yield* executor.runWithStdin(["claude", "--print", ...extraArgs], cwd, config.envVars, prompt);
|
|
72
|
+
if (result.exitCode !== 0) {
|
|
73
|
+
const raw = result.stdout.join("\n");
|
|
74
|
+
return yield* failClassifiedCliError("claude", `claude exited with code ${result.exitCode}`, raw);
|
|
75
|
+
}
|
|
76
|
+
return result.stdout.join("\n");
|
|
77
|
+
});
|
|
78
|
+
const completeStream = (prompt) => Stream.unwrap(Effect.map(Ref.make(undefined), (model) => executor
|
|
79
|
+
.runStreamingWithStdin(["claude", "--print", "--output-format", "stream-json", "--verbose", ...extraArgs], cwd, config.envVars, prompt)
|
|
80
|
+
.pipe(Stream.mapEffect((line) => {
|
|
81
|
+
const initializedModel = claudeCliInitModel(line);
|
|
82
|
+
return initializedModel === undefined
|
|
83
|
+
? Effect.succeed(line)
|
|
84
|
+
: Ref.set(model, initializedModel).pipe(Effect.as(line));
|
|
85
|
+
}), Stream.flatMap((line) => Stream.fromIterable(parseClaudeCliStreamLine(line))), Stream.mapEffect((chunk) => chunk.usage === undefined
|
|
86
|
+
? Effect.succeed(chunk)
|
|
87
|
+
: Effect.map(Ref.get(model), (servedModel) => servedModel === undefined
|
|
88
|
+
? chunk
|
|
89
|
+
: LlmChunk.make({
|
|
90
|
+
...chunk,
|
|
91
|
+
metadata: {
|
|
92
|
+
...chunk.metadata,
|
|
93
|
+
model: servedModel
|
|
94
|
+
}
|
|
95
|
+
}))))));
|
|
96
|
+
const base = makeCliConnector({
|
|
97
|
+
id: ConnectorIds.ClaudeCli,
|
|
98
|
+
interactionSupport: "InteractiveStdin",
|
|
99
|
+
capabilities: ConnectorCapabilities.make({
|
|
100
|
+
interactiveSessions: true,
|
|
101
|
+
askUser: true,
|
|
102
|
+
approval: true,
|
|
103
|
+
resumableSessions: true
|
|
104
|
+
}),
|
|
105
|
+
buildArgv: (prompt, _context) => ["claude", "--print", ...extraArgs, prompt],
|
|
106
|
+
buildInteractiveArgv: (_context) => ["claude", ...extraArgs],
|
|
107
|
+
complete,
|
|
108
|
+
completeStream,
|
|
109
|
+
versionProbe: { executor, binary: "claude", cwd }
|
|
110
|
+
});
|
|
111
|
+
const executeStructuredWithUsage = (prompt, schema, jsonSchema) => Effect.gen(function* () {
|
|
112
|
+
const reply = yield* collect(completeStream(withSchemaHint(prompt, jsonSchema)));
|
|
113
|
+
const value = yield* parseFromText(reply.content, schema, jsonSchema);
|
|
114
|
+
const result = [value, reply.usage, reply.metadata.model];
|
|
115
|
+
return result;
|
|
116
|
+
});
|
|
117
|
+
return {
|
|
118
|
+
...base,
|
|
119
|
+
executeStructured: (prompt, schema, jsonSchema) => Effect.map(executeStructuredWithUsage(prompt, schema, jsonSchema), ([value]) => value),
|
|
120
|
+
executeStructuredWithUsage
|
|
121
|
+
};
|
|
122
|
+
};
|
|
123
|
+
//# sourceMappingURL=ClaudeCliConnector.js.map
|