fix module resolution and config for research agent

Signed-off-by: jsun-m <91754185+jsun-m@users.noreply.github.com>

Author: jsun-m

examples/frameworks/nat_ag2_demo/README.md

@@ -151,7 +151,7 @@ nat run --config_file examples/frameworks/nat_ag2_demo/configs/config-research.y
   --input "What are the latest advances in quantum computing?"
 ```
 
-This config uses the `ag2_research_team` workflow type which wraps the AG2 research agents as a NAT tool, invoked by a `react_agent` orchestrator.
+This config uses the `ag2_research_team` workflow type directly as the workflow, with prompts configured inline.
 
 ## Observability with Phoenix
 

examples/frameworks/nat_ag2_demo/configs/config-research.yml

@@ -25,20 +25,14 @@ llms:
     temperature: 0.0
     max_tokens: 2048
 
-functions:
-  ag2_research:
-    _type: nat_ag2_demo/ag2_research_team
-    llm_name: nim_llm
-    max_rounds: 6
-    researcher_instructions: |
-      You are a research specialist. Investigate the topic thoroughly.
-      Present key facts, data, and sources in a structured format.
-    writer_instructions: |
-      You are a technical writer. Synthesize the researcher's findings into a
-      clear, structured summary with: Key Findings, Analysis, and
-      Recommendations. Keep it under 500 words.
-
 workflow:
-  _type: react_agent
+  _type: nat_ag2_demo/ag2_research_team
   llm_name: nim_llm
-  tool_names: [ag2_research]
+  max_rounds: 6
+  researcher_instructions: |
+    You are a research specialist. Investigate the topic thoroughly.
+    Present key facts, data, and sources in a structured format.
+  writer_instructions: |
+    You are a technical writer. Synthesize the researcher's findings into a
+    clear, structured summary with: Key Findings, Analysis, and
+    Recommendations. Keep it under 500 words.

pyproject.toml

@@ -52,6 +52,7 @@ dependencies = [
 # Keep sorted!!!
 a2a = ["nvidia-nat-a2a == {version}"]
 adk = ["nvidia-nat-adk == {version}"]
+ag2 = ["nvidia-nat-ag2 == {version}"]
 agno = ["nvidia-nat-agno == {version}"]
 app = ["nvidia-nat-app == {version}"]
 autogen = ["nvidia-nat-autogen == {version}"]
@@ -93,6 +94,7 @@ pii-defense = [ "nvidia-nat-core[pii-defense] == {version}" ]
 most = [
   "nvidia-nat-a2a == {version}",
   "nvidia-nat-adk == {version}",
+  "nvidia-nat-ag2 == {version}",
   "nvidia-nat-agno == {version}",
   "nvidia-nat-app == {version}",
   "nvidia-nat-autogen == {version}",
@@ -129,6 +131,7 @@ examples = [
   "nat_agno_personal_finance",
   "nat_agents_examples",
   "nat_alert_triage_agent",
+  "nat_ag2_demo",
   "nat_autogen_demo",
   "nat_automated_description_generation",
   "nat_currency_agent_a2a",
@@ -230,6 +233,7 @@ managed = true
 # Packages
 nvidia-nat-a2a = { path = "packages/nvidia_nat_a2a", editable = true }
 nvidia-nat-adk = { path = "packages/nvidia_nat_adk", editable = true }
+nvidia-nat-ag2 = { path = "packages/nvidia_nat_ag2", editable = true }
 nvidia-nat-agno = { path = "packages/nvidia_nat_agno", editable = true }
 nvidia-nat-app = { path = "packages/nvidia_nat_app", editable = true }
 nvidia-nat-autogen = { path = "packages/nvidia_nat_autogen", editable = true }
@@ -268,6 +272,7 @@ nat_adk_demo = { path = "examples/frameworks/adk_demo", editable = true }
 nat_agno_personal_finance = { path = "examples/frameworks/agno_personal_finance", editable = true }
 nat_agents_examples = { path = "examples/agents", editable = true }
 nat_alert_triage_agent = { path = "examples/advanced_agents/alert_triage_agent", editable = true }
+nat_ag2_demo = { path = "examples/frameworks/nat_ag2_demo", editable = true }
 nat_autogen_demo = { path = "examples/frameworks/nat_autogen_demo", editable = true }
 nat_automated_description_generation = { path = "examples/custom_functions/automated_description_generation", editable = true }
 nat_currency_agent_a2a = { path = "examples/A2A/currency_agent_a2a", editable = true }