Merge pull request #59 from UmstadAI/add-vector-types-for-bot

Add vector types for bot
UmstadAI · Apr 18, 2024 · a4f57d8 · a4f57d8
2 parents 9025e9a + a42a067
commit a4f57d8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/.env.example b/.env.example
@@ -13,6 +13,7 @@ DOCS_VECTOR_TYPE=XXXXX
 CODE_VECTOR_TYPE=XXXXX
 PROJECT_VECTOR_TYPE=XXXXX
 ISSUE_VECTOR_TYPE=XXXXX
+SEARCH_VECTOR_TYPE=XXXXX
 
 # Generate a random secret: https://generate-secret.vercel.app/32 or `openssl rand -base64 32`
 AUTH_SECRET=XXXXXXXX

diff --git a/lib/tools/utils/embeddings.ts b/lib/tools/utils/embeddings.ts
@@ -10,7 +10,7 @@ const openai = new OpenAI(config)
 export async function getEmbeddings(input: string) {
   try {
     const response = await openai.embeddings.create({
-      model: 'text-embedding-ada-002',
+      model: 'text-embedding-3-small',
       input: input.replace(/\n/g, ' ')
     })
 

diff --git a/lib/tools/utils/pinecone.ts b/lib/tools/utils/pinecone.ts
@@ -10,7 +10,7 @@ export type Metadata = {
   hash: string
 }
 
-type VectorType = 'docs' | 'code' | 'project' | 'issue'
+type VectorType = 'docs' | 'code' | 'project' | 'issue' | 'search'
 
 function getVectorType(vector_type: string): string | undefined {
   if (!isVectorType(vector_type)) {
@@ -20,14 +20,15 @@ function getVectorType(vector_type: string): string | undefined {
     docs: process.env.DOCS_VECTOR_TYPE,
     code: process.env.CODE_VECTOR_TYPE,
     project: process.env.PROJECT_VECTOR_TYPE,
-    issue: process.env.ISSUE_VECTOR_TYPE
+    issue: process.env.ISSUE_VECTOR_TYPE,
+    search: process.env.SEARCH_VECTOR_TYPE
   }
 
   return vectorTypeMap[vector_type]
 }
 
 function isVectorType(type: string): type is VectorType {
-  return ['docs', 'code', 'project', 'issue'].includes(type)
+  return ['docs', 'code', 'project', 'issue', 'search'].includes(type)
 }
 
 const getMatchesFromEmbeddings = async (