vectara-agentic 0.4.6__tar.gz → 0.4.8__tar.gz

{vectara_agentic-0.4.6/vectara_agentic.egg-info → vectara_agentic-0.4.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectara_agentic
-Version: 0.4.6
+Version: 0.4.8
 Summary: A Python package for creating AI Assistants and AI Agents with Vectara
 Home-page: https://github.com/vectara/py-vectara-agentic
 Author: Ofer Mendelevitch
@@ -16,34 +16,35 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: llama-index==0.13.3
-Requires-Dist: llama-index-core==0.13.3
-Requires-Dist: llama-index-workflows==1.3.0
-Requires-Dist: llama-index-cli==0.5.0
-Requires-Dist: llama-index-indices-managed-vectara==0.5.0
-Requires-Dist: llama-index-llms-openai==0.5.4
-Requires-Dist: llama-index-llms-openai-like==0.5.0
-Requires-Dist: llama-index-llms-anthropic==0.8.5
-Requires-Dist: llama-index-llms-together==0.4.0
-Requires-Dist: llama-index-llms-groq==0.4.0
-Requires-Dist: llama-index-llms-cohere==0.6.0
-Requires-Dist: llama-index-llms-google-genai==0.3.0
+Requires-Dist: llama-index==0.14.2
+Requires-Dist: llama-index-core==0.14.2
+Requires-Dist: llama-index-workflows==2.2.2
+Requires-Dist: llama-index-cli==0.5.1
+Requires-Dist: llama-index-indices-managed-vectara==0.5.1
+Requires-Dist: llama-index-llms-openai==0.5.6
+Requires-Dist: llama-index-llms-openai-like==0.5.1
+Requires-Dist: llama-index-llms-anthropic==0.8.6
+Requires-Dist: llama-index-llms-together==0.4.1
+Requires-Dist: llama-index-llms-groq==0.4.1
+Requires-Dist: llama-index-llms-cohere==0.6.1
+Requires-Dist: llama-index-llms-google-genai==0.5.0
+Requires-Dist: llama-index-llms-baseten==0.1.4
 Requires-Dist: google_genai>=1.31.0
-Requires-Dist: llama-index-llms-bedrock-converse==0.8.2
-Requires-Dist: llama-index-tools-yahoo-finance==0.4.0
-Requires-Dist: llama-index-tools-arxiv==0.4.0
-Requires-Dist: llama-index-tools-database==0.4.0
-Requires-Dist: llama-index-tools-google==0.6.0
-Requires-Dist: llama-index-tools-tavily_research==0.4.0
-Requires-Dist: llama_index.tools.brave_search==0.4.0
-Requires-Dist: llama-index-tools-neo4j==0.4.0
-Requires-Dist: llama-index-tools-waii==0.4.0
-Requires-Dist: llama-index-graph-stores-kuzu==0.9.0
-Requires-Dist: llama-index-tools-salesforce==0.4.0
-Requires-Dist: llama-index-tools-slack==0.4.0
-Requires-Dist: llama-index-tools-exa==0.4.0
-Requires-Dist: llama-index-tools-wikipedia==0.4.0
-Requires-Dist: llama-index-tools-bing-search==0.4.0
+Requires-Dist: llama-index-llms-bedrock-converse==0.9.2
+Requires-Dist: llama-index-tools-yahoo-finance==0.4.1
+Requires-Dist: llama-index-tools-arxiv==0.4.1
+Requires-Dist: llama-index-tools-database==0.4.1
+Requires-Dist: llama-index-tools-google==0.6.2
+Requires-Dist: llama-index-tools-tavily_research==0.4.1
+Requires-Dist: llama_index.tools.brave_search==0.4.1
+Requires-Dist: llama-index-tools-neo4j==0.4.1
+Requires-Dist: llama-index-tools-waii==0.4.1
+Requires-Dist: llama-index-graph-stores-kuzu==0.9.1
+Requires-Dist: llama-index-tools-salesforce==0.4.1
+Requires-Dist: llama-index-tools-slack==0.4.1
+Requires-Dist: llama-index-tools-exa==0.4.1
+Requires-Dist: llama-index-tools-wikipedia==0.4.1
+Requires-Dist: llama-index-tools-bing-search==0.4.1
 Requires-Dist: openai>=1.99.3
 Requires-Dist: tavily-python>=0.7.10
 Requires-Dist: exa-py>=1.14.20
@@ -736,13 +737,13 @@ If you want to use `agent`, `tools`, `llm` or `verbose` in other events (that ar
 the `Context` of the Workflow as follows:
 
 ```python
-await ctx.set("agent", ev.agent)
+await ctx.store.set("agent", ev.agent)
 ```
 
 and then in any other event you can pull that agent object with
 
 ```python
-agent = await ctx.get("agent")
+agent = await ctx.store.get("agent")
 ```
 
 Similarly you can reuse the `llm`, `tools` or `verbose` arguments within other nodes in the workflow.
@@ -886,7 +887,7 @@ The `AgentConfig` object may include the following items:
 - `main_llm_provider` and `tool_llm_provider`: the LLM provider for main agent and for the tools. Valid values are `OPENAI`, `ANTHROPIC`, `TOGETHER`, `GROQ`, `COHERE`, `BEDROCK`, `GEMINI` (default: `OPENAI`).
 
 > **Note:** Fireworks AI support has been removed. If you were using Fireworks, please migrate to one of the supported providers listed above.
-- `main_llm_model_name` and `tool_llm_model_name`: agent model name for agent and tools (default depends on provider: OpenAI uses gpt-4.1-mini, Gemini uses gemini-2.5-flash-lite).
+- `main_llm_model_name` and `tool_llm_model_name`: agent model name for agent and tools (default depends on provider: OpenAI uses gpt-4.1-mini, Anthropic uses claude-sonnet-4-0, Gemini uses models/gemini-2.5-flash, Together.AI uses deepseek-ai/DeepSeek-V3, GROQ uses openai/gpt-oss-20b, Bedrock uses us.anthropic.claude-sonnet-4-20250514-v1:0, Cohere uses command-a-03-2025).
 - `observer`: the observer type; should be `ARIZE_PHOENIX` or if undefined no observation framework will be used.
 - `endpoint_api_key`: a secret key if using the API endpoint option (defaults to `dev-api-key`)
 

{vectara_agentic-0.4.6 → vectara_agentic-0.4.8}/README.md

@@ -661,13 +661,13 @@ If you want to use `agent`, `tools`, `llm` or `verbose` in other events (that ar
 the `Context` of the Workflow as follows:
 
 ```python
-await ctx.set("agent", ev.agent)
+await ctx.store.set("agent", ev.agent)
 ```
 
 and then in any other event you can pull that agent object with
 
 ```python
-agent = await ctx.get("agent")
+agent = await ctx.store.get("agent")
 ```
 
 Similarly you can reuse the `llm`, `tools` or `verbose` arguments within other nodes in the workflow.
@@ -811,7 +811,7 @@ The `AgentConfig` object may include the following items:
 - `main_llm_provider` and `tool_llm_provider`: the LLM provider for main agent and for the tools. Valid values are `OPENAI`, `ANTHROPIC`, `TOGETHER`, `GROQ`, `COHERE`, `BEDROCK`, `GEMINI` (default: `OPENAI`).
 
 > **Note:** Fireworks AI support has been removed. If you were using Fireworks, please migrate to one of the supported providers listed above.
-- `main_llm_model_name` and `tool_llm_model_name`: agent model name for agent and tools (default depends on provider: OpenAI uses gpt-4.1-mini, Gemini uses gemini-2.5-flash-lite).
+- `main_llm_model_name` and `tool_llm_model_name`: agent model name for agent and tools (default depends on provider: OpenAI uses gpt-4.1-mini, Anthropic uses claude-sonnet-4-0, Gemini uses models/gemini-2.5-flash, Together.AI uses deepseek-ai/DeepSeek-V3, GROQ uses openai/gpt-oss-20b, Bedrock uses us.anthropic.claude-sonnet-4-20250514-v1:0, Cohere uses command-a-03-2025).
 - `observer`: the observer type; should be `ARIZE_PHOENIX` or if undefined no observation framework will be used.
 - `endpoint_api_key`: a secret key if using the API endpoint option (defaults to `dev-api-key`)
 

vectara_agentic-0.4.8/requirements.txt

@@ -0,0 +1,45 @@
+llama-index==0.14.2
+llama-index-core==0.14.2
+llama-index-workflows==2.2.2
+llama-index-cli==0.5.1
+llama-index-indices-managed-vectara==0.5.1
+llama-index-llms-openai==0.5.6
+llama-index-llms-openai-like==0.5.1
+llama-index-llms-anthropic==0.8.6
+llama-index-llms-together==0.4.1
+llama-index-llms-groq==0.4.1
+llama-index-llms-cohere==0.6.1
+llama-index-llms-google-genai==0.5.0
+llama-index-llms-baseten==0.1.4
+google_genai>=1.31.0
+llama-index-llms-bedrock-converse==0.9.2
+llama-index-tools-yahoo-finance==0.4.1
+llama-index-tools-arxiv==0.4.1
+llama-index-tools-database==0.4.1
+llama-index-tools-google==0.6.2
+llama-index-tools-tavily_research==0.4.1
+llama_index.tools.brave_search==0.4.1
+llama-index-tools-neo4j==0.4.1
+llama-index-tools-waii==0.4.1
+llama-index-graph-stores-kuzu==0.9.1
+llama-index-tools-salesforce==0.4.1
+llama-index-tools-slack==0.4.1
+llama-index-tools-exa==0.4.1
+llama-index-tools-wikipedia==0.4.1
+llama-index-tools-bing-search==0.4.1
+openai>=1.99.3
+tavily-python>=0.7.10
+exa-py>=1.14.20
+openinference-instrumentation-llama-index==4.3.4
+opentelemetry-proto>=1.31.0
+arize-phoenix==10.9.1
+arize-phoenix-otel==0.10.3
+protobuf==5.29.5
+tokenizers>=0.20
+pydantic>=2.11.5
+pandas==2.2.3
+retrying==1.4.2
+python-dotenv==1.0.1
+cloudpickle>=3.1.1
+httpx==0.28.1
+commonmark==0.9.1

vectara_agentic-0.4.8/tests/test_bedrock.py

@@ -0,0 +1,170 @@
+# Suppress external dependency warnings before any other imports
+import warnings
+
+warnings.simplefilter("ignore", DeprecationWarning)
+
+import unittest
+import threading
+
+from vectara_agentic.agent import Agent
+from vectara_agentic.tools import ToolsFactory
+from vectara_agentic.tools_catalog import ToolsCatalog
+
+import nest_asyncio
+
+nest_asyncio.apply()
+
+from conftest import (
+    mult,
+    add,
+    fc_config_bedrock,
+    STANDARD_TEST_TOPIC,
+    STANDARD_TEST_INSTRUCTIONS,
+)
+
+ARIZE_LOCK = threading.Lock()
+
+
+class TestBedrock(unittest.IsolatedAsyncioTestCase):
+
+    async def test_multiturn(self):
+        with ARIZE_LOCK:
+            tools = [ToolsFactory().create_tool(mult)]
+            agent = Agent(
+                tools=tools,
+                topic=STANDARD_TEST_TOPIC,
+                custom_instructions=STANDARD_TEST_INSTRUCTIONS,
+                agent_config=fc_config_bedrock,
+            )
+
+            # First calculation: 5 * 10 = 50
+            stream1 = await agent.astream_chat(
+                "What is 5 times 10. Only give the answer, nothing else"
+            )
+            # Consume the stream
+            async for chunk in stream1.async_response_gen():
+                pass
+            _ = await stream1.aget_response()
+
+            # Second calculation: 3 * 7 = 21
+            stream2 = await agent.astream_chat(
+                "what is 3 times 7. Only give the answer, nothing else"
+            )
+            # Consume the stream
+            async for chunk in stream2.async_response_gen():
+                pass
+            _ = await stream2.aget_response()
+
+            # Final calculation: 50 * 21 = 1050
+            stream3 = await agent.astream_chat(
+                "multiply the results of the last two questions. Output only the answer."
+            )
+            # Consume the stream
+            async for chunk in stream3.async_response_gen():
+                pass
+            response3 = await stream3.aget_response()
+
+            self.assertEqual(response3.response, "1050")
+
+    async def test_claude_sonnet_4_multi_tool_chain(self):
+        """Test Claude Sonnet 4 with complex multi-step reasoning chain using multiple tools via Bedrock."""
+        with ARIZE_LOCK:
+            # Use Bedrock config (Claude Sonnet 4)
+            tools_catalog = ToolsCatalog(fc_config_bedrock)
+            tools = [
+                ToolsFactory().create_tool(mult),
+                ToolsFactory().create_tool(add),
+                ToolsFactory().create_tool(tools_catalog.summarize_text),
+                ToolsFactory().create_tool(tools_catalog.rephrase_text),
+            ]
+
+            agent = Agent(
+                agent_config=fc_config_bedrock,
+                tools=tools,
+                topic=STANDARD_TEST_TOPIC,
+                custom_instructions="You are a mathematical reasoning agent that explains your work step by step.",
+            )
+
+            # Complex multi-step reasoning task
+            complex_query = (
+                "Perform this calculation step by step: "
+                "First multiply 5 by 9, then add 13 to that result, "
+                "then multiply the new result by 2. "
+                "After getting the final number, summarize the entire mathematical process "
+                "with expertise in 'mathematics education', "
+                "then rephrase that summary as a 10-year-old would explain it."
+            )
+
+            print("\n🔍 Starting Claude Sonnet 4 multi-tool chain test (Bedrock)")
+            print(f"📝 Query: {complex_query}")
+            print("🌊 Streaming response:\n" + "="*50)
+
+            stream = await agent.astream_chat(complex_query)
+
+            # Capture streaming deltas and tool calls
+            streaming_deltas = []
+            tool_calls_made = []
+            full_response = ""
+
+            async for chunk in stream.async_response_gen():
+                if chunk and chunk.strip():
+                    streaming_deltas.append(chunk)
+                    full_response += chunk
+                    # Display each streaming delta
+                    print(f"📡 Delta: {repr(chunk)}")
+
+                    # Track tool calls in the stream
+                    if "mult" in chunk.lower():
+                        if "mult" not in [call["tool"] for call in tool_calls_made]:
+                            tool_calls_made.append({"tool": "mult", "order": len(tool_calls_made) + 1})
+                            print(f"🔧 Tool call detected: mult (#{len(tool_calls_made)})")
+                    if "add" in chunk.lower():
+                        if "add" not in [call["tool"] for call in tool_calls_made]:
+                            tool_calls_made.append({"tool": "add", "order": len(tool_calls_made) + 1})
+                            print(f"🔧 Tool call detected: add (#{len(tool_calls_made)})")
+                    if "summarize" in chunk.lower():
+                        if "summarize_text" not in [call["tool"] for call in tool_calls_made]:
+                            tool_calls_made.append({"tool": "summarize_text", "order": len(tool_calls_made) + 1})
+                            print(f"🔧 Tool call detected: summarize_text (#{len(tool_calls_made)})")
+                    if "rephrase" in chunk.lower():
+                        if "rephrase_text" not in [call["tool"] for call in tool_calls_made]:
+                            tool_calls_made.append({"tool": "rephrase_text", "order": len(tool_calls_made) + 1})
+                            print(f"🔧 Tool call detected: rephrase_text (#{len(tool_calls_made)})")
+
+            response = await stream.aget_response()
+
+            print("="*50)
+            print(f"✅ Streaming completed. Total deltas: {len(streaming_deltas)}")
+            print(f"🔧 Tool calls made: {[call['tool'] for call in tool_calls_made]}")
+            print(f"📄 Final response length: {len(response.response)} chars")
+            print(f"🎯 Final response: {response.response}")
+
+            # Validate tool usage sequence
+            tools_used = [call["tool"] for call in tool_calls_made]
+            print(f"🧪 Tools used in order: {tools_used}")
+
+            # Check that at least multiplication happened (basic requirement)
+            self.assertIn("mult", tools_used, f"Expected multiplication tool to be used. Tools used: {tools_used}")
+
+            # Check for mathematical results in the full response or streaming deltas
+            # Expected: 5*9=45, 45+13=58, 58*2=116
+            expected_intermediate_results = ["45", "58", "116"]
+            all_text = (full_response + " " + response.response).lower()
+            math_results_found = sum(1 for result in expected_intermediate_results
+                                     if result in all_text)
+
+            print(f"🔢 Mathematical results found: {math_results_found}/3 expected")
+            print(f"🔍 Full text searched: {all_text[:200]}...")
+
+            # More lenient assertion - just check that some mathematical progress was made
+            self.assertGreaterEqual(math_results_found, 1,
+                                    f"Expected at least 1 mathematical result. Found {math_results_found}. "
+                                    f"Full text: {all_text}")
+
+            # Verify that streaming actually produced content
+            self.assertGreater(len(streaming_deltas), 0, "Expected streaming deltas to be produced")
+            self.assertGreater(len(response.response.strip()), 0, "Expected non-empty final response")
+
+
+if __name__ == "__main__":
+    unittest.main()

vectara_agentic-0.4.8/tests/test_gemini.py

@@ -0,0 +1,121 @@
+# Suppress external dependency warnings before any other imports
+import warnings
+
+warnings.simplefilter("ignore", DeprecationWarning)
+
+import unittest
+
+from vectara_agentic.agent import Agent
+from vectara_agentic.tools import ToolsFactory
+from vectara_agentic.tools_catalog import ToolsCatalog
+
+
+import nest_asyncio
+
+nest_asyncio.apply()
+
+from conftest import (
+    mult,
+    add,
+    fc_config_gemini,
+    STANDARD_TEST_TOPIC,
+    STANDARD_TEST_INSTRUCTIONS,
+)
+
+
+class TestGEMINI(unittest.TestCase):
+    def test_gemini(self):
+        tools = [ToolsFactory().create_tool(mult)]
+
+        agent = Agent(
+            agent_config=fc_config_gemini,
+            tools=tools,
+            topic=STANDARD_TEST_TOPIC,
+            custom_instructions=STANDARD_TEST_INSTRUCTIONS,
+        )
+        _ = agent.chat("What is 5 times 10. Only give the answer, nothing else")
+        _ = agent.chat("what is 3 times 7. Only give the answer, nothing else")
+        res = agent.chat(
+            "what is the result of multiplying the results of the last two multiplications. Only give the answer, nothing else."
+        )
+        self.assertIn("1050", res.response)
+
+    def test_gemini_single_prompt(self):
+        tools = [ToolsFactory().create_tool(mult)]
+
+        agent = Agent(
+            agent_config=fc_config_gemini,
+            tools=tools,
+            topic=STANDARD_TEST_TOPIC,
+            custom_instructions=STANDARD_TEST_INSTRUCTIONS,
+        )
+        res = agent.chat(
+            "First, multiply 5 by 10. Then, multiply 3 by 7. Finally, multiply the results of the first two calculations."
+        )
+        self.assertIn("1050", res.response)
+
+    def test_gemini_25_flash_multi_tool_chain(self):
+        """Test Gemini 2.5 Flash with complex multi-step reasoning chain using multiple tools."""
+        # Use Gemini config (Gemini 2.5 Flash)
+        tools_catalog = ToolsCatalog(fc_config_gemini)
+        tools = [
+            ToolsFactory().create_tool(mult),
+            ToolsFactory().create_tool(add),
+            ToolsFactory().create_tool(tools_catalog.summarize_text),
+            ToolsFactory().create_tool(tools_catalog.rephrase_text),
+        ]
+
+        agent = Agent(
+            agent_config=fc_config_gemini,
+            tools=tools,
+            topic=STANDARD_TEST_TOPIC,
+            custom_instructions="You are a mathematical reasoning agent that explains your work step by step.",
+        )
+
+        # Complex multi-step reasoning task
+        complex_query = (
+            "Perform this calculation step by step: "
+            "First multiply 3 by 8, then add 14 to that result, "
+            "then multiply the new result by 3. "
+            "After getting the final number, summarize the entire mathematical process "
+            "with expertise in 'mathematics education', "
+            "then rephrase that summary as a 10-year-old would explain it."
+        )
+
+        print("\n🔍 Starting Gemini 2.5 Flash multi-tool chain test")
+        print(f"📝 Query: {complex_query}")
+
+        # Note: Gemini tests use synchronous chat, not async streaming
+        response = agent.chat(complex_query)
+
+        print(f"🎯 Final response: {response.response}")
+        print(f"📄 Final response length: {len(response.response)} chars")
+
+        # Check for mathematical results in the response
+        # Expected: 3*8=24, 24+14=38, 38*3=114
+        expected_intermediate_results = ["24", "38", "114"]
+        response_text = response.response.lower()
+        math_results_found = sum(1 for result in expected_intermediate_results
+                                 if result in response_text)
+
+        print(f"🔢 Mathematical results found: {math_results_found}/3 expected")
+        print(f"🔍 Response text searched: {response_text[:200]}...")
+
+        # More lenient assertion - just check that some mathematical progress was made
+        self.assertGreaterEqual(math_results_found, 1,
+                                f"Expected at least 1 mathematical result. Found {math_results_found}. "
+                                f"Response: {response.response}")
+
+        # Verify response has content and mentions math concepts
+        self.assertGreater(len(response.response.strip()), 50, "Expected substantial response content")
+
+        # Check for indications of multi-tool usage (math, summary, or explanation content)
+        multi_tool_indicators = ["calculate", "multiply", "add", "summary", "explain", "mathematical", "process"]
+        indicators_found = sum(1 for indicator in multi_tool_indicators
+                               if indicator in response_text)
+        self.assertGreaterEqual(indicators_found, 3,
+                                f"Expected multiple tool usage indicators. Found {indicators_found}: {response.response}")
+
+
+if __name__ == "__main__":
+    unittest.main()