[Bug]: Getting error while trying to use gemini model for evaluation.

[sverma29-nexturn]

What component(s) are affected?

• Python SDK
• Opik UI
• Opik Server
• Documentation

Opik version

• Opik version: 1.2.6

Describe the problem

Getting the following error:

AuthenticationError: litellm.AuthenticationError: geminiException - {
  "error": {
    "code": 400,
    "message": "API key not valid. Please pass a valid API key.",
    "status": "INVALID_ARGUMENT",
    "details": [
      {
        "@type": "type.googleapis.com/google.rpc.ErrorInfo",
        "reason": "API_KEY_INVALID",
        "domain": "googleapis.com",
        "metadata": {
          "service": "generativelanguage.googleapis.com"
        }
      },
      {
        "@type": "type.googleapis.com/google.rpc.LocalizedMessage",
        "locale": "en-US",
        "message": "API key not valid. Please pass a valid API key."
      }
    ]
  }
}

Reproduction steps

import google.generativeai as genai
import litellm
import os
from litellm.integrations.opik.opik import OpikLogger
from opik import track
from opik.opik_context import get_current_span_data

GEMINI_API_KEY = "<API KEY>"
genai.configure(api_key=GEMINI_API_KEY)

model = genai.GenerativeModel("gemini-1.5-flash")

os.environ["OPIK_PROJECT_NAME"] = "gemini-integration-demo"
opik_logger = OpikLogger()
litellm.callbacks = [opik_logger]

prompt = """
Write a short two sentence story about Opik.
"""

response = litellm.completion(
    model="gemini/gemini-pro-1.5",
    messages=[{"role": "user", "content": prompt}],
)

print(response.choices[0].message.content)

[jverre]

Hi @sverma29-nexturn,

Can you try remove the opik_logger and see if you still have the issue ? From the error message it seems to be an issue with the GEMINI_API_KEY rather than Opik