style: apply ruff formatting to AI examples

Author: richardsolomou

examples/example-ai-anthropic/chat.py

@@ -6,7 +6,10 @@
 from posthog import Posthog
 from posthog.ai.anthropic import Anthropic
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 client = Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"], posthog_client=posthog)
 
 tools = [

examples/example-ai-anthropic/extended_thinking.py

@@ -7,15 +7,23 @@
 from posthog import Posthog
 from posthog.ai.anthropic import Anthropic
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 client = Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"], posthog_client=posthog)
 
 message = client.messages.create(
     model="claude-sonnet-4-5-20250929",
     max_tokens=16000,
     posthog_distinct_id="example-user",
     thinking={"type": "enabled", "budget_tokens": 10000},
-    messages=[{"role": "user", "content": "What is the probability of rolling at least one six in four rolls of a fair die?"}],
+    messages=[
+        {
+            "role": "user",
+            "content": "What is the probability of rolling at least one six in four rolls of a fair die?",
+        }
+    ],
 )
 
 for block in message.content:

examples/example-ai-anthropic/streaming.py

@@ -4,7 +4,10 @@
 from posthog import Posthog
 from posthog.ai.anthropic import Anthropic
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 client = Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"], posthog_client=posthog)
 
 stream = client.messages.create(

examples/example-ai-gemini/chat.py

@@ -7,7 +7,10 @@
 from posthog import Posthog
 from posthog.ai.gemini import Client
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 client = Client(api_key=os.environ["GEMINI_API_KEY"], posthog_client=posthog)
 
 tool_declarations = [

examples/example-ai-gemini/image_generation.py

@@ -4,7 +4,10 @@
 from posthog import Posthog
 from posthog.ai.gemini import Client
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 client = Client(api_key=os.environ["GEMINI_API_KEY"], posthog_client=posthog)
 
 response = client.models.generate_content(
@@ -16,7 +19,9 @@
 for candidate in response.candidates:
     for part in candidate.content.parts:
         if hasattr(part, "inline_data") and part.inline_data:
-            print(f"Generated image: {part.inline_data.mime_type}, {len(part.inline_data.data)} bytes")
+            print(
+                f"Generated image: {part.inline_data.mime_type}, {len(part.inline_data.data)} bytes"
+            )
         elif hasattr(part, "text"):
             print(part.text)
 

examples/example-ai-gemini/streaming.py

@@ -4,13 +4,21 @@
 from posthog import Posthog
 from posthog.ai.gemini import Client
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 client = Client(api_key=os.environ["GEMINI_API_KEY"], posthog_client=posthog)
 
 stream = client.models.generate_content_stream(
     model="gemini-2.5-flash",
     posthog_distinct_id="example-user",
-    contents=[{"role": "user", "parts": [{"text": "Explain product analytics in three sentences."}]}],
+    contents=[
+        {
+            "role": "user",
+            "parts": [{"text": "Explain product analytics in three sentences."}],
+        }
+    ],
 )
 
 for chunk in stream:

examples/example-ai-langchain/callback_handler.py

@@ -5,11 +5,14 @@
 import urllib.request
 from langchain_openai import ChatOpenAI
 from langchain_core.tools import tool
-from langchain_core.messages import HumanMessage, ToolMessage
+from langchain_core.messages import HumanMessage
 from posthog import Posthog
 from posthog.ai.langchain import CallbackHandler
 
-posthog = Posthog(os.environ["POSTHOG_API_KEY"], host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"))
+posthog = Posthog(
+    os.environ["POSTHOG_API_KEY"],
+    host=os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com"),
+)
 callback_handler = CallbackHandler(client=posthog)
 
 

examples/example-ai-langchain/otel.py

@@ -17,7 +17,9 @@
 os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Bearer {posthog_api_key}"
 
 tracer_provider = TracerProvider(
-    resource=Resource.create({"service.name": "langchain-example", "user.id": "example-user"})
+    resource=Resource.create(
+        {"service.name": "langchain-example", "user.id": "example-user"}
+    )
 )
 tracer_provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
 trace.set_tracer_provider(tracer_provider)

examples/example-ai-litellm/chat.py

@@ -7,7 +7,9 @@
 
 # Enable PostHog callbacks — LiteLLM has built-in PostHog support
 os.environ["POSTHOG_API_KEY"] = os.environ.get("POSTHOG_API_KEY", "")
-os.environ["POSTHOG_API_URL"] = os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com")
+os.environ["POSTHOG_API_URL"] = os.environ.get(
+    "POSTHOG_HOST", "https://us.i.posthog.com"
+)
 litellm.success_callback = ["posthog"]
 litellm.failure_callback = ["posthog"]
 
@@ -43,7 +45,10 @@ def get_weather(latitude: float, longitude: float, location_name: str) -> str:
 response = litellm.completion(
     model="gpt-4o-mini",
     messages=[
-        {"role": "system", "content": "You are a helpful assistant with access to weather data."},
+        {
+            "role": "system",
+            "content": "You are a helpful assistant with access to weather data.",
+        },
         {"role": "user", "content": "What's the weather in Paris?"},
     ],
     tools=tools,

examples/example-ai-litellm/streaming.py

@@ -4,7 +4,9 @@
 import litellm
 
 os.environ["POSTHOG_API_KEY"] = os.environ.get("POSTHOG_API_KEY", "")
-os.environ["POSTHOG_API_URL"] = os.environ.get("POSTHOG_HOST", "https://us.i.posthog.com")
+os.environ["POSTHOG_API_URL"] = os.environ.get(
+    "POSTHOG_HOST", "https://us.i.posthog.com"
+)
 litellm.success_callback = ["posthog"]
 litellm.failure_callback = ["posthog"]