remove unused modules

databricks · Jan 5, 2025 · 2aef098 · 2aef098
1 parent 78f49b5
commit 2aef098
Show file tree

Hide file tree

Showing 18 changed files with 1 addition and 2,173 deletions.
diff --git a/langgraph_agent_app_sample_code/03_create_tools.ipynb b/langgraph_agent_app_sample_code/03_create_tools.ipynb
diff --git a/langgraph_agent_app_sample_code/04_tool_calling_agent.ipynb b/langgraph_agent_app_sample_code/04_tool_calling_agent.ipynb
@@ -250,7 +250,6 @@
     "import mlflow\n",
     "from mlflow.types.llm import CHAT_MODEL_INPUT_SCHEMA\n",
     "from mlflow.models.rag_signatures import StringResponse, ChatCompletionResponse\n",
-    "from cookbook.agents.utils.signatures import STRING_RESPONSE_WITH_MESSAGES\n",
     "from mlflow.models.signature import ModelSignature\n",
     "\n",
     "\n",

diff --git a/langgraph_agent_app_sample_code/cookbook/agents/utils/chat.py b/langgraph_agent_app_sample_code/cookbook/agents/utils/chat.py
diff --git a/langgraph_agent_app_sample_code/cookbook/agents/utils/execute_function.py b/langgraph_agent_app_sample_code/cookbook/agents/utils/execute_function.py
diff --git a/langgraph_agent_app_sample_code/cookbook/agents/utils/load_config.py b/langgraph_agent_app_sample_code/cookbook/agents/utils/load_config.py
@@ -17,6 +17,7 @@ def load_config():
         agent_conf = Box(yaml.safe_load(Path(fc_agent_config_path).read_text()))
         return agent_conf
     except FileNotFoundError as e:
+        logging.info(f"Could not load config from {fc_agent_config_path}: {e}")
         return load_config_from_mlflow_model_config()
 
 def load_config_from_mlflow_model_config():

diff --git a/langgraph_agent_app_sample_code/cookbook/agents/utils/playground_parser.py b/langgraph_agent_app_sample_code/cookbook/agents/utils/playground_parser.py
diff --git a/langgraph_agent_app_sample_code/cookbook/agents/utils/signatures.py b/langgraph_agent_app_sample_code/cookbook/agents/utils/signatures.py
diff --git a/langgraph_agent_app_sample_code/cookbook/databricks_utils/agent_evaluation/__init__.py b/langgraph_agent_app_sample_code/cookbook/databricks_utils/agent_evaluation/__init__.py