From 7c6d827ca4b50727692f4f73e63449b79f28b907 Mon Sep 17 00:00:00 2001
From: Pratyush Shukla <ps4534@nyu.edu>
Date: Mon, 30 Dec 2024 17:17:22 +0530
Subject: [PATCH] move llama stack client to `llms/providers`

---
 agentops/llms/{ => providers}/llama_stack_client.py | 4 +---
 agentops/llms/tracker.py                            | 3 +--
 2 files changed, 2 insertions(+), 5 deletions(-)
 rename agentops/llms/{ => providers}/llama_stack_client.py (99%)

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 = {}