feat(pinecone): added pinecone search by vector tool & fixed dropdown tag for evaluator

This commit is contained in:
Waleed Latif
2025-02-24 11:10:23 -08:00
parent 1309c71086
commit 1ad9746b2b
5 changed files with 147 additions and 5 deletions

View File

@@ -21,6 +21,7 @@ import { embeddingsTool as openAIEmbeddings } from './openai/embeddings'
import { fetchTool as pineconeFetchTool } from './pinecone/fetch'
import { generateEmbeddingsTool as pineconeGenerateEmbeddingsTool } from './pinecone/generate'
import { searchTextTool as pineconeSearchTextTool } from './pinecone/searchText'
import { searchVectorTool as pineconeSearchVectorTool } from './pinecone/searchVector'
import { upsertTextTool as pineconeUpsertTextTool } from './pinecone/upsertText'
import { opportunitiesTool as salesforceOpportunities } from './salesforce/opportunities'
import { searchTool as serperSearch } from './serper/search'
@@ -69,6 +70,7 @@ export const tools: Record<string, ToolConfig> = {
pinecone_fetch: pineconeFetchTool,
pinecone_generate_embeddings: pineconeGenerateEmbeddingsTool,
pinecone_search_text: pineconeSearchTextTool,
pinecone_search_vector: pineconeSearchVectorTool,
pinecone_upsert_text: pineconeUpsertTextTool,
github_pr: prTool,
github_comment: commentTool,

View File

@@ -0,0 +1,78 @@
import { ToolConfig } from '../types'
import { PineconeResponse, PineconeSearchVectorParams } from './types'
export const searchVectorTool: ToolConfig<PineconeSearchVectorParams, PineconeResponse> = {
id: 'pinecone_search_vector',
name: 'Pinecone Search Vector',
description: 'Search for similar vectors in a Pinecone index',
version: '1.0',
params: {
apiKey: {
type: 'string',
required: true,
requiredForToolCall: true,
description: 'Pinecone API key',
},
indexHost: {
type: 'string',
required: true,
requiredForToolCall: true,
description: 'Full Pinecone index host URL',
},
namespace: { type: 'string', required: false, description: 'Namespace to search in' },
vector: { type: 'array', required: true, description: 'Vector to search for' },
topK: { type: 'number', required: false, description: 'Number of results to return' },
filter: { type: 'object', required: false, description: 'Filter to apply to the search' },
includeValues: {
type: 'boolean',
required: false,
description: 'Include vector values in response',
},
includeMetadata: {
type: 'boolean',
required: false,
description: 'Include metadata in response',
},
},
request: {
method: 'POST',
url: (params) => `${params.indexHost}/query`,
headers: (params) => ({
'Api-Key': params.apiKey,
'Content-Type': 'application/json',
Accept: 'application/json',
}),
body: (params) => ({
namespace: params.namespace,
vector: typeof params.vector === 'string' ? JSON.parse(params.vector) : params.vector,
topK: params.topK ? parseInt(params.topK.toString()) : 10,
filter: params.filter
? typeof params.filter === 'string'
? JSON.parse(params.filter)
: params.filter
: undefined,
includeValues: true, //TODO: Make this dynamic
includeMetadata: true, //TODO: Make this dynamic
}),
},
transformResponse: async (response) => {
const data = await response.json()
return {
success: true,
output: {
matches: data.matches.map((match: any) => ({
id: match.id,
score: match.score,
values: match.values,
metadata: match.metadata,
})),
namespace: data.namespace,
},
}
},
transformError: (error) => `Pinecone vector search failed: ${error.message}`,
}

View File

@@ -150,3 +150,12 @@ export interface PineconeParams {
metadata?: Record<string, any>
}>
}
// Search Vector
export interface PineconeSearchVectorParams extends PineconeBaseParams {
vector: number[] | string
topK?: number | string
filter?: Record<string, any> | string
includeValues?: boolean
includeMetadata?: boolean
}