Merge pull request #18 from jwm4/jwm4-nb1b

feat: Update sample notebook

Author: deanwampler

notebooks/01-responses/nps_mcp_server.py

@@ -23,6 +23,8 @@
 from typing import Optional
 import httpx
 from fastmcp import FastMCP
+from fastmcp.server.dependencies import get_http_request
+from starlette.requests import Request
 import os
 import argparse
 import json
@@ -110,7 +112,7 @@ async def search_parks(
     state_code: Optional[str] = None, 
     park_code: Optional[str] = None,
     query: Optional[str] = None,
-    limit: int = 10
+    limit: int = 10,
 ) -> str:
     """
     Search for national parks by state, park code, or query string.
@@ -126,6 +128,11 @@ async def search_parks(
     """
     # Log input parameters
     get_logger().debug(f"search_parks called with inputs: state_code={state_code}, park_code={park_code}, query={query}, limit={limit}")
+
+    request: Request = get_http_request()
+    headers = dict(request.headers)
+    if headers:
+        get_logger().debug(f"tool call headers: {mask_sensitive_headers(headers)}")
 
     try:
         api_key = get_api_key()

notebooks/01-responses/requirements.txt

@@ -12,6 +12,7 @@ attrs==25.3.0
 Authlib==1.6.1
 autoevals==0.0.129
 braintrust_core==0.0.59
+cachetools==6.2.0
 certifi==2025.7.14
 cffi==1.17.1
 chardet==5.2.0
@@ -53,6 +54,7 @@ ibm-cos-sdk==2.14.2
 ibm-cos-sdk-core==2.14.2
 ibm-cos-sdk-s3transfer==2.14.2
 ibm_watson_machine_learning==1.0.368
+ibm_watsonx_ai==1.3.38
 idna==3.10
 importlib_metadata==8.7.0
 ipykernel==6.30.0
@@ -77,8 +79,8 @@ langchain-text-splitters==0.3.9
 langsmith==0.4.13
 litellm==1.74.9.post1
 llama_api_client==0.1.2
-llama_stack==0.2.17
-llama_stack_client==0.2.17
+llama_stack==0.2.22
+llama_stack_client==0.2.22
 lomond==0.3.3
 markdown-it-py==3.0.0
 MarkupSafe==3.0.2
@@ -92,7 +94,7 @@ multiprocess==0.70.16
 nest-asyncio==1.6.0
 networkx==3.5
 numpy==1.26.4
-openai==1.97.0
+openai==1.108.0
 openapi-pydantic==0.5.1
 opentelemetry-api==1.36.0
 opentelemetry-exporter-otlp==1.36.0
@@ -121,7 +123,7 @@ pyaml==25.7.0
 pyarrow==21.0.0
 pyasn1==0.6.1
 pycparser==2.22
-pydantic==2.11.7
+pydantic==2.11.9
 pydantic-settings==2.10.1
 pydantic_core==2.33.2
 Pygments==2.19.2

notebooks/01-responses/responses-api.ipynb

@@ -106,12 +106,10 @@ providers:
     config:
       openai_api_key: ${env.OPENAI_API_KEY:}
   post_training:
-  - provider_id: huggingface
-    provider_type: inline::huggingface
+  - provider_id: torchtune-cpu
+    provider_type: inline::torchtune-cpu
     config:
-      checkpoint_format: huggingface
-      distributed_backend: null
-      device: cpu
+      checkpoint_format: meta
   tool_runtime:
   - provider_id: brave-search
     provider_type: remote::brave-search
@@ -140,11 +138,6 @@ inference_store:
   type: sqlite
   db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/sample-notebook}/inference_store.db
 models:
-- metadata: {}
-  model_id: watsonx-Llama-3.3-70B-Instruct
-  provider_id: watsonx
-  provider_model_id: meta-llama/llama-3-3-70b-instruct
-  model_type: llm
 - metadata: {}
   model_id: Llama-3.3-70B-Instruct
   provider_model_id: Llama-3.3-70B-Instruct
@@ -160,6 +153,11 @@ models:
   provider_model_id: gpt-4o
   provider_id: openai
   model_type: llm
+- metadata: {}
+  model_id: watsonx-Llama-3.3-70B-Instruct
+  provider_id: watsonx
+  provider_model_id: meta-llama/llama-3-3-70b-instruct
+  model_type: llm
 - metadata:
     embedding_dimension: 768
   model_id: granite-embedding-125m