feat: Refactor subscription streaming and plumb through anthropic sup…

…port (#5105) * Extract/Refactor OpenAI/AzureOpenAI model interfaces * Add Anthropic client * Use new payload type * run anthropic and fix most of the types * use abc * add anthropic to type check dependencies * add anthropic as a container dependency * fix tool calls for openai * add back in token usage * fix type --------- Co-authored-by: Alexander Song <axiomofjoy@gmail.com>
Arize-ai · Oct 19, 2024 · d405f61 · d405f61
1 parent 33eac82
commit d405f61
Show file tree

Hide file tree

Showing 3 changed files with 229 additions and 94 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -123,6 +123,7 @@ pg = [
   "psycopg[binary,pool]",
 ]
 container = [
+  "anthropic",
   "prometheus-client",
   "openai>=1.0.0",
   "opentelemetry-sdk",

diff --git a/requirements/type-check.txt b/requirements/type-check.txt
@@ -1,4 +1,5 @@
 -r ci.txt
+anthropic
 grpcio
 litellm>=1.0.3
 openai>=1.0.0