chore: reorganize imports

Author: skoob13

llm_observability_examples.py

@@ -2,7 +2,7 @@
 import uuid
 
 import posthog
-from posthog.ai import AsyncOpenAI, OpenAI
+from posthog.ai.openai import AsyncOpenAI, OpenAI
 
 # Example credentials - replace these with your own or use environment variables
 posthog.project_api_key = os.getenv("POSTHOG_PROJECT_API_KEY", "your-project-api-key")

posthog/ai/__init__.py

@@ -1,4 +0,0 @@
-from .providers.openai.openai import OpenAI
-from .providers.openai.openai_async import AsyncOpenAI
-
-__all__ = ["OpenAI", "AsyncOpenAI"]

posthog/ai/langchain/__init__.py

@@ -0,0 +1,3 @@
+from .callbacks import PosthogCallbackHandler
+
+__all__ = ["PosthogCallbackHandler"]

posthog/ai/providers/langchain.py posthog/ai/langchain/callbacks.pyposthog/ai/providers/langchain.py renamed to posthog/ai/langchain/callbacks.py

@@ -42,6 +42,10 @@ class RunMetadata(TypedDict, total=False):
 
 
 class PosthogCallbackHandler(BaseCallbackHandler):
+    """
+    A callback handler for LangChain that sends events to PostHog LLM Observability.
+    """
+
     _client: Client
     """PostHog client instance."""
     _distinct_id: Optional[str]

posthog/ai/openai/__init__.py

@@ -0,0 +1,4 @@
+from .openai import OpenAI
+from .openai_async import AsyncOpenAI
+
+__all__ = ["OpenAI", "AsyncOpenAI"]

posthog/ai/providers/openai/openai.py posthog/ai/openai/openai.pyposthog/ai/providers/openai/openai.py renamed to posthog/ai/openai/openai.py

@@ -2,13 +2,13 @@
 import uuid
 from typing import Any, Dict, Optional
 
-import openai.resources
-
 try:
     import openai
 except ImportError:
     raise ModuleNotFoundError("Please install the OpenAI SDK to use this feature: 'pip install openai'")
 
+import openai.resources
+
 from posthog.ai.utils import call_llm_and_track_usage, get_model_params
 from posthog.client import Client as PostHogClient
 

…thog/ai/providers/openai/openai_async.py posthog/ai/openai/openai_async.pyposthog/ai/providers/openai/openai_async.py renamed to posthog/ai/openai/openai_async.py posthog/ai/providers/openai/openai_async.py renamed to posthog/ai/openai/openai_async.py

@@ -2,13 +2,13 @@
 import uuid
 from typing import Any, Dict, Optional
 
-import openai.resources
-
 try:
     import openai
 except ImportError:
     raise ModuleNotFoundError("Please install the OpenAI SDK to use this feature: 'pip install openai'")
 
+import openai.resources
+
 from posthog.ai.utils import call_llm_and_track_usage_async, get_model_params
 from posthog.client import Client as PostHogClient