chores(ci): cleanup naming (#24)

* chore(ci): cleanup imports * rename files --------- Co-authored-by: Averi Kitsch <akitsch@google.com>
googleapis · Feb 19, 2024 · dcca171 · dcca171
1 parent c1f369f
commit dcca171
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 13 deletions.
diff --git a/src/langchain_google_cloud_sql_mysql/__init__.py b/src/langchain_google_cloud_sql_mysql/__init__.py
@@ -12,20 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from langchain_google_cloud_sql_mysql.mysql_chat_message_history import (
-    MySQLChatMessageHistory,
-)
-from langchain_google_cloud_sql_mysql.mysql_engine import MySQLEngine
-from langchain_google_cloud_sql_mysql.mysql_loader import (
-    MySQLDocumentSaver,
-    MySQLLoader,
-)
-from langchain_google_cloud_sql_mysql.version import __version__
+from .chat_message_history import MySQLChatMessageHistory
+from .engine import MySQLEngine
+from .loader import MySQLDocumentSaver, MySQLLoader
+from .version import __version__
 
 __all__ = [
     "MySQLChatMessageHistory",
     "MySQLDocumentSaver",
     "MySQLEngine",
     "MySQLLoader",
-    __version__,
+    "__version__",
 ]
diff --git a/...d_sql_mysql/mysql_chat_message_history.py → ...e_cloud_sql_mysql/chat_message_history.py b/...d_sql_mysql/mysql_chat_message_history.py → ...e_cloud_sql_mysql/chat_message_history.py
@@ -18,7 +18,7 @@
 from langchain_core.chat_history import BaseChatMessageHistory
 from langchain_core.messages import BaseMessage, messages_from_dict
 
-from langchain_google_cloud_sql_mysql.mysql_engine import MySQLEngine
+from .engine import MySQLEngine
 
 
 class MySQLChatMessageHistory(BaseChatMessageHistory):

diff --git a/...in_google_cloud_sql_mysql/mysql_engine.py → ...angchain_google_cloud_sql_mysql/engine.py b/...in_google_cloud_sql_mysql/mysql_engine.py → ...angchain_google_cloud_sql_mysql/engine.py
diff --git a/...in_google_cloud_sql_mysql/mysql_loader.py → ...angchain_google_cloud_sql_mysql/loader.py b/...in_google_cloud_sql_mysql/mysql_loader.py → ...angchain_google_cloud_sql_mysql/loader.py
@@ -20,7 +20,7 @@
 from langchain_community.document_loaders.base import BaseLoader
 from langchain_core.documents import Document
 
-from langchain_google_cloud_sql_mysql.mysql_engine import MySQLEngine
+from .engine import MySQLEngine
 
 DEFAULT_CONTENT_COL = "page_content"
 DEFAULT_METADATA_COL = "langchain_metadata"

diff --git a/tests/unit/test_doc2row.py b/tests/unit/test_doc2row.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 from langchain_core.documents import Document
 
-from langchain_google_cloud_sql_mysql.mysql_loader import (
+from langchain_google_cloud_sql_mysql.loader import (
     DEFAULT_CONTENT_COL,
     DEFAULT_METADATA_COL,
     _parse_doc_from_row,