From 6a99b9a8ac602d7df88d99aef692d460a40f8aa1 Mon Sep 17 00:00:00 2001 From: Teo Date: Thu, 16 Jan 2025 19:44:30 +0100 Subject: [PATCH] Change InstrumentedProvider -> BaseProvider Signed-off-by: Teo --- CONTRIBUTING.md | 6 +++--- agentops/llms/providers/ai21.py | 4 ++-- agentops/llms/providers/anthropic.py | 4 ++-- .../llms/providers/{instrumented_provider.py => base.py} | 2 +- agentops/llms/providers/cohere.py | 4 ++-- agentops/llms/providers/groq.py | 4 ++-- agentops/llms/providers/litellm.py | 4 ++-- agentops/llms/providers/llama_stack_client.py | 4 ++-- agentops/llms/providers/mistral.py | 4 ++-- agentops/llms/providers/ollama.py | 4 ++-- agentops/llms/providers/openai.py | 4 ++-- agentops/llms/providers/taskweaver.py | 4 ++-- 12 files changed, 24 insertions(+), 24 deletions(-) rename agentops/llms/providers/{instrumented_provider.py => base.py} (96%) diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 0b66b8d86..68f456974 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -274,10 +274,10 @@ We use Jupyter notebooks as integration tests for LLM providers. This approach: The `agentops/llms/` directory contains provider implementations. Each provider must: -1. **Inherit from InstrumentedProvider**: +1. **Inherit from BaseProvider**: ```python @singleton - class NewProvider(InstrumentedProvider): + class NewProvider(BaseProvider): def __init__(self, client): super().__init__(client) self._provider_name = "ProviderName" @@ -397,4 +397,4 @@ We encourage active community participation and are here to help! ## License -By contributing to AgentOps, you agree that your contributions will be licensed under the MIT License. \ No newline at end of file +By contributing to AgentOps, you agree that your contributions will be licensed under the MIT License. diff --git a/agentops/llms/providers/ai21.py b/agentops/llms/providers/ai21.py index 8c907d525..f445a8035 100644 --- a/agentops/llms/providers/ai21.py +++ b/agentops/llms/providers/ai21.py @@ -2,7 +2,7 @@ import pprint from typing import Optional -from agentops.llms.providers.instrumented_provider import InstrumentedProvider +from agentops.llms.providers.base import BaseProvider from agentops.time_travel import fetch_completion_override_from_time_travel_cache from agentops.event import ErrorEvent, LLMEvent, ActionEvent, ToolEvent @@ -13,7 +13,7 @@ @singleton -class AI21Provider(InstrumentedProvider): +class AI21Provider(BaseProvider): original_create = None original_create_async = None original_answer = None diff --git a/agentops/llms/providers/anthropic.py b/agentops/llms/providers/anthropic.py index 7ba523d82..02d536fb4 100644 --- a/agentops/llms/providers/anthropic.py +++ b/agentops/llms/providers/anthropic.py @@ -2,7 +2,7 @@ import pprint from typing import Optional -from agentops.llms.providers.instrumented_provider import InstrumentedProvider +from agentops.llms.providers.base import BaseProvider from agentops.time_travel import fetch_completion_override_from_time_travel_cache from agentops.event import ErrorEvent, LLMEvent, ToolEvent @@ -13,7 +13,7 @@ @singleton -class AnthropicProvider(InstrumentedProvider): +class AnthropicProvider(BaseProvider): original_create = None original_create_async = None diff --git a/agentops/llms/providers/instrumented_provider.py b/agentops/llms/providers/base.py similarity index 96% rename from agentops/llms/providers/instrumented_provider.py rename to agentops/llms/providers/base.py index 9b069c535..7a54b5f0e 100644 --- a/agentops/llms/providers/instrumented_provider.py +++ b/agentops/llms/providers/base.py @@ -5,7 +5,7 @@ from agentops.event import LLMEvent -class InstrumentedProvider(ABC): +class BaseProvider(ABC): _provider_name: str = "InstrumentedModel" llm_event: Optional[LLMEvent] = None client = None diff --git a/agentops/llms/providers/cohere.py b/agentops/llms/providers/cohere.py index f5954bbfa..5e4961216 100644 --- a/agentops/llms/providers/cohere.py +++ b/agentops/llms/providers/cohere.py @@ -2,7 +2,7 @@ import pprint from typing import Optional -from .instrumented_provider import InstrumentedProvider +from .base import BaseProvider from agentops.event import ActionEvent, ErrorEvent, LLMEvent from agentops.session import Session from agentops.log_config import logger @@ -11,7 +11,7 @@ @singleton -class CohereProvider(InstrumentedProvider): +class CohereProvider(BaseProvider): original_create = None original_create_stream = None original_create_async = None diff --git a/agentops/llms/providers/groq.py b/agentops/llms/providers/groq.py index 226a9123e..4f07a9fd9 100644 --- a/agentops/llms/providers/groq.py +++ b/agentops/llms/providers/groq.py @@ -1,7 +1,7 @@ import pprint from typing import Optional -from .instrumented_provider import InstrumentedProvider +from .base import BaseProvider from agentops.event import ErrorEvent, LLMEvent from agentops.session import Session from agentops.log_config import logger @@ -10,7 +10,7 @@ @singleton -class GroqProvider(InstrumentedProvider): +class GroqProvider(BaseProvider): original_create = None original_async_create = None diff --git a/agentops/llms/providers/litellm.py b/agentops/llms/providers/litellm.py index dff40765c..488a94b9b 100644 --- a/agentops/llms/providers/litellm.py +++ b/agentops/llms/providers/litellm.py @@ -5,13 +5,13 @@ from agentops.event import LLMEvent, ErrorEvent from agentops.session import Session from agentops.helpers import get_ISO_time, check_call_stack_for_agent_id -from agentops.llms.providers.instrumented_provider import InstrumentedProvider +from agentops.llms.providers.base import BaseProvider from agentops.time_travel import fetch_completion_override_from_time_travel_cache from agentops.singleton import singleton @singleton -class LiteLLMProvider(InstrumentedProvider): +class LiteLLMProvider(BaseProvider): original_create = None original_create_async = None original_oai_create = None diff --git a/agentops/llms/providers/llama_stack_client.py b/agentops/llms/providers/llama_stack_client.py index ecd008cbe..0f7601536 100644 --- a/agentops/llms/providers/llama_stack_client.py +++ b/agentops/llms/providers/llama_stack_client.py @@ -7,10 +7,10 @@ from agentops.session import Session from agentops.log_config import logger from agentops.helpers import get_ISO_time, check_call_stack_for_agent_id -from agentops.llms.providers.instrumented_provider import InstrumentedProvider +from agentops.llms.providers.base import BaseProvider -class LlamaStackClientProvider(InstrumentedProvider): +class LlamaStackClientProvider(BaseProvider): original_complete = None original_create_turn = None diff --git a/agentops/llms/providers/mistral.py b/agentops/llms/providers/mistral.py index 1754cae52..83f090cf0 100644 --- a/agentops/llms/providers/mistral.py +++ b/agentops/llms/providers/mistral.py @@ -7,10 +7,10 @@ from agentops.session import Session from agentops.log_config import logger from agentops.helpers import get_ISO_time, check_call_stack_for_agent_id -from .instrumented_provider import InstrumentedProvider +from .base import BaseProvider -class MistralProvider(InstrumentedProvider): +class MistralProvider(BaseProvider): original_complete = None original_complete_async = None original_stream = None diff --git a/agentops/llms/providers/ollama.py b/agentops/llms/providers/ollama.py index e944469c9..c83a85bc9 100644 --- a/agentops/llms/providers/ollama.py +++ b/agentops/llms/providers/ollama.py @@ -5,14 +5,14 @@ from agentops.event import LLMEvent from agentops.session import Session from agentops.helpers import get_ISO_time, check_call_stack_for_agent_id -from .instrumented_provider import InstrumentedProvider +from .base import BaseProvider from agentops.singleton import singleton original_func = {} @singleton -class OllamaProvider(InstrumentedProvider): +class OllamaProvider(BaseProvider): original_create = None original_create_async = None diff --git a/agentops/llms/providers/openai.py b/agentops/llms/providers/openai.py index dfba0eca0..171b39fe1 100644 --- a/agentops/llms/providers/openai.py +++ b/agentops/llms/providers/openai.py @@ -1,7 +1,7 @@ import pprint from typing import Optional -from agentops.llms.providers.instrumented_provider import InstrumentedProvider +from agentops.llms.providers.base import BaseProvider from agentops.time_travel import fetch_completion_override_from_time_travel_cache from agentops.event import ActionEvent, ErrorEvent, LLMEvent @@ -12,7 +12,7 @@ @singleton -class OpenAiProvider(InstrumentedProvider): +class OpenAiProvider(BaseProvider): original_create = None original_create_async = None original_assistant_methods = None diff --git a/agentops/llms/providers/taskweaver.py b/agentops/llms/providers/taskweaver.py index 2254897f9..ae2fda1c0 100644 --- a/agentops/llms/providers/taskweaver.py +++ b/agentops/llms/providers/taskweaver.py @@ -6,12 +6,12 @@ from agentops.session import Session from agentops.log_config import logger from agentops.helpers import get_ISO_time, check_call_stack_for_agent_id -from agentops.llms.providers.instrumented_provider import InstrumentedProvider +from agentops.llms.providers.base import BaseProvider from agentops.singleton import singleton @singleton -class TaskWeaverProvider(InstrumentedProvider): +class TaskWeaverProvider(BaseProvider): original_chat_completion = None def __init__(self, client):