chore: add otel info for pinecone

core/api/package.json

@@ -106,7 +106,7 @@
  "medici": "^7.0.0",
  "mongoose": "8.3.2",
  "node-cache": "^5.1.2",
- "openai": "^4.43.0",
+ "openai": "^4.46.1",
  "pg": "^8.11.5",
  "pino": "9.0.0",
  "pino-http": "^10.1.0",

core/api/src/services/openai/pinecone.ts

@@ -2,6 +2,7 @@ import axios from "axios"
 
 import { env } from "@/config/env"
 import { UnknownPineconeError } from "@/domain/support/errors"
+import { SemanticAttributes, asyncRunInSpan } from "../tracing"
 
 interface Match {
  metadata?: {
@@ -11,31 +12,42 @@ interface Match {
 
 export const retrieveRelatedQueries = async (
  vector: number[],
-): Promise<string[] | Error> => {
- const topK = 8 // how many results to retrieve
+): Promise<string[] | Error> =>
+ asyncRunInSpan(
+ "app.chatbot.retrieveRelatedQueries",
+ {
+ attributes: {
+ [SemanticAttributes.CODE_FUNCTION]: "retrieveRelatedQueries",
+ [SemanticAttributes.CODE_NAMESPACE]: "chatbot",
+ vector,
+ },
+ },
+ async () => {
+ const topK = 8 // how many results to retrieve
 
- try {
- // TODO: fetch programmatically the endpoint. doc is not clear.
- const url = `https://blink-3072-5c8b1i4.svc.aped-4627-b74a.pinecone.io/query`
- const headers = {
- "Api-Key": env.PINECONE_API_KEY ?? "unknown",
- "Content-Type": "application/json",
- }
+  try {
+  // TODO: fetch programmatically the endpoint. doc is not clear.
+  const url = `https://blink-3072-5c8b1i4.svc.aped-4627-b74a.pinecone.io/query`
+  const headers = {
+  "Api-Key": env.PINECONE_API_KEY ?? "unknown",
+  "Content-Type": "application/json",
+  }
 
- const data = {
- vector,
- topK,
- includeMetadata: true,
- }
+  const data = {
+  vector,
+  topK,
+  includeMetadata: true,
+  }
 
- const response = await axios.post(url, data, { headers })
+  const response = await axios.post(url, data, { headers })
 
- const results = response.data.matches.map(
- (match: Match) => JSON.parse(match.metadata?._node_content ?? "{}").text,
- )
- return results as string[]
- } catch (error) {
- console.error(error)
- return new UnknownPineconeError(error)
- }
-}
+ const results = response.data.matches.map(
+ (match: Match) => JSON.parse(match.metadata?._node_content ?? "{}").text,
+ )
+ return results as string[]
+ } catch (error) {
+ console.error(error)
+ return new UnknownPineconeError(error)
+ }
+ },
+ )