opik-gemini 0.1.1

package/README.md

@@ -0,0 +1,293 @@
+# Opik Gemini Integration
+
+[![npm version](https://img.shields.io/npm/v/opik-gemini.svg)](https://www.npmjs.com/package/opik-gemini)
+[![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/comet-ml/opik/blob/main/LICENSE)
+
+Seamlessly integrate [Opik](https://www.comet.com/docs/opik/) observability with your [Google Gemini](https://ai.google.dev/) applications.
+
+## Features
+
+- 🔍 **Comprehensive Tracing**: Automatically trace Gemini API calls
+- 📊 **Hierarchical Visualization**: View execution as structured traces and spans
+- 📝 **Detailed Metadata**: Record model names, prompts, completions, token usage
+- 🚨 **Error Handling**: Capture and visualize errors with full context
+- 🏷️ **Custom Tagging**: Add custom tags and metadata to organize traces
+- 🔄 **Streaming Support**: Full support for streamed responses
+- ⚡ **Non-blocking**: Minimal performance impact with async batching
+- 🎯 **Type-Safe**: Full TypeScript support with comprehensive types
+
+## Installation
+
+```bash
+npm install opik-gemini @google/genai
+```
+
+### Requirements
+
+- Node.js ≥ 18
+- @google/genai SDK (≥ 1.0.0)
+- Opik SDK (automatically installed as peer dependency)
+
+**Note**: The official Google GenAI SDK package is `@google/genai` (not `@google/generative-ai`). This is Google Deepmind's unified SDK for both Gemini Developer API and Vertex AI.
+
+## Quick Start
+
+### Basic Usage
+
+```typescript
+import { GoogleGenAI } from "@google/genai";
+import { trackGemini } from "opik-gemini";
+
+// Initialize Gemini client
+const genAI = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });
+
+// Wrap with Opik tracking
+const trackedGenAI = trackGemini(genAI, {
+  traceMetadata: {
+    tags: ["production", "my-app"],
+  },
+});
+
+// Use normally - all calls are automatically tracked
+async function main() {
+  const response = await trackedGenAI.models.generateContent({
+    model: "gemini-2.0-flash-001",
+    contents: "What is the capital of France?",
+  });
+
+  console.log(response.text);
+
+  // Ensure all traces are sent before exit
+  await trackedGenAI.flush();
+}
+
+main();
+```
+
+### Streaming Support
+
+```typescript
+import { GoogleGenAI } from "@google/genai";
+import { trackGemini } from "opik-gemini";
+
+const genAI = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });
+const trackedGenAI = trackGemini(genAI);
+
+async function streamExample() {
+  const response = await trackedGenAI.models.generateContentStream({
+    model: "gemini-2.0-flash-001",
+    contents: "Write a haiku about AI",
+  });
+
+  // Stream is automatically tracked
+  let streamedContent = "";
+  for await (const chunk of response) {
+    const chunkText = chunk.text;
+    if (chunkText) {
+      process.stdout.write(chunkText);
+      streamedContent += chunkText;
+    }
+  }
+
+  console.log("\n");
+  await trackedGenAI.flush();
+}
+
+streamExample();
+```
+
+### Using with Existing Opik Client
+
+```typescript
+import { Opik } from "opik";
+import { GoogleGenAI } from "@google/genai";
+import { trackGemini } from "opik-gemini";
+
+const opikClient = new Opik({
+  projectName: "gemini-project",
+});
+
+const genAI = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });
+const trackedGenAI = trackGemini(genAI, {
+  client: opikClient,
+  traceMetadata: {
+    tags: ["gemini", "production"],
+    environment: "prod",
+  },
+});
+
+// All calls will be logged to "gemini-project"
+const response = await trackedGenAI.models.generateContent({
+  model: "gemini-2.0-flash-001",
+  contents: "Hello, Gemini!",
+});
+
+console.log(response.text);
+```
+
+### Custom Generation Names
+
+```typescript
+import { trackGemini } from "opik-gemini";
+
+const trackedGenAI = trackGemini(genAI, {
+  generationName: "MyCustomGeminiCall",
+});
+
+// Traces will appear as "MyCustomGeminiCall" in Opik
+```
+
+### Nested Tracing
+
+```typescript
+import { Opik } from "opik";
+import { trackGemini } from "opik-gemini";
+
+const opikClient = new Opik();
+const trackedGenAI = trackGemini(genAI, { client: opikClient });
+
+async function processQuery(query: string) {
+  // Create parent trace
+  const trace = opikClient.trace({
+    name: "ProcessUserQuery",
+    input: { query },
+  });
+
+  // Gemini call will be nested under this trace
+  const trackedGenAIWithParent = trackGemini(genAI, {
+    parent: trace,
+    client: opikClient,
+  });
+
+  const response = await trackedGenAIWithParent.models.generateContent({
+    model: "gemini-2.0-flash-001",
+    contents: query,
+  });
+
+  trace.update({
+    output: { response: response.text },
+  });
+  trace.end();
+
+  return response;
+}
+```
+
+## Configuration
+
+### TrackOpikConfig Options
+
+```typescript
+interface TrackOpikConfig {
+  /** Opik client instance (optional, creates singleton if not provided) */
+  client?: Opik;
+
+  /** Custom name for the generation (optional, defaults to method name) */
+  generationName?: string;
+
+  /** Parent trace or span for nested tracing (optional) */
+  parent?: Trace | Span;
+
+  /** Additional metadata for traces (optional) */
+  traceMetadata?: {
+    tags?: string[];
+    [key: string]: unknown;
+  };
+}
+```
+
+## What Gets Tracked
+
+The integration automatically captures:
+
+- **Input**: Prompt contents and generation config
+- **Output**: Generated text, candidates, and safety ratings
+- **Model**: Model name/version (e.g., "gemini-pro", "gemini-1.5-flash")
+- **Usage**: Token counts (prompt, completion, total)
+- **Metadata**: Provider info, model settings, safety settings
+- **Errors**: Error messages and stack traces
+- **Timing**: Start/end times and duration
+
+## Best Practices
+
+1. **Always call `flush()` before process exit** (especially in short-lived scripts):
+
+   ```typescript
+   await trackedGenAI.flush();
+   ```
+
+2. **Use descriptive tags** for easier filtering:
+
+   ```typescript
+   trackGemini(genAI, {
+     traceMetadata: {
+       tags: ["production", "customer-support", "v2"],
+     },
+   });
+   ```
+
+3. **Reuse Opik client** across your application for consistency:
+
+   ```typescript
+   const opikClient = new Opik({ projectName: "my-project" });
+   const trackedGenAI = trackGemini(genAI, { client: opikClient });
+   ```
+
+4. **Use nested tracing** for complex workflows to understand call hierarchies.
+
+## Supported Gemini Models
+
+This integration supports all Google Gemini models including:
+
+- `gemini-2.0-flash-001` (Latest, recommended for most use cases)
+- `gemini-1.5-pro`
+- `gemini-1.5-flash`
+- `gemini-pro`
+- `gemini-pro-vision`
+- Any future Gemini models
+
+Refer to [Google's official documentation](https://ai.google.dev/models/gemini) for the complete list of available models and their capabilities.
+
+## Development
+
+### Building
+
+```bash
+npm run build
+```
+
+### Type Checking
+
+```bash
+npm run typecheck
+```
+
+### Testing
+
+```bash
+npm test
+```
+
+## Examples
+
+Check out the [examples directory](../../../../../../examples) for more usage examples.
+
+## Documentation
+
+- [Opik Documentation](https://www.comet.com/docs/opik/)
+- [Google Gemini Documentation](https://ai.google.dev/)
+- [TypeScript SDK Guide](https://www.comet.com/docs/opik/typescript-sdk)
+
+## Support
+
+- [GitHub Issues](https://github.com/comet-ml/opik/issues)
+- [Comet Support](mailto:support@comet.com)
+- [Community Slack](https://www.comet.com/docs/opik/community/)
+
+## License
+
+Apache-2.0
+
+## Contributing
+
+Contributions are welcome! Please see our [Contributing Guide](https://github.com/comet-ml/opik/blob/main/CONTRIBUTING.md).

package/dist/index.cjs

@@ -0,0 +1,2 @@
+'use strict';var opik=require('opik');var R=class R{static getInstance(e){return R.instance||(R.instance=new opik.Opik(e)),R.instance}};R.instance=null;var A=R;var x=a=>{if(a)return a.replace(/^models\//,"")};var G=a=>a!==null&&typeof a=="object"&&Symbol.asyncIterator in a&&typeof a[Symbol.asyncIterator]=="function",P=(a,e="")=>{let o={};for(let[n,t]of Object.entries(a)){let i=e?`${e}.${n}`:n;t&&typeof t=="object"&&!Array.isArray(t)&&!(t instanceof Date)?Object.assign(o,P(t,i)):o[i]=t;}return o};var E=a=>{let e=a.model;e&&(e=x(e));let o={};"contents"in a&&(o.contents=a.contents),"prompt"in a&&(o.prompt=a.prompt),"config"in a&&(o.config=a.config);let n={};if("config"in a&&typeof a.config=="object"&&a.config!==null){let t=a.config;"systemInstruction"in t&&(n.systemInstruction=t.systemInstruction),"generationConfig"in t&&(n.generationConfig=t.generationConfig),"safetySettings"in t&&(n.safetySettings=t.safetySettings),"tools"in t&&(n.tools=t.tools),"toolConfig"in t&&(n.toolConfig=t.toolConfig);let i=["temperature","maxOutputTokens","topP","topK","candidateCount","stopSequences"];for(let c of i)c in t&&(n[c]=t[c]);}return {model:e,input:o,modelParameters:n}},O=a=>{if(!a||typeof a!="object")return;let e=a;if("candidates"in e&&Array.isArray(e.candidates))return e.candidates.length===0?void 0:{candidates:e.candidates}},D=a=>{if(!a||typeof a!="object")return;let e=a,o="usageMetadata"in e&&e.usageMetadata||"usage_metadata"in e&&e.usage_metadata||"usage"in e&&e.usage;if(!o||typeof o!="object")return;let n=o,t={};(typeof n.promptTokenCount=="number"||typeof n.prompt_token_count=="number")&&(t.prompt_tokens=n.promptTokenCount||n.prompt_token_count),(typeof n.candidatesTokenCount=="number"||typeof n.candidates_token_count=="number")&&(t.completion_tokens=n.candidatesTokenCount||n.candidates_token_count),(typeof n.totalTokenCount=="number"||typeof n.total_token_count=="number")&&(t.total_tokens=n.totalTokenCount||n.total_token_count),(typeof n.cachedContentTokenCount=="number"||typeof n.cached_content_token_count=="number")&&(t.cached_content_tokens=n.cachedContentTokenCount||n.cached_content_token_count);let i={...n};delete i.promptTokensDetails,delete i.prompt_tokens_details,delete i.candidatesTokensDetails,delete i.candidates_tokens_details;let c=P(i,"original_usage");for(let[l,s]of Object.entries(c))typeof s=="number"&&(t[l]=s);return Object.keys(t).length>0?t:void 0},M=a=>{if(!a||typeof a!="object")return {isToolCall:false,data:""};let e=a;if("candidates"in e&&Array.isArray(e.candidates)&&e.candidates.length>0){let o=e.candidates[0];if(o&&"content"in o&&o.content&&typeof o.content=="object"){let n=o.content;if("parts"in n&&Array.isArray(n.parts)&&n.parts.length>0){let t=n.parts[0];if("functionCall"in t||"function_call"in t)return {isToolCall:true,data:JSON.stringify(t.functionCall||t.function_call)};if("text"in t&&typeof t.text=="string")return {isToolCall:false,data:t.text}}}}return {isToolCall:false,data:""}},C=a=>{if(!a||typeof a!="object")return {model:void 0,metadata:void 0};let e=a,o,n={};if("modelVersion"in e&&typeof e.modelVersion=="string"?o=x(e.modelVersion):"model_version"in e&&typeof e.model_version=="string"&&(o=x(e.model_version)),"candidates"in e&&Array.isArray(e.candidates)&&e.candidates.length>0){let t=e.candidates[0];"safetyRatings"in t?n.safety_ratings=t.safetyRatings:"safety_ratings"in t&&(n.safety_ratings=t.safety_ratings),"finishReason"in t?n.finish_reason=t.finishReason:"finish_reason"in t&&(n.finish_reason=t.finish_reason);}return "promptFeedback"in e?n.prompt_feedback=e.promptFeedback:"prompt_feedback"in e&&(n.prompt_feedback=e.prompt_feedback),n.created_from="genai",{model:o,metadata:Object.keys(n).length>0?n:void 0}};var v=(a,e)=>(...o)=>{var g;let{model:n,input:t,modelParameters:i}=E(o[0]),{tags:c=[],...l}=(g=e==null?void 0:e.traceMetadata)!=null?g:{},s={name:e.generationName,startTime:new Date,input:t,model:n,provider:e.provider,metadata:{...l,...i},tags:["genai",...c]},r,y=!!(e!=null&&e.parent);e!=null&&e.parent?r=e.parent:r=e.client.trace(s);try{let d=a(...o);if(G(d))return N(d,r,y,s);if(d instanceof Promise)return d.then(p=>{if(G(p))return N(p,r,y,s);let F=O(p),S=D(p),{model:$,metadata:K}=C(p),L=r.span({...s,output:F,endTime:new Date,usage:S,model:$||s.model,type:opik.OpikSpanType.Llm,metadata:{...s.metadata,...K,usage:S}}),j=p;return j.functionCalls&&Array.isArray(j.functionCalls)&&j.functionCalls.forEach(h=>{L.span({name:`function_call: ${h.name}`,startTime:new Date,endTime:new Date,input:{arguments:h.args},output:{function_name:h.name},type:opik.OpikSpanType.Tool,metadata:{function_name:h.name,function_arguments:h.args}});}),y||r.update({output:F,endTime:new Date}),p}).catch(p=>{throw I(p,r,s),p});let k=O(d),m=D(d),{model:b,metadata:f}=C(d),u=r.span({...s,output:k,endTime:new Date,usage:m,model:b||s.model,type:opik.OpikSpanType.Llm,metadata:{...s.metadata,...f,usage:m}}),_=d;return _.functionCalls&&Array.isArray(_.functionCalls)&&_.functionCalls.forEach(w=>{u.span({name:`function_call: ${w.name}`,startTime:new Date,endTime:new Date,input:{arguments:w.args},output:{function_name:w.name},type:opik.OpikSpanType.Tool,metadata:{function_name:w.name,function_arguments:w.args}});}),y||r.update({output:k,endTime:new Date}),d}catch(d){throw I(d,r,s),d}};function N(a,e,o,n){async function*t(){var y,g,d,k;let i=[],c=[],l,s={},r;try{for await(let f of a){i.push(f);let u=M(f);u.isToolCall||c.push(u.data),yield f;}let m=i[i.length-1];if(m){l=D(m),r=C(m);let f=c.join(""),u=O(m);if(u!=null&&u.candidates){s={candidates:structuredClone(u.candidates)};let _=s;(k=(d=(g=(y=_.candidates)==null?void 0:y[0])==null?void 0:g.content)==null?void 0:d.parts)!=null&&k[0]&&(_.candidates[0].content.parts[0].text=f);}}let b=e.span({...n,output:s,endTime:new Date,usage:l,model:(r==null?void 0:r.model)||n.model,type:opik.OpikSpanType.Llm,metadata:{...n.metadata,...r==null?void 0:r.metadata,usage:l}});if(m&&typeof m=="object"){let f=m;f.functionCalls&&Array.isArray(f.functionCalls)&&f.functionCalls.forEach(u=>{b.span({name:`function_call: ${u.name}`,startTime:new Date,endTime:new Date,input:{arguments:u.args},output:{function_name:u.name},type:opik.OpikSpanType.Tool,metadata:{function_name:u.name,function_arguments:u.args}});});}o||e.update({output:s,endTime:new Date});}catch(m){throw I(m,e,n),m}}return t()}var I=(a,e,o)=>{var n;e.span({...o,endTime:new Date,type:opik.OpikSpanType.Llm,errorInfo:{message:a.message,exceptionType:a.name,traceback:(n=a.stack)!=null?n:""}}),e.end();};var V=a=>"vertexai"in a&&a.vertexai?"google_vertexai":"google_ai",z=(a,e)=>{let o=V(a);return new Proxy(a,{get(n,t,i){var g,d,k;let c=n[t],l=((g=a.constructor)==null?void 0:g.name)||"Gemini",r={generationName:(d=e==null?void 0:e.generationName)!=null?d:`${l}.${t.toString()}`,client:(k=e==null?void 0:e.client)!=null?k:A.getInstance(),parent:e==null?void 0:e.parent,traceMetadata:e==null?void 0:e.traceMetadata,provider:o};return t==="flush"?r.client.flush.bind(r.client):typeof c=="function"?v(c.bind(n),r):c&&!Array.isArray(c)&&!(c instanceof Date)&&typeof c=="object"?new Proxy(c,{get(m,b,f){let u=m[b];return typeof u=="function"?v(u.bind(m),r):Reflect.get(m,b,f)}}):Reflect.get(n,t,i)}})};
+exports.trackGemini=z;

package/dist/index.d.cts

@@ -0,0 +1,48 @@
+import { Trace, Span, Opik } from 'opik';
+
+type OpikParent = Trace | Span;
+type TrackOpikConfig = {
+    /**
+     * Project name for this trace
+     */
+    projectName?: string;
+    /**
+     * Parent span or trace for this generation
+     */
+    parent?: OpikParent;
+    /**
+     * Generation name (defaults to "ClassName.methodName")
+     */
+    generationName?: string;
+    /**
+     * Trace metadata (tags, custom fields)
+     */
+    traceMetadata?: Record<string, unknown> & {
+        tags?: string[];
+    };
+    /**
+     * Opik client instance
+     */
+    client?: Opik;
+};
+type OpikExtension = {
+    /**
+     * Flush all pending traces and spans to Opik backend
+     */
+    flush: () => Promise<void>;
+};
+
+/**
+ * Track Gemini client with Opik observability
+ *
+ * Wraps a Gemini SDK instance (GoogleGenerativeAI or any Gemini client) to automatically
+ * create traces and spans for all generation calls.
+ *
+ * @param sdk - The Gemini SDK instance to track
+ * @param opikConfig - Configuration for Opik tracking
+ * @returns Proxied SDK instance with flush() method
+ *
+ */
+declare const trackGemini: <SDKType extends object>(sdk: SDKType, opikConfig?: TrackOpikConfig) => SDKType & OpikExtension;
+
+export { trackGemini };

package/dist/index.d.ts

@@ -0,0 +1,48 @@
+import { Trace, Span, Opik } from 'opik';
+
+type OpikParent = Trace | Span;
+type TrackOpikConfig = {
+    /**
+     * Project name for this trace
+     */
+    projectName?: string;
+    /**
+     * Parent span or trace for this generation
+     */
+    parent?: OpikParent;
+    /**
+     * Generation name (defaults to "ClassName.methodName")
+     */
+    generationName?: string;
+    /**
+     * Trace metadata (tags, custom fields)
+     */
+    traceMetadata?: Record<string, unknown> & {
+        tags?: string[];
+    };
+    /**
+     * Opik client instance
+     */
+    client?: Opik;
+};
+type OpikExtension = {
+    /**
+     * Flush all pending traces and spans to Opik backend
+     */
+    flush: () => Promise<void>;
+};
+
+/**
+ * Track Gemini client with Opik observability
+ *
+ * Wraps a Gemini SDK instance (GoogleGenerativeAI or any Gemini client) to automatically
+ * create traces and spans for all generation calls.
+ *
+ * @param sdk - The Gemini SDK instance to track
+ * @param opikConfig - Configuration for Opik tracking
+ * @returns Proxied SDK instance with flush() method
+ *
+ */
+declare const trackGemini: <SDKType extends object>(sdk: SDKType, opikConfig?: TrackOpikConfig) => SDKType & OpikExtension;
+
+export { trackGemini };

package/dist/index.js

@@ -0,0 +1,2 @@
+import {Opik,OpikSpanType}from'opik';var R=class R{static getInstance(e){return R.instance||(R.instance=new Opik(e)),R.instance}};R.instance=null;var A=R;var x=a=>{if(a)return a.replace(/^models\//,"")};var G=a=>a!==null&&typeof a=="object"&&Symbol.asyncIterator in a&&typeof a[Symbol.asyncIterator]=="function",P=(a,e="")=>{let o={};for(let[n,t]of Object.entries(a)){let i=e?`${e}.${n}`:n;t&&typeof t=="object"&&!Array.isArray(t)&&!(t instanceof Date)?Object.assign(o,P(t,i)):o[i]=t;}return o};var E=a=>{let e=a.model;e&&(e=x(e));let o={};"contents"in a&&(o.contents=a.contents),"prompt"in a&&(o.prompt=a.prompt),"config"in a&&(o.config=a.config);let n={};if("config"in a&&typeof a.config=="object"&&a.config!==null){let t=a.config;"systemInstruction"in t&&(n.systemInstruction=t.systemInstruction),"generationConfig"in t&&(n.generationConfig=t.generationConfig),"safetySettings"in t&&(n.safetySettings=t.safetySettings),"tools"in t&&(n.tools=t.tools),"toolConfig"in t&&(n.toolConfig=t.toolConfig);let i=["temperature","maxOutputTokens","topP","topK","candidateCount","stopSequences"];for(let c of i)c in t&&(n[c]=t[c]);}return {model:e,input:o,modelParameters:n}},O=a=>{if(!a||typeof a!="object")return;let e=a;if("candidates"in e&&Array.isArray(e.candidates))return e.candidates.length===0?void 0:{candidates:e.candidates}},D=a=>{if(!a||typeof a!="object")return;let e=a,o="usageMetadata"in e&&e.usageMetadata||"usage_metadata"in e&&e.usage_metadata||"usage"in e&&e.usage;if(!o||typeof o!="object")return;let n=o,t={};(typeof n.promptTokenCount=="number"||typeof n.prompt_token_count=="number")&&(t.prompt_tokens=n.promptTokenCount||n.prompt_token_count),(typeof n.candidatesTokenCount=="number"||typeof n.candidates_token_count=="number")&&(t.completion_tokens=n.candidatesTokenCount||n.candidates_token_count),(typeof n.totalTokenCount=="number"||typeof n.total_token_count=="number")&&(t.total_tokens=n.totalTokenCount||n.total_token_count),(typeof n.cachedContentTokenCount=="number"||typeof n.cached_content_token_count=="number")&&(t.cached_content_tokens=n.cachedContentTokenCount||n.cached_content_token_count);let i={...n};delete i.promptTokensDetails,delete i.prompt_tokens_details,delete i.candidatesTokensDetails,delete i.candidates_tokens_details;let c=P(i,"original_usage");for(let[l,s]of Object.entries(c))typeof s=="number"&&(t[l]=s);return Object.keys(t).length>0?t:void 0},M=a=>{if(!a||typeof a!="object")return {isToolCall:false,data:""};let e=a;if("candidates"in e&&Array.isArray(e.candidates)&&e.candidates.length>0){let o=e.candidates[0];if(o&&"content"in o&&o.content&&typeof o.content=="object"){let n=o.content;if("parts"in n&&Array.isArray(n.parts)&&n.parts.length>0){let t=n.parts[0];if("functionCall"in t||"function_call"in t)return {isToolCall:true,data:JSON.stringify(t.functionCall||t.function_call)};if("text"in t&&typeof t.text=="string")return {isToolCall:false,data:t.text}}}}return {isToolCall:false,data:""}},C=a=>{if(!a||typeof a!="object")return {model:void 0,metadata:void 0};let e=a,o,n={};if("modelVersion"in e&&typeof e.modelVersion=="string"?o=x(e.modelVersion):"model_version"in e&&typeof e.model_version=="string"&&(o=x(e.model_version)),"candidates"in e&&Array.isArray(e.candidates)&&e.candidates.length>0){let t=e.candidates[0];"safetyRatings"in t?n.safety_ratings=t.safetyRatings:"safety_ratings"in t&&(n.safety_ratings=t.safety_ratings),"finishReason"in t?n.finish_reason=t.finishReason:"finish_reason"in t&&(n.finish_reason=t.finish_reason);}return "promptFeedback"in e?n.prompt_feedback=e.promptFeedback:"prompt_feedback"in e&&(n.prompt_feedback=e.prompt_feedback),n.created_from="genai",{model:o,metadata:Object.keys(n).length>0?n:void 0}};var v=(a,e)=>(...o)=>{var g;let{model:n,input:t,modelParameters:i}=E(o[0]),{tags:c=[],...l}=(g=e==null?void 0:e.traceMetadata)!=null?g:{},s={name:e.generationName,startTime:new Date,input:t,model:n,provider:e.provider,metadata:{...l,...i},tags:["genai",...c]},r,y=!!(e!=null&&e.parent);e!=null&&e.parent?r=e.parent:r=e.client.trace(s);try{let d=a(...o);if(G(d))return N(d,r,y,s);if(d instanceof Promise)return d.then(p=>{if(G(p))return N(p,r,y,s);let F=O(p),S=D(p),{model:$,metadata:K}=C(p),L=r.span({...s,output:F,endTime:new Date,usage:S,model:$||s.model,type:OpikSpanType.Llm,metadata:{...s.metadata,...K,usage:S}}),j=p;return j.functionCalls&&Array.isArray(j.functionCalls)&&j.functionCalls.forEach(h=>{L.span({name:`function_call: ${h.name}`,startTime:new Date,endTime:new Date,input:{arguments:h.args},output:{function_name:h.name},type:OpikSpanType.Tool,metadata:{function_name:h.name,function_arguments:h.args}});}),y||r.update({output:F,endTime:new Date}),p}).catch(p=>{throw I(p,r,s),p});let k=O(d),m=D(d),{model:b,metadata:f}=C(d),u=r.span({...s,output:k,endTime:new Date,usage:m,model:b||s.model,type:OpikSpanType.Llm,metadata:{...s.metadata,...f,usage:m}}),_=d;return _.functionCalls&&Array.isArray(_.functionCalls)&&_.functionCalls.forEach(w=>{u.span({name:`function_call: ${w.name}`,startTime:new Date,endTime:new Date,input:{arguments:w.args},output:{function_name:w.name},type:OpikSpanType.Tool,metadata:{function_name:w.name,function_arguments:w.args}});}),y||r.update({output:k,endTime:new Date}),d}catch(d){throw I(d,r,s),d}};function N(a,e,o,n){async function*t(){var y,g,d,k;let i=[],c=[],l,s={},r;try{for await(let f of a){i.push(f);let u=M(f);u.isToolCall||c.push(u.data),yield f;}let m=i[i.length-1];if(m){l=D(m),r=C(m);let f=c.join(""),u=O(m);if(u!=null&&u.candidates){s={candidates:structuredClone(u.candidates)};let _=s;(k=(d=(g=(y=_.candidates)==null?void 0:y[0])==null?void 0:g.content)==null?void 0:d.parts)!=null&&k[0]&&(_.candidates[0].content.parts[0].text=f);}}let b=e.span({...n,output:s,endTime:new Date,usage:l,model:(r==null?void 0:r.model)||n.model,type:OpikSpanType.Llm,metadata:{...n.metadata,...r==null?void 0:r.metadata,usage:l}});if(m&&typeof m=="object"){let f=m;f.functionCalls&&Array.isArray(f.functionCalls)&&f.functionCalls.forEach(u=>{b.span({name:`function_call: ${u.name}`,startTime:new Date,endTime:new Date,input:{arguments:u.args},output:{function_name:u.name},type:OpikSpanType.Tool,metadata:{function_name:u.name,function_arguments:u.args}});});}o||e.update({output:s,endTime:new Date});}catch(m){throw I(m,e,n),m}}return t()}var I=(a,e,o)=>{var n;e.span({...o,endTime:new Date,type:OpikSpanType.Llm,errorInfo:{message:a.message,exceptionType:a.name,traceback:(n=a.stack)!=null?n:""}}),e.end();};var V=a=>"vertexai"in a&&a.vertexai?"google_vertexai":"google_ai",z=(a,e)=>{let o=V(a);return new Proxy(a,{get(n,t,i){var g,d,k;let c=n[t],l=((g=a.constructor)==null?void 0:g.name)||"Gemini",r={generationName:(d=e==null?void 0:e.generationName)!=null?d:`${l}.${t.toString()}`,client:(k=e==null?void 0:e.client)!=null?k:A.getInstance(),parent:e==null?void 0:e.parent,traceMetadata:e==null?void 0:e.traceMetadata,provider:o};return t==="flush"?r.client.flush.bind(r.client):typeof c=="function"?v(c.bind(n),r):c&&!Array.isArray(c)&&!(c instanceof Date)&&typeof c=="object"?new Proxy(c,{get(m,b,f){let u=m[b];return typeof u=="function"?v(u.bind(m),r):Reflect.get(m,b,f)}}):Reflect.get(n,t,i)}})};
+export{z as trackGemini};

package/package.json

@@ -0,0 +1,79 @@
+{
+  "name": "opik-gemini",
+  "description": "Opik TypeScript and JavaScript SDK integration with Google Gemini AI",
+  "version": "0.1.1",
+  "engines": {
+    "node": ">=18"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/comet-ml/opik.git",
+    "directory": "sdks/typescript/src/opik/integrations/opik-gemini"
+  },
+  "homepage": "https://www.comet.com/docs/opik/",
+  "author": {
+    "name": "Comet",
+    "email": "support@comet.com",
+    "url": "https://github.com/comet-ml"
+  },
+  "bugs": {
+    "url": "https://github.com/comet-ml/opik/issues",
+    "email": "support@comet.com"
+  },
+  "license": "Apache-2.0",
+  "keywords": [
+    "opik",
+    "gemini",
+    "google-ai",
+    "google-generative-ai",
+    "genai",
+    "gemini-integration",
+    "sdk",
+    "javascript",
+    "javascript-sdk",
+    "typescript",
+    "typescript-sdk",
+    "llm",
+    "tracing",
+    "observability",
+    "comet"
+  ],
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "main": "dist/index.cjs",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "type": "module",
+  "scripts": {
+    "build": "tsup",
+    "watch": "tsup --watch",
+    "lint": "eslint '**/*.{ts,tsx}'",
+    "typecheck": "tsc --noEmit",
+    "format": "prettier --write 'src/**/*.{ts,tsx,js,jsx,json,md}'",
+    "test": "vitest"
+  },
+  "files": [
+    "dist/**/*",
+    "README.md"
+  ],
+  "peerDependencies": {
+    "@google/genai": ">=1.0.0",
+    "opik": "^1.7.25"
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.20.0",
+    "eslint": "^9.34.0",
+    "globals": "^15.14.0",
+    "prettier": "^3.6.2",
+    "tsup": "^8.5.0",
+    "typescript": "^5.9.2",
+    "typescript-eslint": "^8.42.0",
+    "vitest": "^3.0.5"
+  }
+}