@sweetoburrito/backstage-plugin-ai-assistant-node 0.4.0 → 0.5.0

package/dist/extensions.cjs.js

@@ -11,8 +11,12 @@ const embeddingsProviderExtensionPoint = backendPluginApi.createExtensionPoint({
 const modelProviderExtensionPoint = backendPluginApi.createExtensionPoint({
   id: "ai-assistant.model-provider"
 });
+const toolExtensionPoint = backendPluginApi.createExtensionPoint({
+  id: "ai-assistant.tool"
+});
 
 exports.dataIngestorExtensionPoint = dataIngestorExtensionPoint;
 exports.embeddingsProviderExtensionPoint = embeddingsProviderExtensionPoint;
 exports.modelProviderExtensionPoint = modelProviderExtensionPoint;
+exports.toolExtensionPoint = toolExtensionPoint;
 //# sourceMappingURL=extensions.cjs.js.map

package/dist/extensions.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"extensions.cjs.js","sources":["../src/extensions.ts"],"sourcesContent":["import { createExtensionPoint } from '@backstage/backend-plugin-api';\nimport { Ingestor } from './types/ingestor';\nimport { Embeddings } from '@langchain/core/embeddings';\nimport { BaseChatModel } from '@langchain/core/language_models/chat_models';\n\nexport type DataIngestorExtensionPoint = {\n  registerIngestor: (ingestor: Ingestor) => void;\n};\n\nexport const dataIngestorExtensionPoint =\n  createExtensionPoint<DataIngestorExtensionPoint>({\n    id: 'ai-assistant.data-ingestor',\n  });\n\nexport type EmbeddingsProvider = {\n  getEmbeddings: () => Promise<Omit<Embeddings, 'caller'>>;\n};\n\nexport type EmbeddingsProviderExtensionPoint = {\n  register: (provider: EmbeddingsProvider) => void;\n};\n\nexport const embeddingsProviderExtensionPoint =\n  createExtensionPoint<EmbeddingsProviderExtensionPoint>({\n    id: 'ai-assistant.embeddings-provider',\n  });\n\nexport type Model = {\n  id: string;\n  chatModel: BaseChatModel;\n};\n\nexport type ModelProviderExtensionPoint = {\n  register: (model: Model) => void;\n};\n\nexport const modelProviderExtensionPoint =\n  createExtensionPoint<ModelProviderExtensionPoint>({\n    id: 'ai-assistant.model-provider',\n  });\n"],"names":["createExtensionPoint"],"mappings":";;;;AASO,MAAM,6BACXA,qCAAA,CAAiD;AAAA,EAC/C,EAAA,EAAI;AACN,CAAC;AAUI,MAAM,mCACXA,qCAAA,CAAuD;AAAA,EACrD,EAAA,EAAI;AACN,CAAC;AAWI,MAAM,8BACXA,qCAAA,CAAkD;AAAA,EAChD,EAAA,EAAI;AACN,CAAC;;;;;;"}
+{"version":3,"file":"extensions.cjs.js","sources":["../src/extensions.ts"],"sourcesContent":["import { createExtensionPoint } from '@backstage/backend-plugin-api';\nimport { Ingestor } from './types/ingestor';\nimport { Embeddings } from '@langchain/core/embeddings';\nimport { BaseChatModel } from '@langchain/core/language_models/chat_models';\nimport { Tool } from './types';\nimport { ZodType } from 'zod';\n\nexport type DataIngestorExtensionPoint = {\n  registerIngestor: (ingestor: Ingestor) => void;\n};\n\nexport const dataIngestorExtensionPoint =\n  createExtensionPoint<DataIngestorExtensionPoint>({\n    id: 'ai-assistant.data-ingestor',\n  });\n\nexport type EmbeddingsProvider = {\n  getEmbeddings: () => Promise<Omit<Embeddings, 'caller'>>;\n};\n\nexport type EmbeddingsProviderExtensionPoint = {\n  register: (provider: EmbeddingsProvider) => void;\n};\n\nexport const embeddingsProviderExtensionPoint =\n  createExtensionPoint<EmbeddingsProviderExtensionPoint>({\n    id: 'ai-assistant.embeddings-provider',\n  });\n\nexport type Model = {\n  id: string;\n  chatModel: BaseChatModel;\n};\n\nexport type ModelProviderExtensionPoint = {\n  register: (model: Model) => void;\n};\n\nexport const modelProviderExtensionPoint =\n  createExtensionPoint<ModelProviderExtensionPoint>({\n    id: 'ai-assistant.model-provider',\n  });\n\nexport type ToolExtensionPoint = {\n  register: (tool: Tool<ZodType>) => void;\n};\n\nexport const toolExtensionPoint = createExtensionPoint<ToolExtensionPoint>({\n  id: 'ai-assistant.tool',\n});\n"],"names":["createExtensionPoint"],"mappings":";;;;AAWO,MAAM,6BACXA,qCAAA,CAAiD;AAAA,EAC/C,EAAA,EAAI;AACN,CAAC;AAUI,MAAM,mCACXA,qCAAA,CAAuD;AAAA,EACrD,EAAA,EAAI;AACN,CAAC;AAWI,MAAM,8BACXA,qCAAA,CAAkD;AAAA,EAChD,EAAA,EAAI;AACN,CAAC;AAMI,MAAM,qBAAqBA,qCAAA,CAAyC;AAAA,EACzE,EAAA,EAAI;AACN,CAAC;;;;;;;"}

package/dist/index.cjs.js

@@ -1,6 +1,7 @@
 'use strict';
 
 var extensions = require('./extensions.cjs.js');
+var tool = require('./utils/tool.cjs.js');
 var streamToString = require('./utils/stream-to-string.cjs.js');
 
 
@@ -8,5 +9,7 @@ var streamToString = require('./utils/stream-to-string.cjs.js');
 exports.dataIngestorExtensionPoint = extensions.dataIngestorExtensionPoint;
 exports.embeddingsProviderExtensionPoint = extensions.embeddingsProviderExtensionPoint;
 exports.modelProviderExtensionPoint = extensions.modelProviderExtensionPoint;
+exports.toolExtensionPoint = extensions.toolExtensionPoint;
+exports.createAssistantTool = tool.createAssistantTool;
 exports.streamToString = streamToString.streamToString;
 //# sourceMappingURL=index.cjs.js.map

package/dist/index.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;"}
+{"version":3,"file":"index.cjs.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;"}

package/dist/index.d.ts

@@ -1,6 +1,7 @@
 import * as _backstage_backend_plugin_api from '@backstage/backend-plugin-api';
 import { LoggerService, RootConfigService, SchedulerService } from '@backstage/backend-plugin-api';
 import { Embeddings } from '@langchain/core/embeddings';
+import { ZodType, z } from 'zod';
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 
 type EmbeddingsSource = string;
@@ -54,6 +55,13 @@ type DataIngestionPipeline = {
     start: () => Promise<void>;
 };
 
+type Tool<T extends ZodType<any, any, any> = ZodType<any, any, any>> = {
+    name: string;
+    description: string;
+    schema: T;
+    func: (params: z.infer<T>) => Promise<string>;
+};
+
 type DataIngestorExtensionPoint = {
     registerIngestor: (ingestor: Ingestor) => void;
 };
@@ -73,8 +81,17 @@ type ModelProviderExtensionPoint = {
     register: (model: Model) => void;
 };
 declare const modelProviderExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<ModelProviderExtensionPoint>;
+type ToolExtensionPoint = {
+    register: (tool: Tool<ZodType>) => void;
+};
+declare const toolExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<ToolExtensionPoint>;
+
+type CreateAssistantToolOptions<T extends ZodType<any, any, any>> = {
+    tool: Tool<T>;
+};
+declare const createAssistantTool: <T extends ZodType<any, any, any>>(options: CreateAssistantToolOptions<T>) => Tool<T>;
 
 declare function streamToString(readableStream: NodeJS.ReadableStream | null): Promise<string>;
 
-export { dataIngestorExtensionPoint, embeddingsProviderExtensionPoint, modelProviderExtensionPoint, streamToString };
-export type { AugmentationIndexer, DataIngestionPipeline, DataIngestionPipelineOptions, DataIngestorExtensionPoint, Embedding, EmbeddingDocument, EmbeddingDocumentMetadata, EmbeddingsProvider, EmbeddingsProviderExtensionPoint, EmbeddingsSource, EntityFilterShape, Ingestor, IngestorOptions, Model, ModelProviderExtensionPoint, VectorStore };
+export { createAssistantTool, dataIngestorExtensionPoint, embeddingsProviderExtensionPoint, modelProviderExtensionPoint, streamToString, toolExtensionPoint };
+export type { AugmentationIndexer, DataIngestionPipeline, DataIngestionPipelineOptions, DataIngestorExtensionPoint, Embedding, EmbeddingDocument, EmbeddingDocumentMetadata, EmbeddingsProvider, EmbeddingsProviderExtensionPoint, EmbeddingsSource, EntityFilterShape, Ingestor, IngestorOptions, Model, ModelProviderExtensionPoint, Tool, ToolExtensionPoint, VectorStore };

package/dist/utils/tool.cjs.js

@@ -0,0 +1,8 @@
+'use strict';
+
+const createAssistantTool = (options) => {
+  return options.tool;
+};
+
+exports.createAssistantTool = createAssistantTool;
+//# sourceMappingURL=tool.cjs.js.map

package/dist/utils/tool.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.cjs.js","sources":["../../src/utils/tool.ts"],"sourcesContent":["import { ZodType } from 'zod';\nimport { Tool } from '../types';\n\ntype CreateAssistantToolOptions<T extends ZodType<any, any, any>> = {\n  tool: Tool<T>;\n};\n\nexport const createAssistantTool = <T extends ZodType<any, any, any>>(\n  options: CreateAssistantToolOptions<T>,\n) => {\n  return options.tool;\n};\n"],"names":[],"mappings":";;AAOO,MAAM,mBAAA,GAAsB,CACjC,OAAA,KACG;AACH,EAAA,OAAO,OAAA,CAAQ,IAAA;AACjB;;;;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sweetoburrito/backstage-plugin-ai-assistant-node",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "license": "Apache-2.0",
   "description": "Node.js library for the ai-assistant plugin",
   "main": "dist/index.cjs.js",
@@ -36,7 +36,8 @@
   ],
   "dependencies": {
     "@backstage/backend-plugin-api": "backstage:^",
-    "@langchain/core": "^0.3.72"
+    "@langchain/core": "^0.3.72",
+    "zod": "^4.1.11"
   },
   "typesVersions": {
     "*": {