@roarpeng/graphflow 0.6.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/AGENTS.md +58 -0
- package/CHANGELOG.md +205 -0
- package/CLAUDE.md +39 -0
- package/LICENSE +17 -0
- package/README.md +434 -0
- package/dist/agents/brainstormer.d.ts +5 -0
- package/dist/agents/brainstormer.d.ts.map +1 -0
- package/dist/agents/brainstormer.js +61 -0
- package/dist/agents/brainstormer.js.map +1 -0
- package/dist/agents/planner.d.ts +14 -0
- package/dist/agents/planner.d.ts.map +1 -0
- package/dist/agents/planner.js +166 -0
- package/dist/agents/planner.js.map +1 -0
- package/dist/agents/validator.d.ts +6 -0
- package/dist/agents/validator.d.ts.map +1 -0
- package/dist/agents/validator.js +133 -0
- package/dist/agents/validator.js.map +1 -0
- package/dist/agents/worker.d.ts +18 -0
- package/dist/agents/worker.d.ts.map +1 -0
- package/dist/agents/worker.js +34 -0
- package/dist/agents/worker.js.map +1 -0
- package/dist/config/defaults.d.ts +16 -0
- package/dist/config/defaults.d.ts.map +1 -0
- package/dist/config/defaults.js +85 -0
- package/dist/config/defaults.js.map +1 -0
- package/dist/config/embedding-factory.d.ts +4 -0
- package/dist/config/embedding-factory.d.ts.map +1 -0
- package/dist/config/embedding-factory.js +37 -0
- package/dist/config/embedding-factory.js.map +1 -0
- package/dist/config/loader.d.ts +4 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +194 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/merge.d.ts +6 -0
- package/dist/config/merge.d.ts.map +1 -0
- package/dist/config/merge.js +130 -0
- package/dist/config/merge.js.map +1 -0
- package/dist/config/paths.d.ts +6 -0
- package/dist/config/paths.d.ts.map +1 -0
- package/dist/config/paths.js +32 -0
- package/dist/config/paths.js.map +1 -0
- package/dist/config/resolve.d.ts +4 -0
- package/dist/config/resolve.d.ts.map +1 -0
- package/dist/config/resolve.js +45 -0
- package/dist/config/resolve.js.map +1 -0
- package/dist/config/scaffold.d.ts +11 -0
- package/dist/config/scaffold.d.ts.map +1 -0
- package/dist/config/scaffold.js +42 -0
- package/dist/config/scaffold.js.map +1 -0
- package/dist/config/schema.d.ts +102 -0
- package/dist/config/schema.d.ts.map +1 -0
- package/dist/config/schema.js +3 -0
- package/dist/config/schema.js.map +1 -0
- package/dist/config/secrets.d.ts +9 -0
- package/dist/config/secrets.d.ts.map +1 -0
- package/dist/config/secrets.js +55 -0
- package/dist/config/secrets.js.map +1 -0
- package/dist/core/dag-engine.d.ts +14 -0
- package/dist/core/dag-engine.d.ts.map +1 -0
- package/dist/core/dag-engine.js +97 -0
- package/dist/core/dag-engine.js.map +1 -0
- package/dist/core/errors.d.ts +15 -0
- package/dist/core/errors.d.ts.map +1 -0
- package/dist/core/errors.js +29 -0
- package/dist/core/errors.js.map +1 -0
- package/dist/core/orchestrator.d.ts +31 -0
- package/dist/core/orchestrator.d.ts.map +1 -0
- package/dist/core/orchestrator.js +321 -0
- package/dist/core/orchestrator.js.map +1 -0
- package/dist/core/state-machine.d.ts +14 -0
- package/dist/core/state-machine.d.ts.map +1 -0
- package/dist/core/state-machine.js +77 -0
- package/dist/core/state-machine.js.map +1 -0
- package/dist/core/triage.d.ts +6 -0
- package/dist/core/triage.d.ts.map +1 -0
- package/dist/core/triage.js +58 -0
- package/dist/core/triage.js.map +1 -0
- package/dist/core/types.d.ts +61 -0
- package/dist/core/types.d.ts.map +1 -0
- package/dist/core/types.js +3 -0
- package/dist/core/types.js.map +1 -0
- package/dist/graph/client-factory.d.ts +22 -0
- package/dist/graph/client-factory.d.ts.map +1 -0
- package/dist/graph/client-factory.js +60 -0
- package/dist/graph/client-factory.js.map +1 -0
- package/dist/graph/context-slicer.d.ts +47 -0
- package/dist/graph/context-slicer.d.ts.map +1 -0
- package/dist/graph/context-slicer.js +246 -0
- package/dist/graph/context-slicer.js.map +1 -0
- package/dist/graph/file-indexer.d.ts +15 -0
- package/dist/graph/file-indexer.d.ts.map +1 -0
- package/dist/graph/file-indexer.js +343 -0
- package/dist/graph/file-indexer.js.map +1 -0
- package/dist/graph/graph-indexer.d.ts +7 -0
- package/dist/graph/graph-indexer.d.ts.map +1 -0
- package/dist/graph/graph-indexer.js +32 -0
- package/dist/graph/graph-indexer.js.map +1 -0
- package/dist/graph/graphify-client.d.ts +30 -0
- package/dist/graph/graphify-client.d.ts.map +1 -0
- package/dist/graph/graphify-client.js +186 -0
- package/dist/graph/graphify-client.js.map +1 -0
- package/dist/graph/graphify-file-client.d.ts +26 -0
- package/dist/graph/graphify-file-client.d.ts.map +1 -0
- package/dist/graph/graphify-file-client.js +140 -0
- package/dist/graph/graphify-file-client.js.map +1 -0
- package/dist/graph/graphify-mcp-client.d.ts +18 -0
- package/dist/graph/graphify-mcp-client.d.ts.map +1 -0
- package/dist/graph/graphify-mcp-client.js +73 -0
- package/dist/graph/graphify-mcp-client.js.map +1 -0
- package/dist/graph/language-indexers/c-cpp.d.ts +3 -0
- package/dist/graph/language-indexers/c-cpp.d.ts.map +1 -0
- package/dist/graph/language-indexers/c-cpp.js +121 -0
- package/dist/graph/language-indexers/c-cpp.js.map +1 -0
- package/dist/graph/language-indexers/go.d.ts +3 -0
- package/dist/graph/language-indexers/go.d.ts.map +1 -0
- package/dist/graph/language-indexers/go.js +97 -0
- package/dist/graph/language-indexers/go.js.map +1 -0
- package/dist/graph/language-indexers/index.d.ts +29 -0
- package/dist/graph/language-indexers/index.d.ts.map +1 -0
- package/dist/graph/language-indexers/index.js +27 -0
- package/dist/graph/language-indexers/index.js.map +1 -0
- package/dist/graph/language-indexers/python.d.ts +3 -0
- package/dist/graph/language-indexers/python.d.ts.map +1 -0
- package/dist/graph/language-indexers/python.js +91 -0
- package/dist/graph/language-indexers/python.js.map +1 -0
- package/dist/graph/language-indexers/rust.d.ts +3 -0
- package/dist/graph/language-indexers/rust.d.ts.map +1 -0
- package/dist/graph/language-indexers/rust.js +99 -0
- package/dist/graph/language-indexers/rust.js.map +1 -0
- package/dist/graph/language-indexers/tree-sitter-loader.d.ts +18 -0
- package/dist/graph/language-indexers/tree-sitter-loader.d.ts.map +1 -0
- package/dist/graph/language-indexers/tree-sitter-loader.js +34 -0
- package/dist/graph/language-indexers/tree-sitter-loader.js.map +1 -0
- package/dist/graph/language-indexers/typescript.d.ts +3 -0
- package/dist/graph/language-indexers/typescript.d.ts.map +1 -0
- package/dist/graph/language-indexers/typescript.js +194 -0
- package/dist/graph/language-indexers/typescript.js.map +1 -0
- package/dist/graph/semantic-enricher.d.ts +17 -0
- package/dist/graph/semantic-enricher.d.ts.map +1 -0
- package/dist/graph/semantic-enricher.js +97 -0
- package/dist/graph/semantic-enricher.js.map +1 -0
- package/dist/graph/sqlite-client.d.ts +24 -0
- package/dist/graph/sqlite-client.d.ts.map +1 -0
- package/dist/graph/sqlite-client.js +215 -0
- package/dist/graph/sqlite-client.js.map +1 -0
- package/dist/hooks/post-run-sync.d.ts +7 -0
- package/dist/hooks/post-run-sync.d.ts.map +1 -0
- package/dist/hooks/post-run-sync.js +26 -0
- package/dist/hooks/post-run-sync.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +42 -0
- package/dist/index.js.map +1 -0
- package/dist/integrations/agent-mcp-installer.d.ts +50 -0
- package/dist/integrations/agent-mcp-installer.d.ts.map +1 -0
- package/dist/integrations/agent-mcp-installer.js +414 -0
- package/dist/integrations/agent-mcp-installer.js.map +1 -0
- package/dist/learning/canary-gate.d.ts +6 -0
- package/dist/learning/canary-gate.d.ts.map +1 -0
- package/dist/learning/canary-gate.js +13 -0
- package/dist/learning/canary-gate.js.map +1 -0
- package/dist/learning/embeddings.d.ts +19 -0
- package/dist/learning/embeddings.d.ts.map +1 -0
- package/dist/learning/embeddings.js +160 -0
- package/dist/learning/embeddings.js.map +1 -0
- package/dist/learning/episodic-memory.d.ts +36 -0
- package/dist/learning/episodic-memory.d.ts.map +1 -0
- package/dist/learning/episodic-memory.js +228 -0
- package/dist/learning/episodic-memory.js.map +1 -0
- package/dist/learning/exporter.d.ts +10 -0
- package/dist/learning/exporter.d.ts.map +1 -0
- package/dist/learning/exporter.js +25 -0
- package/dist/learning/exporter.js.map +1 -0
- package/dist/learning/feedback-collector.d.ts +12 -0
- package/dist/learning/feedback-collector.d.ts.map +1 -0
- package/dist/learning/feedback-collector.js +14 -0
- package/dist/learning/feedback-collector.js.map +1 -0
- package/dist/learning/learning-events.d.ts +4 -0
- package/dist/learning/learning-events.d.ts.map +1 -0
- package/dist/learning/learning-events.js +35 -0
- package/dist/learning/learning-events.js.map +1 -0
- package/dist/learning/local-embedding.d.ts +3 -0
- package/dist/learning/local-embedding.d.ts.map +1 -0
- package/dist/learning/local-embedding.js +21 -0
- package/dist/learning/local-embedding.js.map +1 -0
- package/dist/learning/nightly-trainer.d.ts +14 -0
- package/dist/learning/nightly-trainer.d.ts.map +1 -0
- package/dist/learning/nightly-trainer.js +56 -0
- package/dist/learning/nightly-trainer.js.map +1 -0
- package/dist/learning/reflector.d.ts +19 -0
- package/dist/learning/reflector.d.ts.map +1 -0
- package/dist/learning/reflector.js +130 -0
- package/dist/learning/reflector.js.map +1 -0
- package/dist/learning/sample-builder.d.ts +7 -0
- package/dist/learning/sample-builder.d.ts.map +1 -0
- package/dist/learning/sample-builder.js +10 -0
- package/dist/learning/sample-builder.js.map +1 -0
- package/dist/learning/skill-flywheel.d.ts +52 -0
- package/dist/learning/skill-flywheel.d.ts.map +1 -0
- package/dist/learning/skill-flywheel.js +533 -0
- package/dist/learning/skill-flywheel.js.map +1 -0
- package/dist/learning/vector-store.d.ts +18 -0
- package/dist/learning/vector-store.d.ts.map +1 -0
- package/dist/learning/vector-store.js +73 -0
- package/dist/learning/vector-store.js.map +1 -0
- package/dist/routing/model-router.d.ts +13 -0
- package/dist/routing/model-router.d.ts.map +1 -0
- package/dist/routing/model-router.js +113 -0
- package/dist/routing/model-router.js.map +1 -0
- package/dist/routing/provider-adapters/anthropic.d.ts +3 -0
- package/dist/routing/provider-adapters/anthropic.d.ts.map +1 -0
- package/dist/routing/provider-adapters/anthropic.js +57 -0
- package/dist/routing/provider-adapters/anthropic.js.map +1 -0
- package/dist/routing/provider-adapters/bailian.d.ts +3 -0
- package/dist/routing/provider-adapters/bailian.d.ts.map +1 -0
- package/dist/routing/provider-adapters/bailian.js +56 -0
- package/dist/routing/provider-adapters/bailian.js.map +1 -0
- package/dist/routing/provider-adapters/doubao.d.ts +3 -0
- package/dist/routing/provider-adapters/doubao.d.ts.map +1 -0
- package/dist/routing/provider-adapters/doubao.js +56 -0
- package/dist/routing/provider-adapters/doubao.js.map +1 -0
- package/dist/routing/provider-adapters/openai.d.ts +6 -0
- package/dist/routing/provider-adapters/openai.d.ts.map +1 -0
- package/dist/routing/provider-adapters/openai.js +80 -0
- package/dist/routing/provider-adapters/openai.js.map +1 -0
- package/dist/routing/provider-adapters/openbmb.d.ts +16 -0
- package/dist/routing/provider-adapters/openbmb.d.ts.map +1 -0
- package/dist/routing/provider-adapters/openbmb.js +368 -0
- package/dist/routing/provider-adapters/openbmb.js.map +1 -0
- package/dist/routing/provider-executor.d.ts +21 -0
- package/dist/routing/provider-executor.d.ts.map +1 -0
- package/dist/routing/provider-executor.js +260 -0
- package/dist/routing/provider-executor.js.map +1 -0
- package/dist/routing/provider-health.d.ts +9 -0
- package/dist/routing/provider-health.d.ts.map +1 -0
- package/dist/routing/provider-health.js +66 -0
- package/dist/routing/provider-health.js.map +1 -0
- package/dist/surfaces/cli/index.d.ts +3 -0
- package/dist/surfaces/cli/index.d.ts.map +1 -0
- package/dist/surfaces/cli/index.js +200 -0
- package/dist/surfaces/cli/index.js.map +1 -0
- package/dist/surfaces/cli/init.d.ts +2 -0
- package/dist/surfaces/cli/init.d.ts.map +1 -0
- package/dist/surfaces/cli/init.js +121 -0
- package/dist/surfaces/cli/init.js.map +1 -0
- package/dist/surfaces/cli/output.d.ts +16 -0
- package/dist/surfaces/cli/output.d.ts.map +1 -0
- package/dist/surfaces/cli/output.js +74 -0
- package/dist/surfaces/cli/output.js.map +1 -0
- package/dist/surfaces/cli/runtime.d.ts +218 -0
- package/dist/surfaces/cli/runtime.d.ts.map +1 -0
- package/dist/surfaces/cli/runtime.js +1130 -0
- package/dist/surfaces/cli/runtime.js.map +1 -0
- package/dist/surfaces/mcp/server.d.ts +60 -0
- package/dist/surfaces/mcp/server.d.ts.map +1 -0
- package/dist/surfaces/mcp/server.js +425 -0
- package/dist/surfaces/mcp/server.js.map +1 -0
- package/dist/surfaces/vscode/extension.d.ts +17 -0
- package/dist/surfaces/vscode/extension.d.ts.map +1 -0
- package/dist/surfaces/vscode/extension.js +35 -0
- package/dist/surfaces/vscode/extension.js.map +1 -0
- package/dist/utils/file-lock.d.ts +9 -0
- package/dist/utils/file-lock.d.ts.map +1 -0
- package/dist/utils/file-lock.js +90 -0
- package/dist/utils/file-lock.js.map +1 -0
- package/dist/utils/logger.d.ts +3 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +36 -0
- package/dist/utils/logger.js.map +1 -0
- package/package.json +89 -0
- package/scripts/safe-postinstall.cjs +14 -0
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { AgentRole } from "../core/types";
|
|
2
|
+
export type ModelTier = "smart" | "economy";
|
|
3
|
+
export type ProviderName = "openai" | "anthropic" | "bailian" | "doubao" | "openbmb";
|
|
4
|
+
export type ProviderHealthMap = Record<ProviderName, boolean>;
|
|
5
|
+
export interface ModelSelection {
|
|
6
|
+
provider: ProviderName;
|
|
7
|
+
model: string;
|
|
8
|
+
tier: ModelTier;
|
|
9
|
+
fallbackApplied: boolean;
|
|
10
|
+
}
|
|
11
|
+
export declare function resolveModelForRole(role: AgentRole, configPath?: string): ModelSelection;
|
|
12
|
+
export declare function resolveModelWithFallback(role: AgentRole, providerHealth: ProviderHealthMap, fallbackChain?: ProviderName[], configPath?: string): ModelSelection;
|
|
13
|
+
//# sourceMappingURL=model-router.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-router.d.ts","sourceRoot":"","sources":["../../src/routing/model-router.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG/C,MAAM,MAAM,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;AAC5C,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,GAAG,SAAS,CAAC;AAErF,MAAM,MAAM,iBAAiB,GAAG,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;AAE9D,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,YAAY,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,SAAS,CAAC;IAChB,eAAe,EAAE,OAAO,CAAC;CAC1B;AAiCD,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,cAAc,CAmExF;AAED,wBAAgB,wBAAwB,CACtC,IAAI,EAAE,SAAS,EACf,cAAc,EAAE,iBAAiB,EACjC,aAAa,CAAC,EAAE,YAAY,EAAE,EAC9B,UAAU,CAAC,EAAE,MAAM,GAClB,cAAc,CAuBhB"}
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.resolveModelForRole = resolveModelForRole;
|
|
4
|
+
exports.resolveModelWithFallback = resolveModelWithFallback;
|
|
5
|
+
const resolve_1 = require("../config/resolve");
|
|
6
|
+
const roleTierMap = {
|
|
7
|
+
planner: "smart",
|
|
8
|
+
validator: "smart",
|
|
9
|
+
worker: "economy",
|
|
10
|
+
enricher: "economy",
|
|
11
|
+
evolver: "economy",
|
|
12
|
+
};
|
|
13
|
+
const DEFAULT_MODELS = {
|
|
14
|
+
openai: {
|
|
15
|
+
smart: "gpt-4.1",
|
|
16
|
+
economy: "gpt-4.1-mini",
|
|
17
|
+
},
|
|
18
|
+
anthropic: {
|
|
19
|
+
smart: "claude-3-5-sonnet-latest",
|
|
20
|
+
economy: "claude-3-5-haiku-latest",
|
|
21
|
+
},
|
|
22
|
+
bailian: {
|
|
23
|
+
smart: "qwen-max",
|
|
24
|
+
economy: "qwen-plus",
|
|
25
|
+
},
|
|
26
|
+
doubao: {
|
|
27
|
+
smart: "doubao-pro-32k",
|
|
28
|
+
economy: "doubao-lite-32k",
|
|
29
|
+
},
|
|
30
|
+
openbmb: {
|
|
31
|
+
smart: "minicpm5-1b",
|
|
32
|
+
economy: "minicpm5-1b",
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
function resolveModelForRole(role, configPath) {
|
|
36
|
+
const tier = roleTierMap[role];
|
|
37
|
+
try {
|
|
38
|
+
const config = (0, resolve_1.resolveConfig)(configPath ?? "graphflow.config.json");
|
|
39
|
+
if (role === "enricher") {
|
|
40
|
+
const enrichPolicy = config.graphPolicy.semanticEnrichment;
|
|
41
|
+
const backend = enrichPolicy?.backend ?? "inherit";
|
|
42
|
+
if (backend === "local" || enrichPolicy?.provider === "openbmb") {
|
|
43
|
+
return {
|
|
44
|
+
provider: "openbmb",
|
|
45
|
+
model: enrichPolicy?.model ??
|
|
46
|
+
config.learningPolicy.skillEvolution?.model ??
|
|
47
|
+
DEFAULT_MODELS.openbmb.economy,
|
|
48
|
+
tier,
|
|
49
|
+
fallbackApplied: false,
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
const provider = (backend === "network" && enrichPolicy?.provider
|
|
53
|
+
? enrichPolicy.provider
|
|
54
|
+
: enrichPolicy?.provider ?? config.tiers.economy.provider);
|
|
55
|
+
return {
|
|
56
|
+
provider,
|
|
57
|
+
model: enrichPolicy?.model ??
|
|
58
|
+
config.tiers.economy.model ??
|
|
59
|
+
DEFAULT_MODELS[provider].economy,
|
|
60
|
+
tier,
|
|
61
|
+
fallbackApplied: false,
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
if (role === "evolver") {
|
|
65
|
+
const provider = config.tiers.economy.provider;
|
|
66
|
+
return {
|
|
67
|
+
provider,
|
|
68
|
+
model: config.learningPolicy.skillEvolution?.model ??
|
|
69
|
+
config.tiers.economy.model ??
|
|
70
|
+
DEFAULT_MODELS[provider].economy,
|
|
71
|
+
tier: "economy",
|
|
72
|
+
fallbackApplied: false,
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
const tierConfig = config.tiers[tier];
|
|
76
|
+
const provider = tierConfig.provider;
|
|
77
|
+
return {
|
|
78
|
+
provider,
|
|
79
|
+
model: tierConfig.model || DEFAULT_MODELS[provider][tier],
|
|
80
|
+
tier,
|
|
81
|
+
fallbackApplied: false,
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
catch {
|
|
85
|
+
return {
|
|
86
|
+
provider: "openai",
|
|
87
|
+
model: DEFAULT_MODELS.openai[tier],
|
|
88
|
+
tier,
|
|
89
|
+
fallbackApplied: false,
|
|
90
|
+
};
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
function resolveModelWithFallback(role, providerHealth, fallbackChain, configPath) {
|
|
94
|
+
const base = resolveModelForRole(role, configPath);
|
|
95
|
+
if (providerHealth[base.provider]) {
|
|
96
|
+
return base;
|
|
97
|
+
}
|
|
98
|
+
const chain = fallbackChain ?? ["anthropic", "bailian", "doubao", "openbmb"];
|
|
99
|
+
const available = chain.find((provider) => provider !== base.provider && providerHealth[provider]);
|
|
100
|
+
if (!available) {
|
|
101
|
+
return {
|
|
102
|
+
...base,
|
|
103
|
+
fallbackApplied: true,
|
|
104
|
+
};
|
|
105
|
+
}
|
|
106
|
+
return {
|
|
107
|
+
provider: available,
|
|
108
|
+
model: DEFAULT_MODELS[available][base.tier],
|
|
109
|
+
tier: base.tier,
|
|
110
|
+
fallbackApplied: true,
|
|
111
|
+
};
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=model-router.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-router.js","sourceRoot":"","sources":["../../src/routing/model-router.ts"],"names":[],"mappings":";;AA8CA,kDAmEC;AAED,4DA4BC;AA9ID,+CAAkD;AAclD,MAAM,WAAW,GAAiC;IAChD,OAAO,EAAE,OAAO;IAChB,SAAS,EAAE,OAAO;IAClB,MAAM,EAAE,SAAS;IACjB,QAAQ,EAAE,SAAS;IACnB,OAAO,EAAE,SAAS;CACnB,CAAC;AAEF,MAAM,cAAc,GAAoD;IACtE,MAAM,EAAE;QACN,KAAK,EAAE,SAAS;QAChB,OAAO,EAAE,cAAc;KACxB;IACD,SAAS,EAAE;QACT,KAAK,EAAE,0BAA0B;QACjC,OAAO,EAAE,yBAAyB;KACnC;IACD,OAAO,EAAE;QACP,KAAK,EAAE,UAAU;QACjB,OAAO,EAAE,WAAW;KACrB;IACD,MAAM,EAAE;QACN,KAAK,EAAE,gBAAgB;QACvB,OAAO,EAAE,iBAAiB;KAC3B;IACD,OAAO,EAAE;QACP,KAAK,EAAE,aAAa;QACpB,OAAO,EAAE,aAAa;KACvB;CACF,CAAC;AAEF,SAAgB,mBAAmB,CAAC,IAAe,EAAE,UAAmB;IACtE,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAE/B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAA,uBAAa,EAAC,UAAU,IAAI,uBAAuB,CAAC,CAAC;QAEpE,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACxB,MAAM,YAAY,GAAG,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAAC;YAC3D,MAAM,OAAO,GAAG,YAAY,EAAE,OAAO,IAAI,SAAS,CAAC;YAEnD,IAAI,OAAO,KAAK,OAAO,IAAI,YAAY,EAAE,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAChE,OAAO;oBACL,QAAQ,EAAE,SAAS;oBACnB,KAAK,EACH,YAAY,EAAE,KAAK;wBACnB,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,KAAK;wBAC3C,cAAc,CAAC,OAAO,CAAC,OAAO;oBAChC,IAAI;oBACJ,eAAe,EAAE,KAAK;iBACvB,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,CACf,OAAO,KAAK,SAAS,IAAI,YAAY,EAAE,QAAQ;gBAC7C,CAAC,CAAC,YAAY,CAAC,QAAQ;gBACvB,CAAC,CAAC,YAAY,EAAE,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAC5C,CAAC;YAClB,OAAO;gBACL,QAAQ;gBACR,KAAK,EACH,YAAY,EAAE,KAAK;oBACnB,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK;oBAC1B,cAAc,CAAC,QAAQ,CAAC,CAAC,OAAO;gBAClC,IAAI;gBACJ,eAAe,EAAE,KAAK;aACvB,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAwB,CAAC;YAC/D,OAAO;gBACL,QAAQ;gBACR,KAAK,EACH,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,KAAK;oBAC3C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK;oBAC1B,cAAc,CAAC,QAAQ,CAAC,CAAC,OAAO;gBAClC,IAAI,EAAE,SAAS;gBACf,eAAe,EAAE,KAAK;aACvB,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAwB,CAAC;QACrD,OAAO;YACL,QAAQ;YACR,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI,cAAc,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC;YACzD,IAAI;YACJ,eAAe,EAAE,KAAK;SACvB,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC;YAClC,IAAI;YACJ,eAAe,EAAE,KAAK;SACvB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAgB,wBAAwB,CACtC,IAAe,EACf,cAAiC,EACjC,aAA8B,EAC9B,UAAmB;IAEnB,MAAM,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAEnD,IAAI,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,KAAK,GAAG,aAAa,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,IAAI,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEnG,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO;YACL,GAAG,IAAI;YACP,eAAe,EAAE,IAAI;SACtB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,SAAS;QACnB,KAAK,EAAE,cAAc,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QAC3C,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,eAAe,EAAE,IAAI;KACtB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/routing/provider-adapters/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAEpD,wBAAsB,qBAAqB,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC,CA0DzF"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.anthropicGenerateText = anthropicGenerateText;
|
|
4
|
+
const logger_1 = require("../../utils/logger");
|
|
5
|
+
async function anthropicGenerateText(request) {
|
|
6
|
+
const strict = process.env.GRAPHFLOW_ANTHROPIC_STRICT === "1";
|
|
7
|
+
const apiKey = process.env.ANTHROPIC_API_KEY;
|
|
8
|
+
const baseUrl = (process.env.ANTHROPIC_BASE_URL ?? "https://api.anthropic.com").replace(/\/+$/, "");
|
|
9
|
+
const timeoutMsRaw = Number(process.env.GRAPHFLOW_ANTHROPIC_TIMEOUT_MS ?? 15000);
|
|
10
|
+
const timeoutMs = Number.isFinite(timeoutMsRaw) ? Math.max(1000, Math.floor(timeoutMsRaw)) : 15000;
|
|
11
|
+
if (!apiKey) {
|
|
12
|
+
if (strict) {
|
|
13
|
+
throw new Error("ANTHROPIC_API_KEY is required");
|
|
14
|
+
}
|
|
15
|
+
return `[anthropic:${request.model}] ${request.prompt}`;
|
|
16
|
+
}
|
|
17
|
+
const controller = new AbortController();
|
|
18
|
+
const timer = setTimeout(() => controller.abort(), timeoutMs);
|
|
19
|
+
try {
|
|
20
|
+
const response = await fetch(`${baseUrl}/v1/messages`, {
|
|
21
|
+
method: "POST",
|
|
22
|
+
headers: {
|
|
23
|
+
"content-type": "application/json",
|
|
24
|
+
"x-api-key": apiKey,
|
|
25
|
+
"anthropic-version": "2023-06-01",
|
|
26
|
+
},
|
|
27
|
+
body: JSON.stringify({
|
|
28
|
+
model: request.model,
|
|
29
|
+
max_tokens: 512,
|
|
30
|
+
temperature: 0.1,
|
|
31
|
+
messages: [{ role: "user", content: request.prompt }],
|
|
32
|
+
}),
|
|
33
|
+
signal: controller.signal,
|
|
34
|
+
});
|
|
35
|
+
if (!response.ok) {
|
|
36
|
+
const text = await response.text();
|
|
37
|
+
throw new Error(`anthropic http ${response.status}: ${text.slice(0, 300)}`);
|
|
38
|
+
}
|
|
39
|
+
const payload = (await response.json());
|
|
40
|
+
const text = payload.content?.find((item) => item.type === "text")?.text?.trim();
|
|
41
|
+
if (!text) {
|
|
42
|
+
throw new Error("anthropic response missing text content");
|
|
43
|
+
}
|
|
44
|
+
return text;
|
|
45
|
+
}
|
|
46
|
+
catch (error) {
|
|
47
|
+
logger_1.logger.error({ error: error instanceof Error ? error.message : String(error) }, "Provider adapter caught error");
|
|
48
|
+
if (strict) {
|
|
49
|
+
throw error;
|
|
50
|
+
}
|
|
51
|
+
return `[anthropic:${request.model}] ${request.prompt}`;
|
|
52
|
+
}
|
|
53
|
+
finally {
|
|
54
|
+
clearTimeout(timer);
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/routing/provider-adapters/anthropic.ts"],"names":[],"mappings":";;AAGA,sDA0DC;AA7DD,+CAA4C;AAGrC,KAAK,UAAU,qBAAqB,CAAC,OAA4B;IACtE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,0BAA0B,KAAK,GAAG,CAAC;IAC9D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAC7C,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,2BAA2B,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACpG,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,KAAK,CAAC,CAAC;IACjF,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEnG,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,cAAc,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IAC1D,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAC9D,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,cAAc,EAAE;YACrD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,MAAM;gBACnB,mBAAmB,EAAE,YAAY;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,UAAU,EAAE,GAAG;gBACf,WAAW,EAAE,GAAG;gBAChB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;aACtD,CAAC;YACF,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAC9E,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAErC,CAAC;QACF,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACjF,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,eAAM,CAAC,KAAK,CACV,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACjE,+BAA+B,CAChC,CAAC;QACF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,cAAc,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IAC1D,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bailian.d.ts","sourceRoot":"","sources":["../../../src/routing/provider-adapters/bailian.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAEpD,wBAAsB,mBAAmB,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC,CA2DvF"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.bailianGenerateText = bailianGenerateText;
|
|
4
|
+
const logger_1 = require("../../utils/logger");
|
|
5
|
+
async function bailianGenerateText(request) {
|
|
6
|
+
const strict = process.env.GRAPHFLOW_BAILIAN_STRICT === "1";
|
|
7
|
+
const apiKey = process.env.BAILIAN_API_KEY;
|
|
8
|
+
const baseUrl = (process.env.BAILIAN_BASE_URL ?? "https://dashscope.aliyuncs.com/compatible-mode/v1").replace(/\/+$/, "");
|
|
9
|
+
const timeoutMsRaw = Number(process.env.GRAPHFLOW_BAILIAN_TIMEOUT_MS ?? 15000);
|
|
10
|
+
const timeoutMs = Number.isFinite(timeoutMsRaw) ? Math.max(1000, Math.floor(timeoutMsRaw)) : 15000;
|
|
11
|
+
if (!apiKey) {
|
|
12
|
+
if (strict) {
|
|
13
|
+
throw new Error("BAILIAN_API_KEY is required");
|
|
14
|
+
}
|
|
15
|
+
return `[bailian:${request.model}] ${request.prompt}`;
|
|
16
|
+
}
|
|
17
|
+
const controller = new AbortController();
|
|
18
|
+
const timer = setTimeout(() => controller.abort(), timeoutMs);
|
|
19
|
+
try {
|
|
20
|
+
const response = await fetch(`${baseUrl}/chat/completions`, {
|
|
21
|
+
method: "POST",
|
|
22
|
+
headers: {
|
|
23
|
+
"content-type": "application/json",
|
|
24
|
+
authorization: `Bearer ${apiKey}`,
|
|
25
|
+
},
|
|
26
|
+
body: JSON.stringify({
|
|
27
|
+
model: request.model,
|
|
28
|
+
messages: [{ role: "user", content: request.prompt }],
|
|
29
|
+
temperature: 0.1,
|
|
30
|
+
max_tokens: 512,
|
|
31
|
+
}),
|
|
32
|
+
signal: controller.signal,
|
|
33
|
+
});
|
|
34
|
+
if (!response.ok) {
|
|
35
|
+
const text = await response.text();
|
|
36
|
+
throw new Error(`bailian http ${response.status}: ${text.slice(0, 300)}`);
|
|
37
|
+
}
|
|
38
|
+
const payload = (await response.json());
|
|
39
|
+
const text = payload.choices?.[0]?.message?.content?.trim() ?? payload.choices?.[0]?.text?.trim();
|
|
40
|
+
if (!text) {
|
|
41
|
+
throw new Error("bailian response missing content");
|
|
42
|
+
}
|
|
43
|
+
return text;
|
|
44
|
+
}
|
|
45
|
+
catch (error) {
|
|
46
|
+
logger_1.logger.error({ error: error instanceof Error ? error.message : String(error) }, "Provider adapter caught error");
|
|
47
|
+
if (strict) {
|
|
48
|
+
throw error;
|
|
49
|
+
}
|
|
50
|
+
return `[bailian:${request.model}] ${request.prompt}`;
|
|
51
|
+
}
|
|
52
|
+
finally {
|
|
53
|
+
clearTimeout(timer);
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=bailian.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bailian.js","sourceRoot":"","sources":["../../../src/routing/provider-adapters/bailian.ts"],"names":[],"mappings":";;AAGA,kDA2DC;AA9DD,+CAA4C;AAGrC,KAAK,UAAU,mBAAmB,CAAC,OAA4B;IACpE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,GAAG,CAAC;IAC5D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC3C,MAAM,OAAO,GAAG,CACd,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,mDAAmD,CACpF,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACtB,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,KAAK,CAAC,CAAC;IAC/E,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEnG,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,YAAY,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IACxD,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAC9D,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;YAC1D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrD,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,GAAG;aAChB,CAAC;YACF,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,gBAAgB,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAErC,CAAC;QACF,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAClG,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,eAAM,CAAC,KAAK,CACV,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACjE,+BAA+B,CAChC,CAAC;QACF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,YAAY,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IACxD,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"doubao.d.ts","sourceRoot":"","sources":["../../../src/routing/provider-adapters/doubao.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAEpD,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC,CA4DtF"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.doubaoGenerateText = doubaoGenerateText;
|
|
4
|
+
const logger_1 = require("../../utils/logger");
|
|
5
|
+
async function doubaoGenerateText(request) {
|
|
6
|
+
const strict = process.env.GRAPHFLOW_DOUBAO_STRICT === "1";
|
|
7
|
+
const apiKey = process.env.DOUBAO_API_KEY;
|
|
8
|
+
const baseUrl = (process.env.DOUBAO_BASE_URL ?? "https://ark.cn-beijing.volces.com/api/v3").replace(/\/+$/, "");
|
|
9
|
+
const timeoutMsRaw = Number(process.env.GRAPHFLOW_DOUBAO_TIMEOUT_MS ?? 15000);
|
|
10
|
+
const timeoutMs = Number.isFinite(timeoutMsRaw) ? Math.max(1000, Math.floor(timeoutMsRaw)) : 15000;
|
|
11
|
+
if (!apiKey) {
|
|
12
|
+
if (strict) {
|
|
13
|
+
throw new Error("DOUBAO_API_KEY is required");
|
|
14
|
+
}
|
|
15
|
+
return `[doubao:${request.model}] ${request.prompt}`;
|
|
16
|
+
}
|
|
17
|
+
const controller = new AbortController();
|
|
18
|
+
const timer = setTimeout(() => controller.abort(), timeoutMs);
|
|
19
|
+
try {
|
|
20
|
+
const response = await fetch(`${baseUrl}/chat/completions`, {
|
|
21
|
+
method: "POST",
|
|
22
|
+
headers: {
|
|
23
|
+
"content-type": "application/json",
|
|
24
|
+
authorization: `Bearer ${apiKey}`,
|
|
25
|
+
},
|
|
26
|
+
body: JSON.stringify({
|
|
27
|
+
model: request.model,
|
|
28
|
+
messages: [{ role: "user", content: request.prompt }],
|
|
29
|
+
temperature: 0.1,
|
|
30
|
+
max_tokens: 512,
|
|
31
|
+
}),
|
|
32
|
+
signal: controller.signal,
|
|
33
|
+
});
|
|
34
|
+
if (!response.ok) {
|
|
35
|
+
const text = await response.text();
|
|
36
|
+
throw new Error(`doubao http ${response.status}: ${text.slice(0, 300)}`);
|
|
37
|
+
}
|
|
38
|
+
const payload = (await response.json());
|
|
39
|
+
const text = payload.choices?.[0]?.message?.content?.trim() ?? payload.choices?.[0]?.text?.trim();
|
|
40
|
+
if (!text) {
|
|
41
|
+
throw new Error("doubao response missing content");
|
|
42
|
+
}
|
|
43
|
+
return text;
|
|
44
|
+
}
|
|
45
|
+
catch (error) {
|
|
46
|
+
logger_1.logger.error({ error: error instanceof Error ? error.message : String(error) }, "Provider adapter caught error");
|
|
47
|
+
if (strict) {
|
|
48
|
+
throw error;
|
|
49
|
+
}
|
|
50
|
+
return `[doubao:${request.model}] ${request.prompt}`;
|
|
51
|
+
}
|
|
52
|
+
finally {
|
|
53
|
+
clearTimeout(timer);
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=doubao.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"doubao.js","sourceRoot":"","sources":["../../../src/routing/provider-adapters/doubao.ts"],"names":[],"mappings":";;AAGA,gDA4DC;AA/DD,+CAA4C;AAGrC,KAAK,UAAU,kBAAkB,CAAC,OAA4B;IACnE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,GAAG,CAAC;IAC3D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC1C,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,0CAA0C,CAAC,CAAC,OAAO,CACjG,MAAM,EACN,EAAE,CACH,CAAC;IACF,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,KAAK,CAAC,CAAC;IAC9E,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEnG,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,WAAW,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IACvD,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAC9D,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;YAC1D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrD,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,GAAG;aAChB,CAAC;YACF,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,eAAe,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAErC,CAAC;QACF,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAClG,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,eAAM,CAAC,KAAK,CACV,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACjE,+BAA+B,CAChC,CAAC;QACF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,WAAW,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IACvD,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/routing/provider-adapters/openai.ts"],"names":[],"mappings":"AACA,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;CACf;AA2BD,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC,CAsDtF"}
|
|
@@ -0,0 +1,80 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.openaiGenerateText = openaiGenerateText;
|
|
4
|
+
const logger_1 = require("../../utils/logger");
|
|
5
|
+
function asRecord(value) {
|
|
6
|
+
if (value && typeof value === "object") {
|
|
7
|
+
return value;
|
|
8
|
+
}
|
|
9
|
+
return {};
|
|
10
|
+
}
|
|
11
|
+
function pickContent(payload) {
|
|
12
|
+
const choices = payload.choices;
|
|
13
|
+
if (!Array.isArray(choices) || choices.length === 0) {
|
|
14
|
+
return undefined;
|
|
15
|
+
}
|
|
16
|
+
const first = asRecord(choices[0]);
|
|
17
|
+
const message = asRecord(first.message);
|
|
18
|
+
const content = message.content;
|
|
19
|
+
if (typeof content === "string" && content.trim().length > 0) {
|
|
20
|
+
return content.trim();
|
|
21
|
+
}
|
|
22
|
+
const text = first.text;
|
|
23
|
+
if (typeof text === "string" && text.trim().length > 0) {
|
|
24
|
+
return text.trim();
|
|
25
|
+
}
|
|
26
|
+
return undefined;
|
|
27
|
+
}
|
|
28
|
+
async function openaiGenerateText(request) {
|
|
29
|
+
const strict = process.env.GRAPHFLOW_OPENAI_STRICT === "1";
|
|
30
|
+
const apiKey = process.env.OPENAI_API_KEY;
|
|
31
|
+
const baseUrl = (process.env.OPENAI_BASE_URL ?? "https://api.openai.com/v1").replace(/\/+$/, "");
|
|
32
|
+
const timeoutMsRaw = Number(process.env.GRAPHFLOW_OPENAI_TIMEOUT_MS ?? 15000);
|
|
33
|
+
const timeoutMs = Number.isFinite(timeoutMsRaw) ? Math.max(1000, Math.floor(timeoutMsRaw)) : 15000;
|
|
34
|
+
if (!apiKey) {
|
|
35
|
+
if (strict) {
|
|
36
|
+
throw new Error("OPENAI_API_KEY is required");
|
|
37
|
+
}
|
|
38
|
+
return `[openai:${request.model}] ${request.prompt}`;
|
|
39
|
+
}
|
|
40
|
+
const controller = new AbortController();
|
|
41
|
+
const timer = setTimeout(() => controller.abort(), timeoutMs);
|
|
42
|
+
try {
|
|
43
|
+
const response = await fetch(`${baseUrl}/chat/completions`, {
|
|
44
|
+
method: "POST",
|
|
45
|
+
headers: {
|
|
46
|
+
"content-type": "application/json",
|
|
47
|
+
authorization: `Bearer ${apiKey}`,
|
|
48
|
+
},
|
|
49
|
+
body: JSON.stringify({
|
|
50
|
+
model: request.model,
|
|
51
|
+
messages: [{ role: "user", content: request.prompt }],
|
|
52
|
+
temperature: 0.1,
|
|
53
|
+
max_tokens: 512,
|
|
54
|
+
}),
|
|
55
|
+
signal: controller.signal,
|
|
56
|
+
});
|
|
57
|
+
if (!response.ok) {
|
|
58
|
+
const text = await response.text();
|
|
59
|
+
throw new Error(`openai http ${response.status}: ${text.slice(0, 300)}`);
|
|
60
|
+
}
|
|
61
|
+
const payload = asRecord(await response.json());
|
|
62
|
+
const content = pickContent(payload);
|
|
63
|
+
if (!content) {
|
|
64
|
+
throw new Error("openai response missing content");
|
|
65
|
+
}
|
|
66
|
+
return content;
|
|
67
|
+
}
|
|
68
|
+
catch (error) {
|
|
69
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
70
|
+
logger_1.logger.error({ error: message }, "Provider adapter caught error");
|
|
71
|
+
if (strict) {
|
|
72
|
+
throw error;
|
|
73
|
+
}
|
|
74
|
+
return `[openai:${request.model}] ${request.prompt}`;
|
|
75
|
+
}
|
|
76
|
+
finally {
|
|
77
|
+
clearTimeout(timer);
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/routing/provider-adapters/openai.ts"],"names":[],"mappings":";;AA+BA,gDAsDC;AArFD,+CAA4C;AAM5C,SAAS,QAAQ,CAAC,KAAc;IAC9B,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvC,OAAO,KAAgC,CAAC;IAC1C,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,WAAW,CAAC,OAAgC;IACnD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAChC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAChC,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7D,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IACD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;IACxB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvD,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAEM,KAAK,UAAU,kBAAkB,CAAC,OAA4B;IACnE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,GAAG,CAAC;IAC3D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC1C,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,2BAA2B,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACjG,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,KAAK,CAAC,CAAC;IAC9E,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEnG,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,WAAW,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IACvD,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAE9D,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;YAC1D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrD,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,GAAG;aAChB,CAAC;YACF,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,eAAe,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QAChD,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;QACrC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,eAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,+BAA+B,CAAC,CAAC;QAClE,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,WAAW,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;IACvD,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
export interface ProviderTextRequest {
|
|
2
|
+
prompt: string;
|
|
3
|
+
model: string;
|
|
4
|
+
}
|
|
5
|
+
export type OpenBmbMode = "embedded" | "ollama" | "openai-compat";
|
|
6
|
+
export interface OpenBmbRuntimeOptions {
|
|
7
|
+
mode?: OpenBmbMode;
|
|
8
|
+
baseUrl?: string;
|
|
9
|
+
apiKey?: string;
|
|
10
|
+
timeoutMs?: number;
|
|
11
|
+
maxTokens?: number;
|
|
12
|
+
temperature?: number;
|
|
13
|
+
modelPath?: string;
|
|
14
|
+
}
|
|
15
|
+
export declare function openbmbGenerateText(request: ProviderTextRequest, runtimeOptions?: OpenBmbRuntimeOptions): Promise<string>;
|
|
16
|
+
//# sourceMappingURL=openbmb.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openbmb.d.ts","sourceRoot":"","sources":["../../../src/routing/provider-adapters/openbmb.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,WAAW,GAAG,UAAU,GAAG,QAAQ,GAAG,eAAe,CAAC;AAElE,MAAM,WAAW,qBAAqB;IACpC,IAAI,CAAC,EAAE,WAAW,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AA8aD,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,mBAAmB,EAC5B,cAAc,CAAC,EAAE,qBAAqB,GACrC,OAAO,CAAC,MAAM,CAAC,CAsBjB"}
|