Add ollama and add more event types

src/typescript/constants/common.ts

@@ -12,6 +12,7 @@ import { PineConeFunctionNames, PineConeFunctions } from './pinecone'
 import { QdrantFunctionNames, QdrantFunctions } from './qdrant'
 import { WeaviateFunctionNames, WeaviateFunctions } from './weaviate'
 import { TiktokenModel, TiktokenEncoding } from 'tiktoken'
+import { OllamaFunctionNames, OllamaFunctions } from './ollama'
 
 export const LLMSpanAttributeNames: Array<keyof typeof LLMSpanAttributesObj> = Object.keys(LLMSpanAttributesObj) as Array<keyof typeof LLMSpanAttributesObj>
 export const DatabaseSpanAttributeNames: Array<keyof typeof DatabaseSpanAttributesObj> = Object.keys(DatabaseSpanAttributesObj) as Array<keyof typeof DatabaseSpanAttributesObj>
@@ -28,14 +29,17 @@ export const Vendors = {
   CHROMADB: 'chromadb',
   QDRANT: 'qdrant',
   WEAVIATE: 'weaviate',
-  PG: 'pg'
+  PG: 'pg',
+  OLLAMA: 'ollama'
 } as const
 
 export enum Event {
   STREAM_START = 'stream.start',
   STREAM_OUTPUT = 'stream.output',
   STREAM_END = 'stream.end',
   RESPONSE = 'response',
+  GEN_AI_COMPLETION = 'gen_ai.content.completion',
+  GEN_AI_PROMPT = 'gen_ai.content.prompt',
 }
 export type Vendor = typeof Vendors[keyof typeof Vendors]
 
@@ -49,7 +53,8 @@ interface VendorInstrumentationFunctions {
   chromadb: ChromadbFunctions[]
   qdrant: QdrantFunctions[]
   weaviate: WeaviateFunctions[]
-  pg: PgFunctions[]
+  pg: PgFunctions[],
+  ollama: OllamaFunctions[]
 }
 
 
@@ -68,7 +73,8 @@ export const TracedFunctionsByVendor: VendorTracedFunctions = {
   openai: OpenAIFunctionNames,
   pinecone: PineConeFunctionNames,
   qdrant: QdrantFunctionNames,
-  weaviate: WeaviateFunctionNames
+  weaviate: WeaviateFunctionNames,
+  ollama: OllamaFunctionNames
 } as const
 
 export const TIKTOKEN_MODEL_MAPPING: Record<TiktokenModel | string, TiktokenEncoding> = {

src/typescript/constants/ollama.ts

@@ -0,0 +1,32 @@
+/*
+ * Copyright (c) 2024 Scale3 Labs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+export const APIS = {
+  GENERATE: {
+    METHOD: 'ollama.generate',
+    ENDPOINT: '/api/generate'
+  },
+  CHAT: {
+    METHOD: 'ollama.chat',
+    ENDPOINT: '/api/chat'
+  },
+  EMBEDDINGS: {
+    METHOD: 'ollama.embeddings',
+    ENDPOINT: ' /api/embeddings'
+  }
+} as const
+export type OllamaFunctions = typeof APIS[keyof typeof APIS]['METHOD']
+export const OllamaFunctionNames: OllamaFunctions[] = Object.values(APIS).map((api) => api.METHOD)

src/typescript/index.ts

@@ -11,6 +11,7 @@ import { APIS as LlamaIndexAPIS } from "./constants/llamaindex";
 import { APIS as OpenAIAPIs } from "./constants/openai";
 import { APIS as PineConeAPIS } from "./constants/pinecone";
 import { APIS as QdrantAPIS } from "./constants/qdrant";
+import { APIS as ollamaAPIS } from "./constants/ollama";
 import { queryTypeToFunctionToProps } from "./constants/weaviate";
 import { TIKTOKEN_MODEL_MAPPING } from "./constants/common";
 
@@ -45,6 +46,7 @@ const APIS = {
   openai: OpenAIAPIs,
   pinecone: PineConeAPIS,
   qdrant: QdrantAPIS,
+  ollama: ollamaAPIS
 }
 export {
   LLMSpanAttributeNames,

src/typescript/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@langtrase/trace-attributes",
-    "version": "6.0.1",
+    "version": "6.0.2",
     "description": "LangTrace - Trace Attributes",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",