@drax/ai-back 3.32.0 → 3.35.1

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.
Files changed (59) hide show
  1. package/dist/config/GoogleAiConfig.js +8 -0
  2. package/dist/config/OllamaAiConfig.js +9 -0
  3. package/dist/factory/AiProviderFactory.js +12 -3
  4. package/dist/factory/GoogleAiProviderFactory.js +14 -0
  5. package/dist/factory/OllamaAiProviderFactory.js +14 -0
  6. package/dist/index.js +7 -1
  7. package/dist/providers/GoogleAiProvider.js +367 -0
  8. package/dist/providers/OllamaAiProvider.js +342 -0
  9. package/dist/tools/BuilderTool.js +9 -0
  10. package/package.json +4 -2
  11. package/src/config/GoogleAiConfig.ts +13 -0
  12. package/src/config/OllamaAiConfig.ts +14 -0
  13. package/src/factory/AiProviderFactory.ts +12 -5
  14. package/src/factory/GoogleAiProviderFactory.ts +26 -0
  15. package/src/factory/OllamaAiProviderFactory.ts +27 -0
  16. package/src/index.ts +12 -0
  17. package/src/providers/GoogleAiProvider.ts +489 -0
  18. package/src/providers/OllamaAiProvider.ts +469 -0
  19. package/src/tools/BuilderTool.ts +12 -0
  20. package/test/GoogleAiProvider.test.ts +211 -0
  21. package/test/ToolBuilder.test.ts +48 -0
  22. package/tsconfig.tsbuildinfo +1 -1
  23. package/types/config/GoogleAiConfig.d.ts +8 -0
  24. package/types/config/GoogleAiConfig.d.ts.map +1 -0
  25. package/types/config/OllamaAiConfig.d.ts +9 -0
  26. package/types/config/OllamaAiConfig.d.ts.map +1 -0
  27. package/types/factory/AiProviderFactory.d.ts.map +1 -1
  28. package/types/factory/GoogleAiProviderFactory.d.ts +8 -0
  29. package/types/factory/GoogleAiProviderFactory.d.ts.map +1 -0
  30. package/types/factory/OllamaAiProviderFactory.d.ts +8 -0
  31. package/types/factory/OllamaAiProviderFactory.d.ts.map +1 -0
  32. package/types/index.d.ts.map +1 -1
  33. package/types/providers/GoogleAiProvider.d.ts +63 -0
  34. package/types/providers/GoogleAiProvider.d.ts.map +1 -0
  35. package/types/providers/OllamaAiProvider.d.ts +78 -0
  36. package/types/providers/OllamaAiProvider.d.ts.map +1 -0
  37. package/types/tools/BuilderTool.d.ts.map +1 -1
  38. package/.env +0 -4
  39. package/dist/agents/ChatbotTaskService.js +0 -143
  40. package/dist/agents/ChatbotTaskTools.js +0 -756
  41. package/dist/controllers/AIController.js +0 -150
  42. package/dist/interfaces/IAILog.js +0 -1
  43. package/dist/routes/ChatbotTaskRoutes.js +0 -8
  44. package/dist/tools/ToolBuilder.js +0 -243
  45. package/dist/vectors/ChromaVector.js +0 -65
  46. package/types/agents/ChatbotTaskService.d.ts +0 -42
  47. package/types/agents/ChatbotTaskService.d.ts.map +0 -1
  48. package/types/agents/ChatbotTaskTools.d.ts +0 -54
  49. package/types/agents/ChatbotTaskTools.d.ts.map +0 -1
  50. package/types/controllers/AIController.d.ts +0 -25
  51. package/types/controllers/AIController.d.ts.map +0 -1
  52. package/types/interfaces/IAILog.d.ts +0 -77
  53. package/types/interfaces/IAILog.d.ts.map +0 -1
  54. package/types/routes/ChatbotTaskRoutes.d.ts +0 -4
  55. package/types/routes/ChatbotTaskRoutes.d.ts.map +0 -1
  56. package/types/tools/ToolBuilder.d.ts +0 -47
  57. package/types/tools/ToolBuilder.d.ts.map +0 -1
  58. package/types/vectors/ChromaVector.d.ts +0 -21
  59. package/types/vectors/ChromaVector.d.ts.map +0 -1
@@ -1 +0,0 @@
1
- {"version":3,"file":"IAILog.d.ts","sourceRoot":"","sources":["../../src/interfaces/IAILog.ts"],"names":[],"mappings":"AACA,UAAU,UAAU;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,WAAW,CAAC,EAAE,KAAK,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,IAAI,CAAC,EAAE,MAAM,CAAA;QACb,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,GAAG,CAAC,EAAE,MAAM,CAAA;KACX,CAAC,CAAA;IACF,UAAU,CAAC,EAAE,KAAK,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,IAAI,CAAC,EAAE,MAAM,CAAA;QACb,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,GAAG,CAAC,EAAE,MAAM,CAAA;KACX,CAAC,CAAA;IACF,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,CAAC,EAAE,IAAI,CAAA;IAChB,OAAO,CAAC,EAAE,IAAI,CAAA;IACd,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,GAAG,CAAA;IACZ,IAAI,CAAC,EAAE,GAAG,CAAA;IACV,SAAS,CAAC,EAAE,IAAI,CAAA;IAChB,SAAS,CAAC,EAAE,IAAI,CAAA;CACnB;AAED,UAAU,MAAM;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,WAAW,CAAC,EAAE,KAAK,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,IAAI,CAAC,EAAE,MAAM,CAAA;QACb,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,GAAG,CAAC,EAAE,MAAM,CAAA;KACX,CAAC,CAAA;IACF,UAAU,CAAC,EAAE,KAAK,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,IAAI,CAAC,EAAE,MAAM,CAAA;QACb,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,GAAG,CAAC,EAAE,MAAM,CAAA;KACX,CAAC,CAAA;IACF,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,CAAC,EAAE,IAAI,CAAA;IAChB,OAAO,CAAC,EAAE,IAAI,CAAA;IACd,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,GAAG,CAAA;IACZ,IAAI,CAAC,EAAE,GAAG,CAAA;IACV,SAAS,CAAC,EAAE,IAAI,CAAA;IAChB,SAAS,CAAC,EAAE,IAAI,CAAA;CACnB;AAED,YAAY,EACZ,UAAU,EACV,MAAM,EACL,CAAA"}
@@ -1,4 +0,0 @@
1
- declare function ChatbotTaskFastifyRoutes(fastify: any, options: any): Promise<void>;
2
- export default ChatbotTaskFastifyRoutes;
3
- export { ChatbotTaskFastifyRoutes };
4
- //# sourceMappingURL=ChatbotTaskRoutes.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"ChatbotTaskRoutes.d.ts","sourceRoot":"","sources":["../../src/routes/ChatbotTaskRoutes.ts"],"names":[],"mappings":"AAEA,iBAAe,wBAAwB,CAAC,OAAO,KAAA,EAAE,OAAO,KAAA,iBAKvD;AAED,eAAe,wBAAwB,CAAC;AACxC,OAAO,EAAC,wBAAwB,EAAC,CAAC"}
@@ -1,47 +0,0 @@
1
- import { z, type ZodObject, type ZodRawShape, type ZodTypeAny } from "zod";
2
- import type { IPromptTool } from "../interfaces/IAIProvider.js";
3
- import type { IDraxCrudService } from "@drax/crud-share";
4
- type ToolBuilderMethod = "create" | "update" | "updatePartial" | "delete" | "findById" | "findByIds" | "findOneBy" | "findOne" | "findBy" | "fetchAll" | "search" | "find" | "paginate" | "groupBy";
5
- type ToolBuilderService<T = any, C = any, U = any> = IDraxCrudService<T, C, U> & Record<string, any>;
6
- interface ToolBuilderOptions<T = any, C = any, U = any> {
7
- entityName: string;
8
- entityDescription?: string;
9
- schema: ZodObject<ZodRawShape>;
10
- outputSchema?: ZodObject<ZodRawShape>;
11
- service: ToolBuilderService<T, C, U>;
12
- methods: ToolBuilderMethod[];
13
- toolNamePrefix?: string;
14
- }
15
- declare class ToolBuilder<T = any, C = any, U = any> {
16
- protected options: ToolBuilderOptions<T, C, U>;
17
- protected _inputSchema: ZodObject<ZodRawShape>;
18
- protected _partialInputSchema: ZodObject<ZodRawShape>;
19
- protected _outputSchema?: ZodObject<ZodRawShape>;
20
- constructor(options: ToolBuilderOptions<T, C, U>);
21
- get inputSchema(): z.ZodObject<Readonly<{
22
- [k: string]: z.core.$ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
23
- }>, z.core.$strip>;
24
- get partialInputSchema(): z.ZodObject<Readonly<{
25
- [k: string]: z.core.$ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
26
- }>, z.core.$strip>;
27
- get outputSchema(): z.ZodObject<Readonly<{
28
- [k: string]: z.core.$ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
29
- }>, z.core.$strip>;
30
- getTools(): IPromptTool[];
31
- getSystemPromptSection(): string;
32
- objectParameters(schema: ZodObject<ZodRawShape>): z.core.ZodStandardJSONSchemaPayload<z.ZodObject<Readonly<{
33
- [k: string]: z.core.$ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
34
- }>, z.core.$strip>>;
35
- protected buildTool(method: ToolBuilderMethod): IPromptTool;
36
- protected getToolName(method: ToolBuilderMethod): string;
37
- protected toJsonSchema(schema: ZodObject<ZodRawShape>): z.core.ZodStandardJSONSchemaPayload<z.ZodObject<Readonly<{
38
- [k: string]: z.core.$ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
39
- }>, z.core.$strip>>;
40
- protected getTypeName(field: any): string | undefined;
41
- protected fieldAdapter(field: unknown): ZodTypeAny;
42
- protected schemaAdapter<TObj extends ZodObject<ZodRawShape>>(schema: TObj): TObj;
43
- }
44
- export default ToolBuilder;
45
- export { ToolBuilder };
46
- export type { ToolBuilderMethod, ToolBuilderOptions, ToolBuilderService };
47
- //# sourceMappingURL=ToolBuilder.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"ToolBuilder.d.ts","sourceRoot":"","sources":["../../src/tools/ToolBuilder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAE,KAAK,SAAS,EAAE,KAAK,WAAW,EAAE,KAAK,UAAU,EAAC,MAAM,KAAK,CAAC;AACzE,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,8BAA8B,CAAC;AAC9D,OAAO,KAAK,EAAC,gBAAgB,EAAC,MAAM,kBAAkB,CAAC;AAEvD,KAAK,iBAAiB,GAChB,QAAQ,GACR,QAAQ,GACR,eAAe,GACf,QAAQ,GACR,UAAU,GACV,WAAW,GACX,WAAW,GACX,SAAS,GACT,QAAQ,GACR,UAAU,GACV,QAAQ,GACR,MAAM,GACN,UAAU,GACV,SAAS,CAAC;AAEhB,KAAK,kBAAkB,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG,IAAI,gBAAgB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAErG,UAAU,kBAAkB,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG;IAClD,UAAU,EAAE,MAAM,CAAC;IACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,MAAM,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;IAC/B,YAAY,CAAC,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;IACtC,OAAO,EAAE,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,EAAE,iBAAiB,EAAE,CAAC;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;CAC3B;AAiKD,cAAM,WAAW,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG;IAK3B,SAAS,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAJ1D,SAAS,CAAC,YAAY,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;IAC/C,SAAS,CAAC,mBAAmB,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;IACtD,SAAS,CAAC,aAAa,CAAC,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;gBAE3B,OAAO,EAAE,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAM1D,IAAI,WAAW;;uBAEd;IAED,IAAI,kBAAkB;;uBAErB;IAED,IAAI,YAAY;;uBAEf;IAED,QAAQ,IAAI,WAAW,EAAE;IAIzB,sBAAsB,IAAI,MAAM;IAsBhC,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,WAAW,CAAC;;;IAI/C,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,iBAAiB,GAAG,WAAW;IAgB3D,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE,iBAAiB;IAI/C,SAAS,CAAC,YAAY,CAAC,MAAM,EAAE,SAAS,CAAC,WAAW,CAAC;;;IAIrD,SAAS,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,GAAG,MAAM,GAAG,SAAS;IAIrD,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,UAAU;IA2BlD,SAAS,CAAC,aAAa,CAAC,IAAI,SAAS,SAAS,CAAC,WAAW,CAAC,EAAE,MAAM,EAAE,IAAI,GAAG,IAAI;CAUnF;AAED,eAAe,WAAW,CAAC;AAC3B,OAAO,EAAC,WAAW,EAAC,CAAC;AACrB,YAAY,EAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,EAAC,CAAC"}
@@ -1,21 +0,0 @@
1
- import { Document } from 'chromadb';
2
- interface VectorDocument {
3
- id: string;
4
- content: string;
5
- metadata?: Record<string, any>;
6
- embedding?: number[];
7
- }
8
- declare class ChromaVector {
9
- private client;
10
- private collections;
11
- constructor(path?: string, database?: string);
12
- initializeCollection(collectionName: string): Promise<void>;
13
- addDocument(collectionName: string, document: VectorDocument): Promise<void>;
14
- queryCollection(collectionName: string, query: string, topK?: number): Promise<Document[]>;
15
- deleteDocument(collectionName: string, documentId: string): Promise<void>;
16
- listCollections(): Promise<string[]>;
17
- deleteCollection(collectionName: string): Promise<void>;
18
- }
19
- export default ChromaVector;
20
- export { ChromaVector };
21
- //# sourceMappingURL=ChromaVector.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"ChromaVector.d.ts","sourceRoot":"","sources":["../../src/vectors/ChromaVector.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2B,QAAQ,EAAC,MAAM,UAAU,CAAC;AAE5D,UAAU,cAAc;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,cAAM,YAAY;IAChB,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,WAAW,CAA0B;gBAEjC,IAAI,GAAE,MAAgC,EAAE,QAAS,CAAC,EAAE,MAAM;IAKhE,oBAAoB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU3D,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAe5E,eAAe,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,MAAU,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAc7F,cAAc,CAAC,cAAc,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAWzE,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAIpC,gBAAgB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAI9D;AAED,eAAe,YAAY,CAAC;AAC5B,OAAO,EAAC,YAAY,EAAC,CAAA"}