vectara-agentic 0.2.18__py3-none-any.whl → 0.2.20__py3-none-any.whl

tests/test_agent.py

@@ -11,6 +11,7 @@ from vectara_agentic._prompts import GENERAL_INSTRUCTIONS
 
 
 def mult(x: float, y: float) -> float:
+    "Multiply two numbers"
     return x * y
 
 

tests/test_bedrock.py

@@ -0,0 +1,42 @@
+import unittest
+
+from vectara_agentic.agent import Agent, AgentType
+from vectara_agentic.agent_config import AgentConfig
+from vectara_agentic.tools import ToolsFactory
+from vectara_agentic.types import ModelProvider
+
+import nest_asyncio
+nest_asyncio.apply()
+
+
+def mult(x: float, y: float) -> float:
+    "Multiply two numbers"
+    return x * y
+
+
+fc_config_bedrock = AgentConfig(
+    agent_type=AgentType.FUNCTION_CALLING,
+    main_llm_provider=ModelProvider.BEDROCK,
+    tool_llm_provider=ModelProvider.BEDROCK,
+)
+
+class TestBedrock(unittest.TestCase):
+
+    def test_multiturn(self):
+        tools = [ToolsFactory().create_tool(mult)]
+        topic = "AI topic"
+        instructions = "Always do as your father tells you, if your mother agrees!"
+        agent = Agent(
+            tools=tools,
+            topic=topic,
+            custom_instructions=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("multiply the results of the last two questions. Output only the answer.")
+        self.assertEqual(res.response, "1050")
+
+
+if __name__ == "__main__":
+    unittest.main()

vectara_agentic/_version.py

@@ -1,4 +1,4 @@
 """
 Define the version of the package.
 """
-__version__ = "0.2.18"
+__version__ = "0.2.20"

vectara_agentic/llm_utils.py

@@ -1,7 +1,9 @@
 """
 Utilities for the Vectara agentic.
 """
+
 from typing import Tuple, Callable, Optional
+import os
 from functools import lru_cache
 import tiktoken
 
@@ -14,11 +16,11 @@ from .agent_config import AgentConfig
 
 provider_to_default_model_name = {
     ModelProvider.OPENAI: "gpt-4o",
-    ModelProvider.ANTHROPIC: "claude-3-7-sonnet-latest",
+    ModelProvider.ANTHROPIC: "claude-sonnet-4-20250514",
     ModelProvider.TOGETHER: "Qwen/Qwen2.5-72B-Instruct-Turbo",
     ModelProvider.GROQ: "meta-llama/llama-4-scout-17b-16e-instruct",
     ModelProvider.FIREWORKS: "accounts/fireworks/models/firefunction-v2",
-    ModelProvider.BEDROCK: "anthropic.claude-3-7-sonnet-20250219-v1:0",
+    ModelProvider.BEDROCK: "us.anthropic.claude-sonnet-4-20250514-v1:0",
     ModelProvider.COHERE: "command-a-03-2025",
     ModelProvider.GEMINI: "models/gemini-2.0-flash",
 }
@@ -136,9 +138,18 @@ def get_llm(role: LLMRole, config: Optional[AgentConfig] = None) -> LLM:
 
         llm = Fireworks(model=model_name, temperature=0, max_tokens=max_tokens)
     elif model_provider == ModelProvider.BEDROCK:
-        from llama_index.llms.bedrock import Bedrock
+        from llama_index.llms.bedrock_converse import BedrockConverse
+
+        aws_profile_name = os.getenv("AWS_PROFILE", None)
+        aws_region = os.getenv("AWS_REGION", "us-east-2")
 
-        llm = Bedrock(model=model_name, temperature=0, max_tokens=max_tokens)
+        llm = BedrockConverse(
+            model=model_name,
+            temperature=0,
+            max_tokens=max_tokens,
+            profile_name=aws_profile_name,
+            region_name=aws_region,
+        )
     elif model_provider == ModelProvider.COHERE:
         from llama_index.llms.cohere import Cohere
 

{vectara_agentic-0.2.18.dist-info → vectara_agentic-0.2.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectara_agentic
-Version: 0.2.18
+Version: 0.2.20
 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,20 +16,21 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: llama-index==0.12.35
+Requires-Dist: llama-index==0.12.37
+Requires-Dist: llama-index-core==0.12.37
 Requires-Dist: llama-index-indices-managed-vectara==0.4.5
-Requires-Dist: llama-index-agent-llm-compiler==0.3.0
+Requires-Dist: llama-index-agent-llm-compiler==0.3.1
 Requires-Dist: llama-index-agent-lats==0.3.0
 Requires-Dist: llama-index-agent-openai==0.4.8
-Requires-Dist: llama-index-llms-openai==0.3.42
+Requires-Dist: llama-index-llms-openai==0.3.44
 Requires-Dist: llama-index-llms-openai-like>=0.3.5
-Requires-Dist: llama-index-llms-anthropic==0.6.10
+Requires-Dist: llama-index-llms-anthropic==0.6.19
 Requires-Dist: llama-index-llms-together==0.3.1
 Requires-Dist: llama-index-llms-groq==0.3.1
 Requires-Dist: llama-index-llms-fireworks==0.3.2
 Requires-Dist: llama-index-llms-cohere==0.4.1
-Requires-Dist: llama-index-llms-google-genai==0.1.12
-Requires-Dist: llama-index-llms-bedrock==0.3.8
+Requires-Dist: llama-index-llms-google-genai==0.1.14
+Requires-Dist: llama-index-llms-bedrock-converse==0.6.0
 Requires-Dist: llama-index-tools-yahoo-finance==0.3.0
 Requires-Dist: llama-index-tools-arxiv==0.3.0
 Requires-Dist: llama-index-tools-database==0.3.0
@@ -44,8 +45,9 @@ Requires-Dist: llama-index-tools-slack==0.3.0
 Requires-Dist: llama-index-tools-exa==0.3.0
 Requires-Dist: llama-index-tools-wikipedia==0.3.0
 Requires-Dist: llama-index-tools-bing-search==0.3.0
-Requires-Dist: tavily-python==0.7.2
-Requires-Dist: exa-py==1.12.1
+Requires-Dist: openai>=1.82.0
+Requires-Dist: tavily-python==0.7.3
+Requires-Dist: exa-py==1.13.1
 Requires-Dist: openinference-instrumentation-llama-index==4.2.1
 Requires-Dist: opentelemetry-proto>=1.31.0
 Requires-Dist: arize-phoenix==8.26.1
@@ -144,7 +146,11 @@ Check out our example AI assistants:
 - [Vectara account](https://console.vectara.com/signup/?utm_source=github&utm_medium=code&utm_term=DevRel&utm_content=vectara-agentic&utm_campaign=github-code-DevRel-vectara-agentic)
 - A Vectara corpus with an [API key](https://docs.vectara.com/docs/api-keys)
 - [Python 3.10 or higher](https://www.python.org/downloads/)
-- OpenAI API key (or API keys for Anthropic, TOGETHER.AI, Fireworks AI, Bedrock, Cohere, GEMINI or GROQ, if you choose to use them)
+- OpenAI API key (or API keys for Anthropic, TOGETHER.AI, Fireworks AI, Cohere, GEMINI or GROQ, if you choose to use them).
+  To use AWS Bedrock, make sure that
+  * The Bedrock models you need are enabled on your account
+  * Your environment includes `AWS_PROFILE` with your AWS profile name.
+  * Your environment includes `AWS_REGION` set to the region where you want to consume the AWS Bedrock services (defaults to us-east-2)
 
 ### Installation
 

{vectara_agentic-0.2.18.dist-info → vectara_agentic-0.2.20.dist-info}/RECORD

@@ -1,9 +1,10 @@
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/endpoint.py,sha256=frnpdZQpnuQNNKNYgAn2rFTarNG8MCJaNA77Bw_W22A,1420
-tests/test_agent.py,sha256=o5U3K1AJllsSDvucrgFJPQRdAmHPq3LCuFpsnECUTFk,5483
+tests/test_agent.py,sha256=BmpJrYN-BLwYasgPhY5Dji-kIpc723e2F6I-nj4EQgc,5510
 tests/test_agent_planning.py,sha256=JwEebGooROAvsQ9JZoaH6KEcrSyv1F0lL4TD4FjP8a8,2213
 tests/test_agent_type.py,sha256=mWo-pTQNDj4fWFPETm5jnb7Y5N48aW35keTVvxdIaCc,7173
 tests/test_api_endpoint.py,sha256=M9YGFCy_Jphzq9JznP4ftHqxZ_yu6dgWdX1jRvdsORA,5002
+tests/test_bedrock.py,sha256=23A5_2FRaEl47PzgfUPDL6dgFDPJ9iktz-A-B1UwIhg,1224
 tests/test_fallback.py,sha256=M5YD7NHZ0joVU1frYIr9_OiRAIje5mrXrYVcekzlyGs,2829
 tests/test_gemini.py,sha256=QUBYWhZkX9AjnhPn5qa7sREf6YHZWeJEmYzKwVC23Io,4081
 tests/test_groq.py,sha256=5RA6uFC6qra-Do55f6HUotk3EQqOosw0GjOGiHDBS4o,4071
@@ -17,20 +18,20 @@ vectara_agentic/__init__.py,sha256=2GLDS3U6KckK-dBRl9v_x1kSV507gEhjOfuMmmu0Qxg,8
 vectara_agentic/_callback.py,sha256=ron49t1t-ox-736WaXzrZ99vhN4NI9bMiHFyj0iIPqg,13062
 vectara_agentic/_observability.py,sha256=UbJxiOJFOdLq3b1t0-Y7swMC3BzJu3IOlTUM-c1oUk8,4328
 vectara_agentic/_prompts.py,sha256=vAb02oahA7GKRgLOsDGqgKl-BLBop2AjOlCTgLrf3M4,9694
-vectara_agentic/_version.py,sha256=6bWhPhOhATgGaKBsmcgPdRKvZBguw3zZOhD6CJaNJPs,66
+vectara_agentic/_version.py,sha256=UFpxkXKeHSevaFea4W2brcejYQjkcb-LSSVAYobX5C0,66
 vectara_agentic/agent.py,sha256=zJ7ucFf8jc0VO4mTFqujfwREz2B-rJCpIgCJKAtNlEk,54884
 vectara_agentic/agent_config.py,sha256=E-rtYMcpoGxnEAyy8231bizo2n0uGQ2qWxuSgTEfwdQ,4327
 vectara_agentic/agent_endpoint.py,sha256=PzIN7HhEHv8Mq_Zo5cZ2xYrgdv2AN6kx6dc_2AJq28I,7497
 vectara_agentic/db_tools.py,sha256=Kfz6n-rSj5TQEbAiJnWGmqWtcwB0A5GpxD7d1UwGzlc,11194
-vectara_agentic/llm_utils.py,sha256=_dkxA9DcBwyIzg-BOTi7NwZZhhjV5G2cnaVlzd9J7do,5687
+vectara_agentic/llm_utils.py,sha256=K-rtHq4iNQiT2Jqp_gIJdbFtCheawe_3mvA6mr8JczY,5967
 vectara_agentic/sub_query_workflow.py,sha256=cPeossVPFajpSAwy45fSXhTXbQOfzv_l66pxSa4molM,12366
 vectara_agentic/tool_utils.py,sha256=jv98vCMYb9afFa-HaPxI2A8BXxplfQRv2Z9b5w7ztZc,18919
 vectara_agentic/tools.py,sha256=2_9YBqszFqYDpvlTIZfdfplRKffe660jQRxp0akM-cE,32918
 vectara_agentic/tools_catalog.py,sha256=cAN_kDOWZUoW4GNFwY5GdS6ImMUQNnF2sggx9OGK9Cg,4906
 vectara_agentic/types.py,sha256=HcS7vR8P2v2xQTlOc6ZFV2vvlr3OpzSNWhtcLMxqUZc,1792
 vectara_agentic/utils.py,sha256=R9HitEG5K3Q_p2M_teosT181OUxkhs1-hnj98qDYGbE,2545
-vectara_agentic-0.2.18.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vectara_agentic-0.2.18.dist-info/METADATA,sha256=FO_4pmBWrl_-7DWPWF32HwOHbbicHdB86YvRi4cA67Y,29946
-vectara_agentic-0.2.18.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-vectara_agentic-0.2.18.dist-info/top_level.txt,sha256=Y7TQTFdOYGYodQRltUGRieZKIYuzeZj2kHqAUpfCUfg,22
-vectara_agentic-0.2.18.dist-info/RECORD,,
+vectara_agentic-0.2.20.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vectara_agentic-0.2.20.dist-info/METADATA,sha256=Q4SUWtoG_rVZ6mfPFexFq8YYMVV-3G1Aa2qoegJkmjs,30323
+vectara_agentic-0.2.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vectara_agentic-0.2.20.dist-info/top_level.txt,sha256=Y7TQTFdOYGYodQRltUGRieZKIYuzeZj2kHqAUpfCUfg,22
+vectara_agentic-0.2.20.dist-info/RECORD,,

{vectara_agentic-0.2.18.dist-info → vectara_agentic-0.2.20.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any