From e821a6195a1b8898b66dd0b8a54fd48fb68aa035 Mon Sep 17 00:00:00 2001
From: Pavlo Paliychuk <pavlo.paliychuk.ca@gmail.com>
Date: Tue, 27 Aug 2024 16:03:08 -0400
Subject: [PATCH] chore: Move anthropic to dev deps, remove anthropic and groq
 clients from __init__ (#61)

---
 graphiti_core/llm_client/__init__.py | 4 +---
 poetry.lock                          | 5 ++---
 pyproject.toml                       | 3 +--
 3 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/graphiti_core/llm_client/__init__.py b/graphiti_core/llm_client/__init__.py
index 9c1255b1..24f9a46f 100644
--- a/graphiti_core/llm_client/__init__.py
+++ b/graphiti_core/llm_client/__init__.py
@@ -1,7 +1,5 @@
-from .anthropic_client import AnthropicClient
 from .client import LLMClient
 from .config import LLMConfig
-from .groq_client import GroqClient
 from .openai_client import OpenAIClient
 
-__all__ = ['LLMClient', 'OpenAIClient', 'LLMConfig', 'AnthropicClient', 'GroqClient']
+__all__ = ['LLMClient', 'OpenAIClient', 'LLMConfig']
diff --git a/poetry.lock b/poetry.lock
index 22653da0..902d58a0 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 1.8.3 and should not be changed by hand.
+# This file is automatically @generated by Poetry 1.7.1 and should not be changed by hand.
 
 [[package]]
 name = "annotated-types"
@@ -1804,7 +1804,6 @@ description = "Nvidia JIT LTO Library"
 optional = false
 python-versions = ">=3"
 files = [
-    {file = "nvidia_nvjitlink_cu12-12.6.20-py3-none-manylinux2014_aarch64.whl", hash = "sha256:84fb38465a5bc7c70cbc320cfd0963eb302ee25a5e939e9f512bbba55b6072fb"},
     {file = "nvidia_nvjitlink_cu12-12.6.20-py3-none-manylinux2014_x86_64.whl", hash = "sha256:562ab97ea2c23164823b2a89cb328d01d45cb99634b8c65fe7cd60d14562bd79"},
     {file = "nvidia_nvjitlink_cu12-12.6.20-py3-none-win_amd64.whl", hash = "sha256:ed3c43a17f37b0c922a919203d2d36cbef24d41cc3e6b625182f8b58203644f6"},
 ]
@@ -3758,4 +3757,4 @@ test = ["websockets"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "001663dfc8078ad473675c994b15191db1f53a844e23f40ffa4a704379a61132"
+content-hash = "425298614e73ee8b1333343bdf27e876849d56b98671d1ad7b30ca6c6584d203"
diff --git a/pyproject.toml b/pyproject.toml
index 956e48d4..1c6770b1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,9 +20,7 @@ fastapi = "^0.112.0"
 neo4j = "^5.23.0"
 sentence-transformers = "^3.0.1"
 diskcache = "^5.6.3"
-arrow = "^1.3.0"
 openai = "^1.38.0"
-anthropic = "^0.34.1"
 tenacity = "^9.0.0"
 
 [tool.poetry.dev-dependencies]
@@ -36,6 +34,7 @@ ruff = "^0.6.2"
 pydantic = "^2.8.2"
 mypy = "^1.11.1"
 groq = "^0.9.0"
+anthropic = "^0.34.1"
 ipykernel = "^6.29.5"
 jupyterlab = "^4.2.4"
 diskcache-stubs = "^5.6.3.6.20240818"