Merge pull request #22 from davidzhangbj/modifyPackageName

optimize: rename all MySQL-related package names, file names, and var…

Author: xxsc0529

langgraph-checkpoint-oceanbase-plugin/README.md

@@ -24,13 +24,13 @@ In LangGraph's definition, Checkpointer refers to short-term memory (i.e. memory
 ```python
 from langchain.chat_models import init_chat_model
 from langgraph.graph import StateGraph, MessagesState, START
-from langgraph.checkpoint.mysql.pymysql import PyMySQLSaver
+from langgraph.checkpoint.oceanbase.pyoceanbase import PyOceanBaseSaver
 from langchain_core.runnables.config import RunnableConfig
 from langchain_core.messages import HumanMessage
 model = init_chat_model(model="qwen-max-latest", api_key="xxx",
                         base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", model_provider="openai",temperature=0)
 DB_URI = "mysql://username:password@ip:port/database"
-with PyMySQLSaver.from_conn_string(DB_URI) as checkpointer:
+with PyOceanBaseSaver.from_conn_string(DB_URI) as checkpointer:
     checkpointer.setup()
 
     def call_model(state: MessagesState):
@@ -63,7 +63,7 @@ with PyMySQLSaver.from_conn_string(DB_URI) as checkpointer:
 from langchain_core.runnables import RunnableConfig
 from langgraph.config import get_store
 from langgraph.prebuilt import create_react_agent
-from langgraph.store.mysql import PyMySQLStore
+from langgraph.store.oceanbase import PyOceanBaseStore
 from langchain.chat_models import init_chat_model
 from langchain_core.messages import HumanMessage
 from typing_extensions import TypedDict
@@ -86,7 +86,7 @@ def save_user_info(user_info: UserInfo, config: RunnableConfig) -> str:
     return "Successfully saved user info."
 
 
-with PyMySQLStore.from_conn_string(DB_URI) as store:
+with PyOceanBaseStore.from_conn_string(DB_URI) as store:
     store.setup()
     agent=create_react_agent(
         model=model,

langgraph-checkpoint-oceanbase-plugin/README_CN.md

@@ -24,13 +24,13 @@ langgraph-checkpoint-oceanbase 已经上传到 PyPI。可以使用下面的命
 ```python
 from langchain.chat_models import init_chat_model
 from langgraph.graph import StateGraph, MessagesState, START
-from langgraph.checkpoint.mysql.pymysql import PyMySQLSaver
+from langgraph.checkpoint.oceanbase.pyoceanbase import PyOceanBaseSaver
 from langchain_core.runnables.config import RunnableConfig
 from langchain_core.messages import HumanMessage
 model = init_chat_model(model="qwen-max-latest", api_key="xxx",
                         base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", model_provider="openai",temperature=0)
 DB_URI = "mysql://username:password@ip:port/database"
-with PyMySQLSaver.from_conn_string(DB_URI) as checkpointer:
+with PyOceanBaseSaver.from_conn_string(DB_URI) as checkpointer:
     checkpointer.setup()
 
     def call_model(state: MessagesState):
@@ -63,7 +63,7 @@ with PyMySQLSaver.from_conn_string(DB_URI) as checkpointer:
 from langchain_core.runnables import RunnableConfig
 from langgraph.config import get_store
 from langgraph.prebuilt import create_react_agent
-from langgraph.store.mysql import PyMySQLStore
+from langgraph.store.oceanbase import PyOceanBaseStore
 from langchain.chat_models import init_chat_model
 from langchain_core.messages import HumanMessage
 from typing_extensions import TypedDict
@@ -86,7 +86,7 @@ def save_user_info(user_info: UserInfo, config: RunnableConfig) -> str:
     return "Successfully saved user info."
 
 
-with PyMySQLStore.from_conn_string(DB_URI) as store:
+with PyOceanBaseStore.from_conn_string(DB_URI) as store:
     store.setup()
     agent=create_react_agent(
         model=model,

langgraph-checkpoint-oceanbase-plugin/langgraph-tests/tests/conftest_checkpointer.py

@@ -6,9 +6,9 @@
 import pymysql
 from sqlalchemy import Engine, create_engine
 
-from langgraph.checkpoint.mysql.aio import AIOMySQLSaver
-from langgraph.checkpoint.mysql.asyncmy import AsyncMySaver
-from langgraph.checkpoint.mysql.pymysql import PyMySQLSaver
+from langgraph.checkpoint.oceanbase.aio import AIOMySQLSaver
+from langgraph.checkpoint.oceanbase.asyncmy import AsyncMySaver
+from langgraph.checkpoint.oceanbase.pyoceanbase import PyOceanBaseSaver
 
 DEFAULT_MYSQL_URI = "mysql://mysql:mysql@localhost:5441/"
 
@@ -23,19 +23,19 @@ def _checkpointer_pymysql():
     database = f"test_{uuid4().hex[:16]}"
 
     # create unique db
-    with pymysql.connect(**PyMySQLSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+    with pymysql.connect(**PyOceanBaseSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
         with conn.cursor() as cursor:
             cursor.execute(f"CREATE DATABASE {database}")
     try:
         # yield checkpointer
-        with PyMySQLSaver.from_conn_string(
+        with PyOceanBaseSaver.from_conn_string(
             DEFAULT_MYSQL_URI + database
         ) as checkpointer:
             checkpointer.setup()
             yield checkpointer
     finally:
         # drop unique db
-        with pymysql.connect(**PyMySQLSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+        with pymysql.connect(**PyOceanBaseSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
             with conn.cursor() as cursor:
                 cursor.execute(f"DROP DATABASE {database}")
 
@@ -45,17 +45,17 @@ def _checkpointer_pymysql_pool():
     database = f"test_{uuid4().hex[:16]}"
 
     # create unique db
-    with pymysql.connect(**PyMySQLSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+    with pymysql.connect(**PyOceanBaseSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
         with conn.cursor() as cursor:
             cursor.execute(f"CREATE DATABASE {database}")
     try:
         pool = get_pymysql_sqlalchemy_engine(DEFAULT_MYSQL_URI + database)
-        checkpointer = PyMySQLSaver(pool.raw_connection)
+        checkpointer = PyOceanBaseSaver(pool.raw_connection)
         checkpointer.setup()
         yield checkpointer
     finally:
         # drop unique db
-        with pymysql.connect(**PyMySQLSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+        with pymysql.connect(**PyOceanBaseSaver.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
             with conn.cursor() as cursor:
                 cursor.execute(f"DROP DATABASE {database}")
 

langgraph-checkpoint-oceanbase-plugin/langgraph-tests/tests/conftest_store.py

@@ -6,9 +6,9 @@
 import pymysql
 from sqlalchemy import Engine, create_engine
 
-from langgraph.store.mysql.aio import AIOMySQLStore
-from langgraph.store.mysql.asyncmy import AsyncMyStore
-from langgraph.store.mysql.pymysql import PyMySQLStore
+from langgraph.store.oceanbase.aio import AIOMySQLStore
+from langgraph.store.oceanbase.asyncmy import AsyncMyStore
+from langgraph.store.oceanbase.pyoceanbase import PyOceanBaseStore
 
 DEFAULT_MYSQL_URI = "mysql://mysql:mysql@localhost:5441/"
 
@@ -23,17 +23,17 @@ def _store_pymysql():
     database = f"test_{uuid4().hex[:16]}"
 
     # create unique db
-    with pymysql.connect(**PyMySQLStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+    with pymysql.connect(**PyOceanBaseStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
         with conn.cursor() as cursor:
             cursor.execute(f"CREATE DATABASE {database}")
     try:
         # yield store
-        with PyMySQLStore.from_conn_string(DEFAULT_MYSQL_URI + database) as store:
+        with PyOceanBaseStore.from_conn_string(DEFAULT_MYSQL_URI + database) as store:
             store.setup()
             yield store
     finally:
         # drop unique db
-        with pymysql.connect(**PyMySQLStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+        with pymysql.connect(**PyOceanBaseStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
             with conn.cursor() as cursor:
                 cursor.execute(f"DROP DATABASE {database}")
 
@@ -43,18 +43,18 @@ def _store_pymysql_pool():
     database = f"test_{uuid4().hex[:16]}"
 
     # create unique db
-    with pymysql.connect(**PyMySQLStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+    with pymysql.connect(**PyOceanBaseStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
         with conn.cursor() as cursor:
             cursor.execute(f"CREATE DATABASE {database}")
     try:
         # yield store
         engine = get_pymysql_sqlalchemy_engine(DEFAULT_MYSQL_URI + database)
-        store = PyMySQLStore(engine.raw_connection)
+        store = PyOceanBaseStore(engine.raw_connection)
         store.setup()
         yield store
     finally:
         # drop unique db
-        with pymysql.connect(**PyMySQLStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
+        with pymysql.connect(**PyOceanBaseStore.parse_conn_string(DEFAULT_MYSQL_URI), autocommit=True) as conn:
             with conn.cursor() as cursor:
                 cursor.execute(f"DROP DATABASE {database}")
 

…n/langgraph/checkpoint/mysql/__init__.py …nggraph/checkpoint/oceanbase/__init__.pylanggraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/__init__.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/__init__.py langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/__init__.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/__init__.py

@@ -18,9 +18,9 @@
     get_checkpoint_id,
     get_checkpoint_metadata,
 )
-from langgraph.checkpoint.mysql import _internal
-from langgraph.checkpoint.mysql.base import BaseMySQLSaver
-from langgraph.checkpoint.mysql.utils import (
+from langgraph.checkpoint.oceanbase import _internal
+from langgraph.checkpoint.oceanbase.base import BaseMySQLSaver
+from langgraph.checkpoint.oceanbase.utils import (
     deserialize_channel_values,
     deserialize_pending_sends,
     deserialize_pending_writes,
@@ -115,7 +115,7 @@ def list(
             Iterator[CheckpointTuple]: An iterator of checkpoint tuples.
 
         Examples:
-            >>> from langgraph.checkpoint.mysql import PyMySQLSaver
+            >>> from langgraph.checkpoint.oceanbase import PyOceanBaseSaver
             >>> DB_URI = "mysql://mysql:mysql@localhost:5432/mysql"
             >>> with PyMySQLSaver.from_conn_string(DB_URI) as memory:
             ... # Run a graph, then list the checkpoints
@@ -284,9 +284,9 @@ def put(
 
         Examples:
 
-            >>> from langgraph.checkpoint.mysql import PyMySQLSaver
+            >>> from langgraph.checkpoint.oceanbase import PyOceanBaseSaver
             >>> DB_URI = "mysql://mysql:mysql@localhost:5432/mysql"
-            >>> with PyMySQLSaver.from_conn_string(DB_URI) as memory:
+            >>> with PyOceanBaseSaver.from_conn_string(DB_URI) as memory:
             >>>     config = {"configurable": {"thread_id": "1", "checkpoint_ns": ""}}
             >>>     checkpoint = {"ts": "2024-05-04T06:32:42.235444+00:00", "id": "1ef4f797-8335-6428-8001-8a1503f9b875", "channel_values": {"key": "value"}}
             >>>     saved_config = memory.put(config, checkpoint, {"source": "input", "step": 1, "writes": {"key": "value"}}, {})

…langgraph/checkpoint/mysql/_ainternal.py …graph/checkpoint/oceanbase/_ainternal.pylanggraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/_ainternal.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/_ainternal.py langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/_ainternal.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/_ainternal.py

@@ -9,9 +9,9 @@
 import aiomysql  # type: ignore
 from typing_extensions import Self, override
 
-from langgraph.checkpoint.mysql import _ainternal
-from langgraph.checkpoint.mysql.aio_base import BaseAsyncMySQLSaver
-from langgraph.checkpoint.mysql.shallow import BaseShallowAsyncMySQLSaver
+from langgraph.checkpoint.oceanbase import _ainternal
+from langgraph.checkpoint.oceanbase.aio_base import BaseAsyncMySQLSaver
+from langgraph.checkpoint.oceanbase.shallow import BaseShallowAsyncMySQLSaver
 from langgraph.checkpoint.serde.base import SerializerProtocol
 
 Conn = _ainternal.Conn[aiomysql.Connection]  # For backward compatibility

…/langgraph/checkpoint/mysql/_internal.py …ggraph/checkpoint/oceanbase/_internal.pylanggraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/_internal.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/_internal.py langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/_internal.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/_internal.py

@@ -18,9 +18,9 @@
     get_checkpoint_id,
     get_checkpoint_metadata,
 )
-from langgraph.checkpoint.mysql import _ainternal
-from langgraph.checkpoint.mysql.base import BaseMySQLSaver
-from langgraph.checkpoint.mysql.utils import (
+from langgraph.checkpoint.oceanbase import _ainternal
+from langgraph.checkpoint.oceanbase.base import BaseMySQLSaver
+from langgraph.checkpoint.oceanbase.utils import (
     deserialize_channel_values,
     deserialize_pending_sends,
     deserialize_pending_writes,

…plugin/langgraph/checkpoint/mysql/aio.py …in/langgraph/checkpoint/oceanbase/aio.pylanggraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/aio.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/aio.py langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/mysql/aio.py renamed to langgraph-checkpoint-oceanbase-plugin/langgraph/checkpoint/oceanbase/aio.py

@@ -10,8 +10,8 @@
 from asyncmy.cursors import DictCursor  # type: ignore
 from typing_extensions import Self, override
 
-from langgraph.checkpoint.mysql.aio_base import BaseAsyncMySQLSaver
-from langgraph.checkpoint.mysql.shallow import BaseShallowAsyncMySQLSaver
+from langgraph.checkpoint.oceanbase.aio_base import BaseAsyncMySQLSaver
+from langgraph.checkpoint.oceanbase.shallow import BaseShallowAsyncMySQLSaver
 from langgraph.checkpoint.serde.base import SerializerProtocol