Enhance documentation and functionality for trace management (#1012)

* Enhance documentation and functionality for trace management

- Updated `mint.json` to include new usage paths for trace decorators and manual trace control.
- Revised `introduction.mdx` to reflect changes in session tracking terminology, replacing "sessions" with "traces" for clarity.
- Expanded `quickstart.mdx` to introduce the `@trace` decorator and manual trace management examples.
- Added new documentation files for `manual-trace-control.mdx` and `trace-decorator.mdx`, detailing advanced trace management techniques and usage patterns.
- Updated `sdk-reference.mdx` to include new parameters for trace management and improved examples for initializing the AgentOps SDK.
- Enhanced `tracking-agents.mdx` to demonstrate multi-agent workflows and coordination using the `@trace` decorator.

These changes improve the clarity and usability of the AgentOps SDK, particularly around trace management and agent tracking.

* Enhance documentation for trace management and agent tracking

- Updated `introduction.mdx` to clarify the use of the `@trace` decorator for custom trace creation.
- Revised `quickstart.mdx` to include examples of initializing AgentOps with manual trace control.
- Improved `examples.mdx` by updating image links to use raw GitHub URLs for better accessibility.
- Enhanced `manual-trace-control.mdx` to provide detailed examples of managing traces manually.
- Updated `recording-operations.mdx` to emphasize the use of the `@trace` decorator for grouping operations.
- Refined `sdk-reference.mdx` to clarify parameters for ending traces and the use of decorators.
- Improved `tracking-agents.mdx` to demonstrate structured tracking with the `@agent` decorator in multi-agent systems.

These changes improve the clarity and usability of the AgentOps SDK, particularly around trace management and agent tracking.

* Add Google Generative AI instrumentation module

- Introduced a new module for Google Generative AI (Gemini) API instrumentation, including `GoogleGenAIInstrumentor` for capturing telemetry data.
- Added attribute extraction functions for model and chat interactions, enhancing observability of API calls.
- Implemented wrappers for synchronous and asynchronous streaming methods to track performance metrics.
- Updated documentation to include usage examples and details on supported features for the new instrumentation.
- Refactored existing code to accommodate the new module structure and improve clarity.

These changes enhance the AgentOps SDK by providing comprehensive support for monitoring Google Generative AI interactions.

* Update mint.json to reorder integration paths, moving 'anthropic' to the correct position in the integrations list. This change improves the organization and accessibility of integration documentation.

Author: Dwij1704

agentops/instrumentation/__init__.py

@@ -183,8 +183,8 @@ class InstrumentorConfig(TypedDict):
         "min_version": "0.1.0",
     },
     "google.genai": {
-        "module_name": "agentops.instrumentation.google_generativeai",
-        "class_name": "GoogleGenerativeAIInstrumentor",
+        "module_name": "agentops.instrumentation.google_genai",
+        "class_name": "GoogleGenAIInstrumentor",
         "min_version": "0.1.0",
         "package_name": "google-genai",  # Actual pip package name
     },

agentops/instrumentation/google_generativeai/README.md renamed to agentops/instrumentation/google_genai/README.md

@@ -31,10 +31,10 @@ def get_version() -> str:
 logger = logging.getLogger(__name__)
 
 # Import after defining constants to avoid circular imports
-from agentops.instrumentation.google_generativeai.instrumentor import GoogleGenerativeAIInstrumentor  # noqa: E402
+from agentops.instrumentation.google_genai.instrumentor import GoogleGenAIInstrumentor  # noqa: E402
 
 __all__ = [
     "LIBRARY_NAME",
     "LIBRARY_VERSION",
-    "GoogleGenerativeAIInstrumentor",
+    "GoogleGenAIInstrumentor",
 ]

agentops/instrumentation/google_generativeai/__init__.py renamed to agentops/instrumentation/google_genai/__init__.py

@@ -1,16 +1,16 @@
 """Attribute extractors for Google Generative AI instrumentation."""
 
-from agentops.instrumentation.google_generativeai.attributes.common import (
+from agentops.instrumentation.google_genai.attributes.common import (
     get_common_instrumentation_attributes,
     extract_request_attributes,
 )
-from agentops.instrumentation.google_generativeai.attributes.model import (
+from agentops.instrumentation.google_genai.attributes.model import (
     get_model_attributes,
     get_generate_content_attributes,
     get_stream_attributes,
     get_token_counting_attributes,
 )
-from agentops.instrumentation.google_generativeai.attributes.chat import (
+from agentops.instrumentation.google_genai.attributes.chat import (
     get_chat_attributes,
 )
 

agentops/instrumentation/google_generativeai/attributes/__init__.py renamed to agentops/instrumentation/google_genai/attributes/__init__.py

@@ -5,11 +5,11 @@
 from agentops.logging import logger
 from agentops.semconv import SpanAttributes, LLMRequestTypeValues, MessageAttributes
 from agentops.instrumentation.common.attributes import AttributeMap
-from agentops.instrumentation.google_generativeai.attributes.common import (
+from agentops.instrumentation.google_genai.attributes.common import (
     extract_request_attributes,
     get_common_instrumentation_attributes,
 )
-from agentops.instrumentation.google_generativeai.attributes.model import (
+from agentops.instrumentation.google_genai.attributes.model import (
     _extract_content_from_prompt,
     _set_response_attributes,
 )

agentops/instrumentation/google_generativeai/attributes/chat.py renamed to agentops/instrumentation/google_genai/attributes/chat.py

@@ -9,7 +9,7 @@
     get_common_attributes,
     _extract_attributes_from_mapping,
 )
-from agentops.instrumentation.google_generativeai import LIBRARY_NAME, LIBRARY_VERSION
+from agentops.instrumentation.google_genai import LIBRARY_NAME, LIBRARY_VERSION
 
 # Common mapping for config parameters
 REQUEST_CONFIG_ATTRIBUTES: AttributeMap = {

agentops/instrumentation/google_generativeai/attributes/common.py renamed to agentops/instrumentation/google_genai/attributes/common.py

@@ -5,7 +5,7 @@
 from agentops.logging import logger
 from agentops.semconv import SpanAttributes, LLMRequestTypeValues, MessageAttributes
 from agentops.instrumentation.common.attributes import AttributeMap
-from agentops.instrumentation.google_generativeai.attributes.common import (
+from agentops.instrumentation.google_genai.attributes.common import (
     extract_request_attributes,
     get_common_instrumentation_attributes,
 )

agentops/instrumentation/google_generativeai/attributes/model.py renamed to agentops/instrumentation/google_genai/attributes/model.py

@@ -16,12 +16,12 @@
 
 from agentops.logging import logger
 from agentops.instrumentation.common.wrappers import WrapConfig, wrap, unwrap
-from agentops.instrumentation.google_generativeai import LIBRARY_NAME, LIBRARY_VERSION
-from agentops.instrumentation.google_generativeai.attributes.model import (
+from agentops.instrumentation.google_genai import LIBRARY_NAME, LIBRARY_VERSION
+from agentops.instrumentation.google_genai.attributes.model import (
     get_generate_content_attributes,
     get_token_counting_attributes,
 )
-from agentops.instrumentation.google_generativeai.stream_wrapper import (
+from agentops.instrumentation.google_genai.stream_wrapper import (
     generate_content_stream_wrapper,
     generate_content_stream_async_wrapper,
 )
@@ -96,7 +96,7 @@
 ]
 
 
-class GoogleGenerativeAIInstrumentor(BaseInstrumentor):
+class GoogleGenAIInstrumentor(BaseInstrumentor):
     """An instrumentor for Google Generative AI (Gemini) API.
 
     This class provides instrumentation for Google's Generative AI API by wrapping key methods

agentops/instrumentation/google_generativeai/instrumentor.py renamed to agentops/instrumentation/google_genai/instrumentor.py

@@ -14,11 +14,11 @@
 
 from agentops.semconv import SpanAttributes, LLMRequestTypeValues, CoreAttributes, MessageAttributes
 from agentops.instrumentation.common.wrappers import _with_tracer_wrapper
-from agentops.instrumentation.google_generativeai.attributes.model import (
+from agentops.instrumentation.google_genai.attributes.model import (
     get_generate_content_attributes,
     get_stream_attributes,
 )
-from agentops.instrumentation.google_generativeai.attributes.common import (
+from agentops.instrumentation.google_genai.attributes.common import (
     extract_request_attributes,
 )
 

agentops/instrumentation/google_generativeai/stream_wrapper.py renamed to agentops/instrumentation/google_genai/stream_wrapper.py

@@ -167,8 +167,9 @@
     {
       "group": "Integrations",
       "pages": [
-        "v2/integrations/anthropic",
+        "v2/integrations/ag2",
         "v2/integrations/autogen",
+        "v2/integrations/anthropic",
         "v2/integrations/crewai",
         "v2/integrations/google_adk",
         "v2/integrations/gemini",
@@ -188,7 +189,9 @@
         "v2/usage/advanced-configuration",
         "v2/usage/tracking-llm-calls",
         "v2/usage/tracking-agents",
-        "v2/usage/recording-operations"
+        "v2/usage/recording-operations",
+        "v2/usage/trace-decorator",
+        "v2/usage/manual-trace-control"
       ],
       "version": "v2"
     },