move llama stack client to llms/providers

AgentOps-AI · Dec 30, 2024 · 7c6d827 · 7c6d827
1 parent 82f9bb9
commit 7c6d827
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/agentops/llms/llama_stack_client.py → ...tops/llms/providers/llama_stack_client.py b/agentops/llms/llama_stack_client.py → ...tops/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
@@ -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 = {}