update dependencies and rely on LLAMA_TOOLCHAIN_DIR for dev purposes

llama_toolchain/agentic_system/providers.py

@@ -16,6 +16,7 @@ def available_agentic_system_providers() -> List[ProviderSpec]:
             provider_id="meta-reference",
             pip_packages=[
                 "codeshield",
+                "pillow",
                 "torch",
                 "transformers",
             ],

llama_toolchain/distribution/install_distribution.sh

@@ -54,10 +54,14 @@ ensure_conda_env_python310() {
   conda deactivate && conda activate "${env_name}"
 
   # Re-installing llama-toolchain in the new conda environment
-  if git rev-parse --is-inside-work-tree &>/dev/null; then
+  if [ -n "$LLAMA_TOOLCHAIN_DIR" ]; then
-    repo_root=$(git rev-parse --show-toplevel)
+    if [ ! -d "$LLAMA_TOOLCHAIN_DIR" ]; then
-    cd "$repo_root"
+      echo -e "${RED}Warning: LLAMA_TOOLCHAIN_DIR is set but directory does not exist: $LLAMA_TOOLCHAIN_DIR${NC}" >&2
-    pip install -e .
+      exit 1
+    fi
+
+    echo "Installing from LLAMA_TOOLCHAIN_DIR: $LLAMA_TOOLCHAIN_DIR"
+    pip install -e "$LLAMA_TOOLCHAIN_DIR"
   else
     pip install llama-toolchain
   fi

llama_toolchain/inference/ollama/ollama.py

@@ -66,10 +66,7 @@ class OllamaInference(Inference):
         try:
             await self.client.ps()
         except httpx.ConnectError:
-            print(
+            raise RuntimeError("Ollama Server is not running, start it using `ollama serve` in a separate terminal")
-                "Ollama Server is not running, start it using `ollama serve` in a separate terminal"
-            )
-            raise
 
     async def shutdown(self) -> None:
         pass

llama_toolchain/safety/providers.py

@@ -15,6 +15,7 @@ def available_safety_providers() -> List[ProviderSpec]:
             api=Api.safety,
             provider_id="meta-reference",
             pip_packages=[
+                "accelerate",
                 "codeshield",
                 "torch",
                 "transformers",