Test lasya

libs/arangodb/Makefile

@@ -10,14 +10,14 @@ integration_test integration_tests: TEST_FILE = tests/integration_tests/
 
 # unit tests are run with the --disable-socket flag to prevent network calls
 test tests:
-	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE) --cov-report=term-missing --cov=langchain_arangodb
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE) --cov-report=term-missing --cov=langchain_arangodb --cov-report=html
 
 test_watch:
 	poetry run ptw --snapshot-update --now . -- -vv $(TEST_FILE)
 
 # integration tests are run without the --disable-socket flag to allow network calls
 integration_test integration_tests:
-	poetry run pytest $(TEST_FILE) --cov-report=term-missing --cov=langchain_arangodb
+	poetry run pytest $(TEST_FILE) --cov-report=term-missing --cov=langchain_arangodb --cov-report=html
 
 ######################
 # LINTING AND FORMATTING

libs/arangodb/langchain_arangodb/chains/graph_qa/arangodb.py

@@ -19,7 +19,7 @@
     AQL_GENERATION_PROMPT,
     AQL_QA_PROMPT,
 )
-from langchain_arangodb.graphs.arangodb_graph import ArangoGraph
+from langchain_arangodb.graphs.graph_store import GraphStore
 
 AQL_WRITE_OPERATIONS: List[str] = [
     "INSERT",
@@ -45,7 +45,7 @@ class ArangoGraphQAChain(Chain):
         See https://python.langchain.com/docs/security for more information.
     """
 
-    graph: ArangoGraph = Field(exclude=True)
+    graph: GraphStore = Field(exclude=True)
     aql_generation_chain: Runnable[Dict[str, Any], Any]
     aql_fix_chain: Runnable[Dict[str, Any], Any]
     qa_chain: Runnable[Dict[str, Any], Any]

libs/arangodb/langchain_arangodb/graphs/arangodb_graph.py

@@ -42,10 +42,10 @@ def get_arangodb_client(
     Returns:
         An arango.database.StandardDatabase.
     """
-    _url: str = url or os.environ.get("ARANGODB_URL", "http://localhost:8529")  # type: ignore[assignment]
-    _dbname: str = dbname or os.environ.get("ARANGODB_DBNAME", "_system")  # type: ignore[assignment]
-    _username: str = username or os.environ.get("ARANGODB_USERNAME", "root")  # type: ignore[assignment]
-    _password: str = password or os.environ.get("ARANGODB_PASSWORD", "")  # type: ignore[assignment]
+    _url: str = url or str(os.environ.get("ARANGODB_URL", "http://localhost:8529"))
+    _dbname: str = dbname or str(os.environ.get("ARANGODB_DBNAME", "_system"))
+    _username: str = username or str(os.environ.get("ARANGODB_USERNAME", "root"))
+    _password: str = password or str(os.environ.get("ARANGODB_PASSWORD", ""))
 
     return ArangoClient(_url).db(_dbname, _username, _password, verify=True)
 
@@ -408,7 +408,7 @@ def embed_text(text: str) -> list[float]:
 
         if capitalization_strategy == "none":
             capitalization_fn = lambda x: x  # noqa: E731
-        if capitalization_strategy == "lower":
+        elif capitalization_strategy == "lower":
             capitalization_fn = str.lower
         elif capitalization_strategy == "upper":
             capitalization_fn = str.upper
@@ -500,7 +500,7 @@ def embed_text(text: str) -> list[float]:
             # 2. Process Nodes
             node_key_map = {}
             for i, node in enumerate(document.nodes, 1):
-                node.id = capitalization_fn(str(node.id))
+                node.id = str(capitalization_fn(str(node.id)))
                 node_key = self._hash(node.id)
                 node_key_map[node.id] = node_key