@drax/ai-back 3.33.0 → 3.36.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/dist/config/GoogleAiConfig.js +0 -1
- package/dist/config/OllamaAiConfig.js +9 -0
- package/dist/factory/AiProviderFactory.js +4 -0
- package/dist/factory/GoogleAiProviderFactory.js +1 -1
- package/dist/factory/OllamaAiProviderFactory.js +14 -0
- package/dist/index.js +4 -1
- package/dist/providers/OllamaAiProvider.js +342 -0
- package/package.json +3 -3
- package/src/config/GoogleAiConfig.ts +0 -1
- package/src/config/OllamaAiConfig.ts +14 -0
- package/src/factory/AiProviderFactory.ts +4 -0
- package/src/factory/GoogleAiProviderFactory.ts +1 -1
- package/src/factory/OllamaAiProviderFactory.ts +27 -0
- package/src/index.ts +6 -0
- package/src/providers/OllamaAiProvider.ts +469 -0
- package/tsconfig.tsbuildinfo +1 -1
- package/types/config/GoogleAiConfig.d.ts +0 -1
- package/types/config/GoogleAiConfig.d.ts.map +1 -1
- package/types/config/OllamaAiConfig.d.ts +9 -0
- package/types/config/OllamaAiConfig.d.ts.map +1 -0
- package/types/factory/AiProviderFactory.d.ts.map +1 -1
- package/types/factory/OllamaAiProviderFactory.d.ts +8 -0
- package/types/factory/OllamaAiProviderFactory.d.ts.map +1 -0
- package/types/index.d.ts +7 -1
- package/types/index.d.ts.map +1 -1
- package/types/providers/OllamaAiProvider.d.ts +78 -0
- package/types/providers/OllamaAiProvider.d.ts.map +1 -0
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"GoogleAiConfig.d.ts","sourceRoot":"","sources":["../../src/config/GoogleAiConfig.ts"],"names":[],"mappings":"AACA,aAAK,cAAc;IAEf,cAAc,sBAAsB;IACpC,
|
|
1
|
+
{"version":3,"file":"GoogleAiConfig.d.ts","sourceRoot":"","sources":["../../src/config/GoogleAiConfig.ts"],"names":[],"mappings":"AACA,aAAK,cAAc;IAEf,cAAc,sBAAsB;IACpC,aAAa,oBAAoB;IACjC,mBAAmB,2BAA2B;CAEjD;AAGD,eAAe,cAAc,CAAC;AAE9B,OAAO,EAAC,cAAc,EAAC,CAAA"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
declare enum OllamaAiConfig {
|
|
2
|
+
OllamaAiBaseUrl = "OLLAMA_AI_BASE_URL",
|
|
3
|
+
OllamaAiModel = "OLLAMA_AI_MODEL",
|
|
4
|
+
OllamaAiVisionModel = "OLLAMA_AI_VISION_MODEL",
|
|
5
|
+
OllamaAiEmbeddingModel = "OLLAMA_AI_EMBEDDING_MODEL"
|
|
6
|
+
}
|
|
7
|
+
export default OllamaAiConfig;
|
|
8
|
+
export { OllamaAiConfig };
|
|
9
|
+
//# sourceMappingURL=OllamaAiConfig.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OllamaAiConfig.d.ts","sourceRoot":"","sources":["../../src/config/OllamaAiConfig.ts"],"names":[],"mappings":"AACA,aAAK,cAAc;IAEf,eAAe,uBAAuB;IACtC,aAAa,oBAAoB;IACjC,mBAAmB,2BAA2B;IAC9C,sBAAsB,8BAA8B;CAEvD;AAGD,eAAe,cAAc,CAAC;AAE9B,OAAO,EAAC,cAAc,EAAC,CAAA"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AiProviderFactory.d.ts","sourceRoot":"","sources":["../../src/factory/AiProviderFactory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,2BAA2B,CAAA;
|
|
1
|
+
{"version":3,"file":"AiProviderFactory.d.ts","sourceRoot":"","sources":["../../src/factory/AiProviderFactory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,2BAA2B,CAAA;AAK1D,cAAM,iBAAiB;IACnB,OAAO,CAAC,MAAM,CAAC,UAAU,CAAmC;WAE9C,QAAQ,CAAC,QAAQ,GAAE,MAAiB,GAAG,WAAW;CAkBnE;AAED,eAAe,iBAAiB,CAAA;AAChC,OAAO,EACH,iBAAiB,EACpB,CAAA"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { IAIProvider } from "../interfaces/IAIProvider";
|
|
2
|
+
declare class OllamaAiProviderFactory {
|
|
3
|
+
private static singleton;
|
|
4
|
+
static instance(): IAIProvider;
|
|
5
|
+
}
|
|
6
|
+
export default OllamaAiProviderFactory;
|
|
7
|
+
export { OllamaAiProviderFactory };
|
|
8
|
+
//# sourceMappingURL=OllamaAiProviderFactory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OllamaAiProviderFactory.d.ts","sourceRoot":"","sources":["../../src/factory/OllamaAiProviderFactory.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,2BAA2B,CAAA;AAI1D,cAAM,uBAAuB;IACzB,OAAO,CAAC,MAAM,CAAC,SAAS,CAAc;WAExB,QAAQ,IAAI,WAAW;CAYxC;AAED,eAAe,uBAAuB,CAAA;AACtC,OAAO,EACH,uBAAuB,EAC1B,CAAA"}
|
package/types/index.d.ts
CHANGED
|
@@ -1,12 +1,18 @@
|
|
|
1
1
|
import { OpenAiConfig } from "./config/OpenAiConfig.js";
|
|
2
|
+
import { GoogleAiConfig } from "./config/GoogleAiConfig.js";
|
|
3
|
+
import { OllamaAiConfig } from "./config/OllamaAiConfig.js";
|
|
2
4
|
import { AILogSchema, AILogBaseSchema } from "./schemas/AILogSchema.js";
|
|
3
5
|
import AILogModel from "./models/AILogModel.js";
|
|
4
6
|
import AILogMongoRepository from "./repository/mongo/AILogMongoRepository.js";
|
|
5
7
|
import AILogSqliteRepository from "./repository/sqlite/AILogSqliteRepository.js";
|
|
6
8
|
import { OpenAiProviderFactory } from "./factory/OpenAiProviderFactory.js";
|
|
9
|
+
import { GoogleAiProviderFactory } from "./factory/GoogleAiProviderFactory.js";
|
|
10
|
+
import { OllamaAiProviderFactory } from "./factory/OllamaAiProviderFactory.js";
|
|
7
11
|
import { AiProviderFactory } from "./factory/AiProviderFactory.js";
|
|
8
12
|
import AILogServiceFactory from "./factory/services/AILogServiceFactory.js";
|
|
9
13
|
import { OpenAiProvider } from "./providers/OpenAiProvider.js";
|
|
14
|
+
import { GoogleAiProvider } from "./providers/GoogleAiProvider.js";
|
|
15
|
+
import { OllamaAiProvider } from "./providers/OllamaAiProvider.js";
|
|
10
16
|
import { BuilderTool } from "./tools/BuilderTool.js";
|
|
11
17
|
import { KnowledgeService } from "./services/KnowledgeService.js";
|
|
12
18
|
import { AILogService } from "./services/AILogService.js";
|
|
@@ -31,5 +37,5 @@ import type { DraxAgentControllerOptions } from "./interfaces/IDraxAgentControll
|
|
|
31
37
|
import type { DraxAgentFastifyRoutesOptions } from "./interfaces/IDraxAgentRoutes.js";
|
|
32
38
|
import type { DraxAgentConfig, DraxAgentMessageInput, DraxAgentMessageOutput, DraxAgentPromptContext, DraxAgentSession, DraxAgentSessionInput, DraxAgentToolBuilder, DraxAgentToolBuilderSource, DraxAgentToolSource, DraxAgentSystemPrompt } from "./interfaces/IDraxAgent.js";
|
|
33
39
|
export type { IAILogRepository, IAIProvider, IPromptParams, IPromptMessage, IPromptMemory, IPromptTool, IPromptImage, IPromptImageDetail, IPromptContentPart, IPromptContentPartImage, IPromptContentPartText, IPromptResponse, ToolBuilderMethod, ToolBuilderOptions, ToolBuilderService, DraxAgentControllerOptions, DraxAgentFastifyRoutesOptions, DraxAgentConfig, DraxAgentMessageInput, DraxAgentMessageOutput, DraxAgentPromptContext, DraxAgentSession, DraxAgentSessionInput, DraxAgentToolBuilder, DraxAgentToolBuilderSource, DraxAgentToolSource, DraxAgentSystemPrompt, };
|
|
34
|
-
export { OpenAiConfig, AILogSchema, AILogBaseSchema, AILogModel, AILogMongoRepository, AILogSqliteRepository, OpenAiProviderFactory, AiProviderFactory, AILogServiceFactory, OpenAiProvider, BuilderTool, KnowledgeService, AILogService, AILogPermissions, AgentPermissions, AIPermissions, AgentSessionPermissions, AILogController, AICrudController, AIGenericController, DraxAgentController, AgentSessionController, DraxAgent, AILogRoutes, AIRoutes, DraxAgentRoutes, AgentSessionRoutes };
|
|
40
|
+
export { OpenAiConfig, GoogleAiConfig, OllamaAiConfig, AILogSchema, AILogBaseSchema, AILogModel, AILogMongoRepository, AILogSqliteRepository, OpenAiProviderFactory, GoogleAiProviderFactory, OllamaAiProviderFactory, AiProviderFactory, AILogServiceFactory, OpenAiProvider, GoogleAiProvider, OllamaAiProvider, BuilderTool, KnowledgeService, AILogService, AILogPermissions, AgentPermissions, AIPermissions, AgentSessionPermissions, AILogController, AICrudController, AIGenericController, DraxAgentController, AgentSessionController, DraxAgent, AILogRoutes, AIRoutes, DraxAgentRoutes, AgentSessionRoutes };
|
|
35
41
|
//# sourceMappingURL=index.d.ts.map
|
package/types/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAC,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAC,cAAc,EAAC,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAC,WAAW,EAAE,eAAe,EAAC,MAAM,0BAA0B,CAAC;AACtE,OAAO,UAAU,MAAM,wBAAwB,CAAC;AAChD,OAAO,oBAAoB,MAAM,4CAA4C,CAAC;AAC9E,OAAO,qBAAqB,MAAM,8CAA8C,CAAC;AACjF,OAAO,EAAC,qBAAqB,EAAC,MAAM,oCAAoC,CAAC;AACzE,OAAO,EAAC,uBAAuB,EAAC,MAAM,sCAAsC,CAAC;AAC7E,OAAO,EAAC,iBAAiB,EAAC,MAAM,gCAAgC,CAAC;AACjE,OAAO,mBAAmB,MAAM,2CAA2C,CAAC;AAC5E,OAAO,EAAC,cAAc,EAAC,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAC,gBAAgB,EAAC,MAAM,iCAAiC,CAAC;AACjE,OAAO,EAAC,WAAW,EAAC,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAC,gBAAgB,EAAC,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAC,YAAY,EAAC,MAAM,4BAA4B,CAAC;AACxD,OAAO,gBAAgB,MAAM,mCAAmC,CAAC;AACjE,OAAO,gBAAgB,MAAM,mCAAmC,CAAC;AACjE,OAAO,uBAAuB,MAAM,0CAA0C,CAAC;AAC/E,OAAO,aAAa,MAAM,gCAAgC,CAAC;AAC3D,OAAO,eAAe,MAAM,kCAAkC,CAAC;AAC/D,OAAO,gBAAgB,MAAM,mCAAmC,CAAC;AACjE,OAAO,mBAAmB,MAAM,sCAAsC,CAAC;AACvE,OAAO,mBAAmB,MAAM,sCAAsC,CAAC;AACvE,OAAO,sBAAsB,MAAM,yCAAyC,CAAC;AAC7E,OAAO,WAAW,MAAM,yBAAyB,CAAC;AAClD,OAAO,QAAQ,MAAM,sBAAsB,CAAC;AAC5C,OAAO,eAAe,MAAM,6BAA6B,CAAC;AAC1D,OAAO,kBAAkB,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAC,SAAS,EAAC,MAAM,uBAAuB,CAAC;AAChD,OAAO,KAAK,EAAC,gBAAgB,EAAC,MAAM,kCAAkC,CAAC;AACvE,OAAO,KAAK,EACR,WAAW,EACX,kBAAkB,EAClB,uBAAuB,EACvB,sBAAsB,EACtB,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACd,aAAa,EACb,aAAa,EACb,eAAe,EACf,WAAW,EACd,MAAM,6BAA6B,CAAC;AACrC,OAAO,KAAK,EACR,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EACrB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EACR,0BAA0B,EAC7B,MAAM,sCAAsC,CAAC;AAC9C,OAAO,KAAK,EACR,6BAA6B,EAChC,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EACR,eAAe,EACf,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,qBAAqB,EACrB,oBAAoB,EACpB,0BAA0B,EAC1B,mBAAmB,EACnB,qBAAqB,EACxB,MAAM,4BAA4B,CAAC;AAEpC,YAAY,EAER,gBAAgB,EAChB,WAAW,EACX,aAAa,EACb,cAAc,EACd,aAAa,EACb,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,kBAAkB,EAClB,uBAAuB,EACvB,sBAAsB,EACtB,eAAe,EACf,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,0BAA0B,EAC1B,6BAA6B,EAC7B,eAAe,EACf,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,qBAAqB,EACrB,oBAAoB,EACpB,0BAA0B,EAC1B,mBAAmB,EACnB,qBAAqB,GACxB,CAAA;AAED,OAAO,EACH,YAAY,EACZ,cAAc,EACd,WAAW,EACX,eAAe,EACf,UAAU,EACV,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,EACrB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,EACnB,cAAc,EACd,gBAAgB,EAChB,WAAW,EAEX,gBAAgB,EAChB,YAAY,EAEZ,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EACb,uBAAuB,EAEvB,eAAe,EACf,gBAAgB,EAChB,mBAAmB,EACnB,mBAAmB,EACnB,sBAAsB,EACtB,SAAS,EACT,WAAW,EACX,QAAQ,EACR,eAAe,EACf,kBAAkB,EACrB,CAAA"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAC,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAC,cAAc,EAAC,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAC,cAAc,EAAC,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAC,WAAW,EAAE,eAAe,EAAC,MAAM,0BAA0B,CAAC;AACtE,OAAO,UAAU,MAAM,wBAAwB,CAAC;AAChD,OAAO,oBAAoB,MAAM,4CAA4C,CAAC;AAC9E,OAAO,qBAAqB,MAAM,8CAA8C,CAAC;AACjF,OAAO,EAAC,qBAAqB,EAAC,MAAM,oCAAoC,CAAC;AACzE,OAAO,EAAC,uBAAuB,EAAC,MAAM,sCAAsC,CAAC;AAC7E,OAAO,EAAC,uBAAuB,EAAC,MAAM,sCAAsC,CAAC;AAC7E,OAAO,EAAC,iBAAiB,EAAC,MAAM,gCAAgC,CAAC;AACjE,OAAO,mBAAmB,MAAM,2CAA2C,CAAC;AAC5E,OAAO,EAAC,cAAc,EAAC,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAC,gBAAgB,EAAC,MAAM,iCAAiC,CAAC;AACjE,OAAO,EAAC,gBAAgB,EAAC,MAAM,iCAAiC,CAAC;AACjE,OAAO,EAAC,WAAW,EAAC,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAC,gBAAgB,EAAC,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAC,YAAY,EAAC,MAAM,4BAA4B,CAAC;AACxD,OAAO,gBAAgB,MAAM,mCAAmC,CAAC;AACjE,OAAO,gBAAgB,MAAM,mCAAmC,CAAC;AACjE,OAAO,uBAAuB,MAAM,0CAA0C,CAAC;AAC/E,OAAO,aAAa,MAAM,gCAAgC,CAAC;AAC3D,OAAO,eAAe,MAAM,kCAAkC,CAAC;AAC/D,OAAO,gBAAgB,MAAM,mCAAmC,CAAC;AACjE,OAAO,mBAAmB,MAAM,sCAAsC,CAAC;AACvE,OAAO,mBAAmB,MAAM,sCAAsC,CAAC;AACvE,OAAO,sBAAsB,MAAM,yCAAyC,CAAC;AAC7E,OAAO,WAAW,MAAM,yBAAyB,CAAC;AAClD,OAAO,QAAQ,MAAM,sBAAsB,CAAC;AAC5C,OAAO,eAAe,MAAM,6BAA6B,CAAC;AAC1D,OAAO,kBAAkB,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAC,SAAS,EAAC,MAAM,uBAAuB,CAAC;AAChD,OAAO,KAAK,EAAC,gBAAgB,EAAC,MAAM,kCAAkC,CAAC;AACvE,OAAO,KAAK,EACR,WAAW,EACX,kBAAkB,EAClB,uBAAuB,EACvB,sBAAsB,EACtB,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACd,aAAa,EACb,aAAa,EACb,eAAe,EACf,WAAW,EACd,MAAM,6BAA6B,CAAC;AACrC,OAAO,KAAK,EACR,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EACrB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EACR,0BAA0B,EAC7B,MAAM,sCAAsC,CAAC;AAC9C,OAAO,KAAK,EACR,6BAA6B,EAChC,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EACR,eAAe,EACf,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,qBAAqB,EACrB,oBAAoB,EACpB,0BAA0B,EAC1B,mBAAmB,EACnB,qBAAqB,EACxB,MAAM,4BAA4B,CAAC;AAEpC,YAAY,EAER,gBAAgB,EAChB,WAAW,EACX,aAAa,EACb,cAAc,EACd,aAAa,EACb,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,kBAAkB,EAClB,uBAAuB,EACvB,sBAAsB,EACtB,eAAe,EACf,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,0BAA0B,EAC1B,6BAA6B,EAC7B,eAAe,EACf,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,qBAAqB,EACrB,oBAAoB,EACpB,0BAA0B,EAC1B,mBAAmB,EACnB,qBAAqB,GACxB,CAAA;AAED,OAAO,EACH,YAAY,EACZ,cAAc,EACd,cAAc,EACd,WAAW,EACX,eAAe,EACf,UAAU,EACV,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,EACrB,uBAAuB,EACvB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,EACnB,cAAc,EACd,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EAEX,gBAAgB,EAChB,YAAY,EAEZ,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EACb,uBAAuB,EAEvB,eAAe,EACf,gBAAgB,EAChB,mBAAmB,EACnB,mBAAmB,EACnB,sBAAsB,EACtB,SAAS,EACT,WAAW,EACX,QAAQ,EACR,eAAe,EACf,kBAAkB,EACrB,CAAA"}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
import type { IAIProvider, IPromptContentPart, IPromptMessage, IPromptParams, IPromptResponse, IPromptTool } from "../interfaces/IAIProvider";
|
|
2
|
+
import type { AILogService } from "../services/AILogService";
|
|
3
|
+
import type { IAILogBase } from "@drax/ai-share";
|
|
4
|
+
type OllamaMessage = {
|
|
5
|
+
role: "system" | "user" | "assistant" | "tool";
|
|
6
|
+
content: string;
|
|
7
|
+
images?: string[];
|
|
8
|
+
name?: string;
|
|
9
|
+
};
|
|
10
|
+
type OllamaToolCall = {
|
|
11
|
+
function?: {
|
|
12
|
+
name?: string;
|
|
13
|
+
arguments?: string | object;
|
|
14
|
+
};
|
|
15
|
+
};
|
|
16
|
+
declare class OllamaAiProvider implements IAIProvider {
|
|
17
|
+
protected _baseUrl: string;
|
|
18
|
+
protected _model: string;
|
|
19
|
+
protected _visionModel?: string;
|
|
20
|
+
protected _embeddingModel?: string;
|
|
21
|
+
protected _aiLogService?: AILogService;
|
|
22
|
+
constructor(baseUrl: string, model: string, visionModel?: string, embeddingModel?: string, aiLogService?: AILogService);
|
|
23
|
+
get model(): string;
|
|
24
|
+
protected get visionModel(): string;
|
|
25
|
+
protected get embeddingModel(): string;
|
|
26
|
+
protected post<T>(path: string, body: object): Promise<T>;
|
|
27
|
+
protected buildUserMessage(input: IPromptParams): Promise<OllamaMessage>;
|
|
28
|
+
protected mapContentPartsToMessage(content: IPromptContentPart[], role?: "user" | "assistant" | "system"): Promise<OllamaMessage>;
|
|
29
|
+
protected imageUrlToBase64(url: string): Promise<string>;
|
|
30
|
+
protected mapHistory(history?: IPromptMessage[]): Promise<OllamaMessage[]>;
|
|
31
|
+
protected hasImageInput(input: IPromptParams): boolean;
|
|
32
|
+
protected serializePromptInput(input: IPromptParams, systemPrompt: string): string;
|
|
33
|
+
protected serializePromptOutput(output: unknown): string;
|
|
34
|
+
protected buildLogPayload(input: IPromptParams, params: {
|
|
35
|
+
model: string;
|
|
36
|
+
systemPrompt: string;
|
|
37
|
+
startedAt: Date;
|
|
38
|
+
endedAt?: Date;
|
|
39
|
+
inputTokens?: number;
|
|
40
|
+
outputTokens?: number;
|
|
41
|
+
tokens?: number;
|
|
42
|
+
output?: unknown;
|
|
43
|
+
success: boolean;
|
|
44
|
+
errorMessage?: string;
|
|
45
|
+
}): IAILogBase;
|
|
46
|
+
protected registerPromptLog(input: IPromptParams, params: {
|
|
47
|
+
model: string;
|
|
48
|
+
systemPrompt: string;
|
|
49
|
+
startedAt: Date;
|
|
50
|
+
endedAt?: Date;
|
|
51
|
+
inputTokens?: number;
|
|
52
|
+
outputTokens?: number;
|
|
53
|
+
tokens?: number;
|
|
54
|
+
output?: unknown;
|
|
55
|
+
success: boolean;
|
|
56
|
+
errorMessage?: string;
|
|
57
|
+
}): Promise<void>;
|
|
58
|
+
generateEmbedding({ text, model }: {
|
|
59
|
+
text: string;
|
|
60
|
+
model?: string;
|
|
61
|
+
}): Promise<number[]>;
|
|
62
|
+
protected mapTools(tools?: IPromptTool[]): {
|
|
63
|
+
type: string;
|
|
64
|
+
function: {
|
|
65
|
+
name: string;
|
|
66
|
+
description: string;
|
|
67
|
+
parameters: object;
|
|
68
|
+
};
|
|
69
|
+
}[];
|
|
70
|
+
protected normalizeResponseFormat(input: IPromptParams): any;
|
|
71
|
+
protected parseToolArguments(args: string | object | undefined): any;
|
|
72
|
+
protected serializeToolOutput(output: unknown): string;
|
|
73
|
+
protected buildToolMessages(toolCalls?: OllamaToolCall[], tools?: IPromptTool[]): Promise<OllamaMessage[]>;
|
|
74
|
+
prompt(input: IPromptParams): Promise<IPromptResponse>;
|
|
75
|
+
}
|
|
76
|
+
export default OllamaAiProvider;
|
|
77
|
+
export { OllamaAiProvider };
|
|
78
|
+
//# sourceMappingURL=OllamaAiProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OllamaAiProvider.d.ts","sourceRoot":"","sources":["../../src/providers/OllamaAiProvider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACR,WAAW,EACX,kBAAkB,EAClB,cAAc,EACd,aAAa,EACb,eAAe,EACf,WAAW,EACd,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAC,YAAY,EAAC,MAAM,0BAA0B,CAAC;AAC3D,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,gBAAgB,CAAC;AAE/C,KAAK,aAAa,GAAG;IACjB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;CACjB,CAAA;AAED,KAAK,cAAc,GAAG;IAClB,QAAQ,CAAC,EAAE;QACP,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;KAC/B,CAAA;CACJ,CAAA;AAED,cAAM,gBAAiB,YAAW,WAAW;IACzC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAA;IAC1B,SAAS,CAAC,MAAM,EAAE,MAAM,CAAA;IACxB,SAAS,CAAC,YAAY,CAAC,EAAE,MAAM,CAAA;IAC/B,SAAS,CAAC,eAAe,CAAC,EAAE,MAAM,CAAA;IAClC,SAAS,CAAC,aAAa,CAAC,EAAE,YAAY,CAAA;gBAE1B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,cAAc,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,YAAY;IAgBtH,IAAI,KAAK,WAKR;IAED,SAAS,KAAK,WAAW,WAExB;IAED,SAAS,KAAK,cAAc,WAE3B;cAEe,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC;cAiB/C,gBAAgB,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;cAmB9D,wBAAwB,CAAC,OAAO,EAAE,kBAAkB,EAAE,EAAE,IAAI,GAAE,MAAM,GAAG,WAAW,GAAG,QAAiB,GAAG,OAAO,CAAC,aAAa,CAAC;cAoB/H,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;cAiB9C,UAAU,CAAC,OAAO,GAAE,cAAc,EAAO,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAkBpF,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,aAAa;IAc5C,SAAS,CAAC,oBAAoB,CAAC,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM;IAgBzE,SAAS,CAAC,qBAAqB,CAAC,MAAM,EAAE,OAAO;IAY/C,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE;QACpD,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,EAAE,IAAI,CAAC;QAChB,OAAO,CAAC,EAAE,IAAI,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,OAAO,EAAE,OAAO,CAAC;QACjB,YAAY,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,UAAU;cA+BE,iBAAiB,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE;QAC5D,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,EAAE,IAAI,CAAC;QAChB,OAAO,CAAC,EAAE,IAAI,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,OAAO,EAAE,OAAO,CAAC;QACjB,YAAY,CAAC,EAAE,MAAM,CAAC;KACzB;IAgBK,iBAAiB,CAAC,EAAC,IAAI,EAAE,KAAK,EAAC,EAAE;QAAC,IAAI,EAAC,MAAM,CAAC;QAAC,KAAK,CAAC,EAAC,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IASxF,SAAS,CAAC,QAAQ,CAAC,KAAK,GAAE,WAAW,EAAO;;;;;;;;IAe5C,SAAS,CAAC,uBAAuB,CAAC,KAAK,EAAE,aAAa;IAoBtD,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAgB9D,SAAS,CAAC,mBAAmB,CAAC,MAAM,EAAE,OAAO;cAY7B,iBAAiB,CAAC,SAAS,GAAE,cAAc,EAAO,EAAE,KAAK,GAAE,WAAW,EAAO;IAwBvF,MAAM,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,eAAe,CAAC;CAsG/D;AAGD,eAAe,gBAAgB,CAAA;AAC/B,OAAO,EAAC,gBAAgB,EAAC,CAAA"}
|