Skip to content

Commit 28ff6d8

Browse files
iamemilioashwinb
andauthored
fix: remove telemetry_traceable (#4205)
# What does this PR do? Removes stale data from llama stack about old telemetry system **Depends on** #4127 Co-authored-by: Ashwin Bharambe <[email protected]>
1 parent 7da7330 commit 28ff6d8

File tree

10 files changed

+0
-41
lines changed

10 files changed

+0
-41
lines changed

src/llama_stack_api/common/tracing.py

Lines changed: 0 additions & 22 deletions
This file was deleted.

src/llama_stack_api/conversations.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99

1010
from pydantic import BaseModel, Field
1111

12-
from llama_stack_api.common.tracing import telemetry_traceable
1312
from llama_stack_api.openai_responses import (
1413
OpenAIResponseInputFunctionToolCallOutput,
1514
OpenAIResponseMCPApprovalRequest,
@@ -157,7 +156,6 @@ class ConversationItemDeletedResource(BaseModel):
157156

158157

159158
@runtime_checkable
160-
@telemetry_traceable
161159
class Conversations(Protocol):
162160
"""Conversations
163161

src/llama_stack_api/files.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
from pydantic import BaseModel, Field
1212

1313
from llama_stack_api.common.responses import Order
14-
from llama_stack_api.common.tracing import telemetry_traceable
1514
from llama_stack_api.schema_utils import json_schema_type, webmethod
1615
from llama_stack_api.version import LLAMA_STACK_API_V1
1716

@@ -102,7 +101,6 @@ class OpenAIFileDeleteResponse(BaseModel):
102101

103102

104103
@runtime_checkable
105-
@telemetry_traceable
106104
class Files(Protocol):
107105
"""Files
108106

src/llama_stack_api/inference.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
from llama_stack_api.common.responses import (
2323
Order,
2424
)
25-
from llama_stack_api.common.tracing import telemetry_traceable
2625
from llama_stack_api.models import Model
2726
from llama_stack_api.schema_utils import json_schema_type, register_schema, webmethod
2827
from llama_stack_api.version import LLAMA_STACK_API_V1, LLAMA_STACK_API_V1ALPHA
@@ -989,7 +988,6 @@ class OpenAIEmbeddingsRequestWithExtraBody(BaseModel, extra="allow"):
989988

990989

991990
@runtime_checkable
992-
@telemetry_traceable
993991
class InferenceProvider(Protocol):
994992
"""
995993
This protocol defines the interface that should be implemented by all inference providers.

src/llama_stack_api/models.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99

1010
from pydantic import BaseModel, ConfigDict, Field, field_validator
1111

12-
from llama_stack_api.common.tracing import telemetry_traceable
1312
from llama_stack_api.resource import Resource, ResourceType
1413
from llama_stack_api.schema_utils import json_schema_type, webmethod
1514
from llama_stack_api.version import LLAMA_STACK_API_V1
@@ -106,7 +105,6 @@ class OpenAIListModelsResponse(BaseModel):
106105

107106

108107
@runtime_checkable
109-
@telemetry_traceable
110108
class Models(Protocol):
111109
async def list_models(self) -> ListModelsResponse:
112110
"""List all models.

src/llama_stack_api/prompts.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010

1111
from pydantic import BaseModel, Field, field_validator, model_validator
1212

13-
from llama_stack_api.common.tracing import telemetry_traceable
1413
from llama_stack_api.schema_utils import json_schema_type, webmethod
1514
from llama_stack_api.version import LLAMA_STACK_API_V1
1615

@@ -93,7 +92,6 @@ class ListPromptsResponse(BaseModel):
9392

9493

9594
@runtime_checkable
96-
@telemetry_traceable
9795
class Prompts(Protocol):
9896
"""Prompts
9997

src/llama_stack_api/safety.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99

1010
from pydantic import BaseModel, Field
1111

12-
from llama_stack_api.common.tracing import telemetry_traceable
1312
from llama_stack_api.inference import OpenAIMessageParam
1413
from llama_stack_api.schema_utils import json_schema_type, webmethod
1514
from llama_stack_api.shields import Shield
@@ -94,7 +93,6 @@ async def get_shield(self, identifier: str) -> Shield: ...
9493

9594

9695
@runtime_checkable
97-
@telemetry_traceable
9896
class Safety(Protocol):
9997
"""Safety
10098

src/llama_stack_api/shields.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88

99
from pydantic import BaseModel
1010

11-
from llama_stack_api.common.tracing import telemetry_traceable
1211
from llama_stack_api.resource import Resource, ResourceType
1312
from llama_stack_api.schema_utils import json_schema_type, webmethod
1413
from llama_stack_api.version import LLAMA_STACK_API_V1
@@ -49,7 +48,6 @@ class ListShieldsResponse(BaseModel):
4948

5049

5150
@runtime_checkable
52-
@telemetry_traceable
5351
class Shields(Protocol):
5452
@webmethod(route="/shields", method="GET", level=LLAMA_STACK_API_V1)
5553
async def list_shields(self) -> ListShieldsResponse:

src/llama_stack_api/tools.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
from typing_extensions import runtime_checkable
1212

1313
from llama_stack_api.common.content_types import URL, InterleavedContent
14-
from llama_stack_api.common.tracing import telemetry_traceable
1514
from llama_stack_api.resource import Resource, ResourceType
1615
from llama_stack_api.schema_utils import json_schema_type, webmethod
1716
from llama_stack_api.version import LLAMA_STACK_API_V1
@@ -109,7 +108,6 @@ class ListToolDefsResponse(BaseModel):
109108

110109

111110
@runtime_checkable
112-
@telemetry_traceable
113111
class ToolGroups(Protocol):
114112
@webmethod(route="/toolgroups", method="POST", level=LLAMA_STACK_API_V1, deprecated=True)
115113
async def register_tool_group(
@@ -191,7 +189,6 @@ class SpecialToolGroup(Enum):
191189

192190

193191
@runtime_checkable
194-
@telemetry_traceable
195192
class ToolRuntime(Protocol):
196193
tool_store: ToolStore | None = None
197194

src/llama_stack_api/vector_io.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from fastapi import Body, Query
1414
from pydantic import BaseModel, Field, field_validator
1515

16-
from llama_stack_api.common.tracing import telemetry_traceable
1716
from llama_stack_api.inference import InterleavedContent
1817
from llama_stack_api.schema_utils import json_schema_type, register_schema, webmethod
1918
from llama_stack_api.vector_stores import VectorStore
@@ -572,7 +571,6 @@ def get_vector_store(self, vector_store_id: str) -> VectorStore | None: ...
572571

573572

574573
@runtime_checkable
575-
@telemetry_traceable
576574
class VectorIO(Protocol):
577575
vector_store_table: VectorStoreTable | None = None
578576

0 commit comments

Comments
 (0)