feat: clean up code, remove duplication and add examples for streaming with async or not

Author: k11kirky

llm_observability_examples.py

@@ -33,7 +33,7 @@ def main_sync():
 
 async def main_async():
     try:
-        #await basic_async_openai_call()
+        await basic_async_openai_call()
         await streaming_async_openai_call()
     except Exception as e:
         print("Error during OpenAI call:", str(e))
@@ -93,6 +93,9 @@ async def streaming_async_openai_call():
 
     return response
 
+# HOW TO RUN:
+# comment out one of these to run the other
+
 if __name__ == "__main__":
     main_sync()
 

posthog/ai/__init__.py

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