diff --git a/packages/ai-sdk/package.json b/packages/ai-sdk/package.json index 24e4933c..43de702d 100644 --- a/packages/ai-sdk/package.json +++ b/packages/ai-sdk/package.json @@ -1,6 +1,6 @@ { "name": "@supermemory/ai-sdk", - "version": "1.0.1", + "version": "1.0.3", "scripts": { "build": "tsdown", "dev": "tsdown --watch", diff --git a/packages/ai-sdk/src/infinite-chat.ts b/packages/ai-sdk/src/infinite-chat.ts index 4062e2d4..50e8bb0e 100644 --- a/packages/ai-sdk/src/infinite-chat.ts +++ b/packages/ai-sdk/src/infinite-chat.ts @@ -35,15 +35,15 @@ const providerMap = { export const createSupermemoryInfiniteChat = ( apiKey: SupermemoryApiKey, - config: SupermemoryInfiniteChatConfig, + config?: SupermemoryInfiniteChatConfig, ) => createOpenAI({ - apiKey: config.providerApiKey, - baseURL: config.providerName + apiKey: config?.providerApiKey, + baseURL: config?.providerName ? providerMap[config.providerName] - : config.providerUrl, + : config?.providerUrl, headers: { "x-supermemory-api-key": apiKey, - ...config.headers, + ...config?.headers, }, }).chat diff --git a/packages/ai-sdk/src/tools.ts b/packages/ai-sdk/src/tools.ts index 7d99e2c7..fd78940c 100644 --- a/packages/ai-sdk/src/tools.ts +++ b/packages/ai-sdk/src/tools.ts @@ -17,16 +17,16 @@ export interface SupermemoryToolsConfig { */ export function supermemoryTools( apiKey: string, - config: SupermemoryToolsConfig, + config?: SupermemoryToolsConfig, ) { const client = new Supermemory({ apiKey, - ...(config.baseUrl && { baseURL: config.baseUrl }), + ...(config?.baseUrl && { baseURL: config.baseUrl }), }) - const containerTags = config.projectId - ? [`sm_project_${config.projectId}`] - : (config.containerTags ?? ["sm_project_default"]) + const containerTags = config?.projectId + ? [`sm_project_${config?.projectId}`] + : (config?.containerTags ?? ["sm_project_default"]) const searchMemories = tool({ description: @@ -124,7 +124,7 @@ export function supermemoryTools( // Export individual tool creators for more flexibility export const searchMemoriesTool = ( apiKey: string, - config: SupermemoryToolsConfig, + config?: SupermemoryToolsConfig, ) => { const { searchMemories } = supermemoryTools(apiKey, config) return searchMemories @@ -132,7 +132,7 @@ export const searchMemoriesTool = ( export const addMemoryTool = ( apiKey: string, - config: SupermemoryToolsConfig, + config?: SupermemoryToolsConfig, ) => { const { addMemory } = supermemoryTools(apiKey, config) return addMemory @@ -140,7 +140,7 @@ export const addMemoryTool = ( export const fetchMemoryTool = ( apiKey: string, - config: SupermemoryToolsConfig, + config?: SupermemoryToolsConfig, ) => { const { fetchMemory } = supermemoryTools(apiKey, config) return fetchMemory