x

langgraph_tutorials/__init__.py

@@ -0,0 +1 @@
+"""Supporting code for LangGraph tutorials."""

langgraph_tutorials/customer_support/db.py

@@ -117,9 +117,7 @@ class DatabaseManager:
                 f"The dirty database file '{self.dirty_file}' does not exist.\n"
                 f"Please run 'manager.initialize()' first to set up the database."
             )
-            raise FileNotFoundError(
-                msg
-            )
+            raise FileNotFoundError(msg)
         conn: sqlite3.Connection = sqlite3.connect(self.dirty_file)
         cursor: sqlite3.Cursor = conn.cursor()
         try:

langgraph_tutorials/customer_support/flights.py

@@ -85,7 +85,9 @@ def update_ticket_to_new_flight(
 
     with DB.get_cursor() as cursor:
         cursor.execute(
-            "SELECT departure_airport, arrival_airport, scheduled_departure FROM flights WHERE flight_id = ?",
+            "SELECT departure_airport, arrival_airport, scheduled_departure "
+            "FROM flights "
+            "WHERE flight_id = ?",
             (new_flight_id,),
         )
         new_flight = cursor.fetchone()

langgraph_tutorials/customer_support/policy.py

@@ -33,9 +33,7 @@ class PolicyRetriever:
             self._model = embedding_model
         else:
             msg = "embedding must be an Embeddings instance or a model name string"
-            raise ValueError(
-                msg
-            )
+            raise TypeError(msg)
 
         self._docs: list[dict] | None = None
         self._arr: np.ndarray | None = None
@@ -81,9 +79,7 @@ class PolicyRetriever:
                 "Retriever is not initialized. "
                 "Please call retriever.initialize() first."
             )
-            raise RuntimeError(
-                msg
-            )
+            raise RuntimeError(msg)
 
         query_vector = np.array(self._model.embed_query(query))
         scores = query_vector @ self._arr.T