Skip to content

Commit 5703096

Browse files
committed
refactor: move blobpart and filedatapart to instrumentation-google-genai package.
1 parent 46b10d1 commit 5703096

File tree

2 files changed

+19
-19
lines changed
  • instrumentation-genai/opentelemetry-instrumentation-google-genai/src/opentelemetry/instrumentation/google_genai
  • util/opentelemetry-util-genai/src/opentelemetry/util/genai

2 files changed

+19
-19
lines changed

instrumentation-genai/opentelemetry-instrumentation-google-genai/src/opentelemetry/instrumentation/google_genai/message.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,12 @@
1616

1717
import logging
1818
from enum import Enum
19+
from dataclasses import dataclass
20+
from typing import Literal
1921

2022
from google.genai import types as genai_types
2123

2224
from opentelemetry.util.genai.types import (
23-
BlobPart,
24-
FileDataPart,
2525
FinishReason,
2626
InputMessage,
2727
MessagePart,
@@ -39,6 +39,23 @@ class Role(str, Enum):
3939
TOOL = "tool"
4040

4141

42+
@dataclass
43+
class BlobPart:
44+
data: bytes
45+
mime_type: str
46+
type: Literal["blob"] = "blob"
47+
48+
49+
@dataclass
50+
class FileDataPart:
51+
mime_type: str
52+
uri: str
53+
type: Literal["file_data"] = "file_data"
54+
55+
class Config:
56+
extra = "allow"
57+
58+
4259
_logger = logging.getLogger(__name__)
4360

4461

util/opentelemetry-util-genai/src/opentelemetry/util/genai/types.py

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -55,23 +55,6 @@ class Text:
5555
type: Literal["text"] = "text"
5656

5757

58-
@dataclass
59-
class BlobPart:
60-
data: bytes
61-
mime_type: str
62-
type: Literal["blob"] = "blob"
63-
64-
65-
@dataclass
66-
class FileDataPart:
67-
mime_type: str
68-
uri: str
69-
type: Literal["file_data"] = "file_data"
70-
71-
class Config:
72-
extra = "allow"
73-
74-
7558
MessagePart = Union[Text, ToolCall, ToolCallResponse, Any]
7659

7760

0 commit comments

Comments
 (0)