Merge pull request #133 from stealthrocket/openapi-integration

OpenAI integration

Author: chriso

src/dispatch/integrations/__init__.py

@@ -6,7 +6,7 @@
 
 # Automatically register error and output types from
 # commonly used libraries.
-integrations = ("httpx", "requests", "slack")
+integrations = ("httpx", "requests", "slack", "openai")
 for name in integrations:
     try:
         importlib.import_module(f"dispatch.integrations.{name}")

src/dispatch/integrations/openai.py

@@ -0,0 +1,19 @@
+import openai  # type: ignore
+
+from dispatch.integrations.http import http_response_code_status
+from dispatch.status import Status, register_error_type
+
+
+def openai_error_status(error: Exception) -> Status:
+    # See https://github.com/openai/openai-python/blob/main/src/openai/_exceptions.py
+    match error:
+        case openai.APITimeoutError():
+            return Status.TIMEOUT
+        case openai.APIStatusError():
+            return http_response_code_status(error.status_code)
+
+    return Status.TEMPORARY_ERROR
+
+
+# Register base exception.
+register_error_type(openai.OpenAIError, openai_error_status)