Rename InstrumentedProvider to BaseProvider (#645)

Rename InstrumentedProvider to BaseProvider

Signed-off-by: Teo <[email protected]>

Author: teocns

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.
+By contributing to AgentOps, you agree that your contributions will be licensed under the MIT License.

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

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
 

…/llms/providers/instrumented_provider.py agentops/llms/providers/base.pyagentops/llms/providers/instrumented_provider.py renamed to agentops/llms/providers/base.py agentops/llms/providers/instrumented_provider.py renamed to 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

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

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
 

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

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
 

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

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