diff --git a/agentops/llms/llama_stack_client.py b/agentops/llms/providers/llama_stack_client.py similarity index 99% rename from agentops/llms/llama_stack_client.py rename to agentops/llms/providers/llama_stack_client.py index 50f733b17..ecd008cbe 100644 --- a/agentops/llms/llama_stack_client.py +++ b/agentops/llms/providers/llama_stack_client.py @@ -1,9 +1,7 @@ import inspect import pprint -import sys -from typing import Any, AsyncGenerator, Dict, Optional, List +from typing import Any, AsyncGenerator, Dict, Optional, List, Union import logging -from typing import Union from agentops.event import LLMEvent, ErrorEvent, ToolEvent from agentops.session import Session diff --git a/agentops/llms/tracker.py b/agentops/llms/tracker.py index 928cfa26d..615e3dfac 100644 --- a/agentops/llms/tracker.py +++ b/agentops/llms/tracker.py @@ -4,8 +4,6 @@ from packaging.version import Version, parse -from agentops.llms.llama_stack_client import LlamaStackClientProvider - from ..log_config import logger from .providers.cohere import CohereProvider @@ -16,6 +14,7 @@ from .providers.anthropic import AnthropicProvider from .providers.mistral import MistralProvider from .providers.ai21 import AI21Provider +from .providers.llama_stack_client import LlamaStackClientProvider from .providers.taskweaver import TaskWeaverProvider original_func = {}