vectara-agentic 0.2.14__tar.gz → 0.2.16__tar.gz

{vectara_agentic-0.2.14/vectara_agentic.egg-info → vectara_agentic-0.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectara_agentic
-Version: 0.2.14
+Version: 0.2.16
 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,18 +16,18 @@ 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.33
-Requires-Dist: llama-index-indices-managed-vectara==0.4.4
+Requires-Dist: llama-index==0.12.34
+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-lats==0.3.0
-Requires-Dist: llama-index-agent-openai==0.4.6
+Requires-Dist: llama-index-agent-openai==0.4.7
 Requires-Dist: llama-index-llms-openai==0.3.38
 Requires-Dist: llama-index-llms-anthropic==0.6.10
 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.8
+Requires-Dist: llama-index-llms-google-genai==0.1.12
 Requires-Dist: llama-index-llms-bedrock==0.3.8
 Requires-Dist: llama-index-tools-yahoo-finance==0.3.0
 Requires-Dist: llama-index-tools-arxiv==0.3.0

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/requirements.txt

@@ -1,15 +1,15 @@
-llama-index==0.12.33
-llama-index-indices-managed-vectara==0.4.4
+llama-index==0.12.34
+llama-index-indices-managed-vectara==0.4.5
 llama-index-agent-llm-compiler==0.3.0
 llama-index-agent-lats==0.3.0
-llama-index-agent-openai==0.4.6
+llama-index-agent-openai==0.4.7
 llama-index-llms-openai==0.3.38
 llama-index-llms-anthropic==0.6.10
 llama-index-llms-together==0.3.1
 llama-index-llms-groq==0.3.1
 llama-index-llms-fireworks==0.3.2
 llama-index-llms-cohere==0.4.1
-llama-index-llms-google-genai ==0.1.8
+llama-index-llms-google-genai ==0.1.12
 llama-index-llms-bedrock==0.3.8
 llama-index-tools-yahoo-finance==0.3.0
 llama-index-tools-arxiv==0.3.0

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/tests/test_agent.py

@@ -124,7 +124,7 @@ class TestAgentPackage(unittest.TestCase):
         self.assertEqual(res.response, "1050")
 
     def test_custom_general_instruction(self):
-        general_instructions = "Always respond with 'I DIDNT DO IT'"
+        general_instructions = "Always respond with: I DIDN'T DO IT"
         agent = Agent.from_corpus(
             tool_name="RAG Tool",
             vectara_corpus_key="corpus_key",
@@ -135,7 +135,7 @@ class TestAgentPackage(unittest.TestCase):
         )
 
         res = agent.chat("What is the meaning of the universe?")
-        self.assertEqual(res.response, "I DIDNT DO IT")
+        self.assertEqual(res.response, "I DIDN'T DO IT")
 
 
 if __name__ == "__main__":

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/tests/test_agent_planning.py

@@ -4,7 +4,7 @@ from vectara_agentic.agent_config import AgentConfig
 from vectara_agentic.agent import Agent
 from vectara_agentic.tools import VectaraToolFactory
 
-# SETUP speical test account credentials for vectara
+# SETUP special test account credentials for vectara
 # It's okay to expose these credentials in the test code
 vectara_corpus_key = "vectara-docs_1"
 vectara_api_key = 'zqt_UXrBcnI2UXINZkrv4g1tQPhzj02vfdtqYJIDiA'

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/tests/test_groq.py

@@ -113,7 +113,9 @@ class TestGROQ(unittest.TestCase):
             agent_config=fc_config_groq,
         )
         res = agent.chat("What is the stock price?")
-        self.assertIn("I don't know", str(res))
+        self.assertTrue(
+            any(sub in str(res) for sub in ["I don't know", "I do not have"])
+        )
 
 
 if __name__ == "__main__":

vectara_agentic-0.2.16/tests/test_return_direct.py

@@ -0,0 +1,49 @@
+import unittest
+
+from vectara_agentic.agent import Agent
+from vectara_agentic.tools import VectaraToolFactory
+
+vectara_corpus_key = "vectara-docs_1"
+vectara_api_key = "zqt_UXrBcnI2UXINZkrv4g1tQPhzj02vfdtqYJIDiA"
+
+
+class TestAgentPackage(unittest.TestCase):
+
+    def test_return_direct1(self):
+        vec_factory = VectaraToolFactory(vectara_corpus_key, vectara_api_key)
+
+        query_tool = vec_factory.create_rag_tool(
+            tool_name="rag_tool",
+            tool_description="""
+            A dummy tool for testing return_direct.
+            """,
+            return_direct=True,
+        )
+
+        agent = Agent(
+            tools=[query_tool],
+            topic="Sample topic",
+            custom_instructions="You are a helpful assistant.",
+        )
+        res = agent.chat("What is Vectara?")
+        self.assertIn("Response:", str(res))
+        self.assertIn("fcs_score", str(res))
+        self.assertIn("References:", str(res))
+
+    def test_from_corpus(self):
+        agent = Agent.from_corpus(
+            tool_name="rag_tool",
+            vectara_corpus_key=vectara_corpus_key,
+            vectara_api_key=vectara_api_key,
+            data_description="stuff about Vectara",
+            assistant_specialty="question answering",
+            return_direct=True,
+        )
+        res = agent.chat("What is Vectara?")
+        self.assertIn("Response:", str(res))
+        self.assertIn("fcs_score", str(res))
+        self.assertIn("References:", str(res))
+
+
+if __name__ == "__main__":
+    unittest.main()

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/tests/test_tools.py

@@ -1,6 +1,7 @@
 import unittest
 from pydantic import Field, BaseModel
-
+from unittest.mock import patch, MagicMock
+import requests
 from vectara_agentic.tools import (
     VectaraTool,
     VectaraToolFactory,
@@ -17,6 +18,7 @@ from llama_index.core.tools import FunctionTool
 vectara_corpus_key = "vectara-docs_1"
 vectara_api_key = "zqt_UXrBcnI2UXINZkrv4g1tQPhzj02vfdtqYJIDiA"
 
+from typing import Optional
 
 class TestToolsPackage(unittest.TestCase):
 
@@ -89,7 +91,7 @@ class TestToolsPackage(unittest.TestCase):
                 description="The ticker symbol for the company",
                 examples=["AAPL", "GOOG"],
             )
-            year: int | str = Field(
+            year: Optional[int | str] = Field(
                 default=None,
                 description="The year this query relates to. An integer between 2015 and 2024 or a string specifying a condition on the year",
                 examples=[
@@ -109,6 +111,7 @@ class TestToolsPackage(unittest.TestCase):
             tool_args_schema=QueryToolArgs,
         )
 
+        # test an invalid argument name
         res = query_tool(
             query="What is the stock price?",
             the_year=2023,
@@ -126,6 +129,86 @@ class TestToolsPackage(unittest.TestCase):
         )
         self.assertIn("got an unexpected keyword argument 'the_year'", str(res))
 
+    @patch.object(requests.Session, "post")
+    def test_vectara_tool_ranges(self, mock_post):
+        # Configure the mock to return a dummy response.
+        response_text = "ALL GOOD"
+        mock_response = MagicMock()
+        mock_response.status_code = 200
+        mock_response.json.return_value = {
+            'summary': response_text,
+            'search_results': [
+                {'text': 'ALL GOOD', 'document_id': '12345', 'score': 0.9},
+            ]
+        }
+        mock_post.return_value = mock_response
+
+        vec_factory = VectaraToolFactory(vectara_corpus_key, vectara_api_key)
+
+        class QueryToolArgs(BaseModel):
+            ticker: str = Field(
+                description="The ticker symbol for the company",
+                examples=["AAPL", "GOOG"],
+            )
+            year: int | str = Field(
+                default=None,
+                description="The year this query relates to. An integer between 2015 and 2024 or a string specifying a condition on the year",
+                examples=[
+                    2020,
+                    ">2021",
+                    "<2023",
+                    ">=2021",
+                    "<=2023",
+                    "[2021, 2023]",
+                    "[2021, 2023)",
+                ],
+            )
+
+        query_tool = vec_factory.create_rag_tool(
+            tool_name="rag_tool",
+            tool_description="Returns a response (str) to the user query based on the data in this corpus.",
+            tool_args_schema=QueryToolArgs,
+        )
+
+        # test an invalid argument name
+        res = query_tool(
+            query="What is the stock price?",
+            year=">2023"
+        )
+        self.assertIn(response_text, str(res))
+
+        # Test a valid range
+        res = query_tool(
+            query="What is the stock price?",
+            year="[2021, 2023]",
+        )
+        self.assertIn(response_text, str(res))
+
+        # Test a valid half closed range
+        res = query_tool(
+            query="What is the stock price?",
+            year="[2020, 2023)",
+        )
+        self.assertIn(response_text, str(res))
+
+        # Test an operator
+        res = query_tool(
+            query="What is the stock price?",
+            year=">2022",
+        )
+        self.assertIn(response_text, str(res))
+
+        search_tool = vec_factory.create_search_tool(
+            tool_name="search_tool",
+            tool_description="Returns a list of documents (str) that match the user query.",
+            tool_args_schema=QueryToolArgs,
+        )
+        res = search_tool(
+            query="What is the stock price?",
+            the_year=2023,
+        )
+        self.assertIn("got an unexpected keyword argument 'the_year'", str(res))
+
     def test_tool_factory(self):
         def mult(x: float, y: float) -> float:
             return x * y
@@ -152,8 +235,6 @@ class TestToolsPackage(unittest.TestCase):
             self.assertEqual(tool.metadata.tool_type, ToolType.QUERY)
 
     def test_tool_with_many_arguments(self):
-        vectara_corpus_key = "corpus_key"
-        vectara_api_key = "api_key"
         vec_factory = VectaraToolFactory(vectara_corpus_key, vectara_api_key)
 
         class QueryToolArgs(BaseModel):
@@ -169,24 +250,18 @@ class TestToolsPackage(unittest.TestCase):
             arg10: str = Field(description="the tenth argument", examples=["val10"])
             arg11: str = Field(description="the eleventh argument", examples=["val11"])
             arg12: str = Field(description="the twelfth argument", examples=["val12"])
-            arg13: str = Field(
-                description="the thirteenth argument", examples=["val13"]
-            )
-            arg14: str = Field(
-                description="the fourteenth argument", examples=["val14"]
-            )
-            arg15: str = Field(description="the fifteenth argument", examples=["val15"])
+            arg13: str = Field(description="the thirteenth argument", examples=["val13"])
 
         query_tool_1 = vec_factory.create_rag_tool(
             tool_name="rag_tool",
             tool_description="""
-            A dummy tool that takes 15 arguments and returns a response (str) to the user query based on the data in this corpus.
+            A dummy tool that takes 13 arguments and returns a response (str) to the user query based on the data in this corpus.
             We are using this tool to test the tool factory works and does not crash with OpenAI.
             """,
             tool_args_schema=QueryToolArgs,
         )
 
-        # Test with 15 arguments which go over the 1024 limit.
+        # Test with 13 arguments which go over the 1024 limit.
         config = AgentConfig(
             agent_type=AgentType.OPENAI
         )
@@ -196,7 +271,7 @@ class TestToolsPackage(unittest.TestCase):
             custom_instructions="Call the tool with 15 arguments for OPENAI",
             agent_config=config,
         )
-        res = agent.chat("What is the stock price?")
+        res = agent.chat("What is the stock price for Yahoo on 12/31/22?")
         self.assertIn("maximum length of 1024 characters", str(res))
 
         # Same test but with GROQ
@@ -208,7 +283,7 @@ class TestToolsPackage(unittest.TestCase):
         agent = Agent(
             tools=[query_tool_1],
             topic="Sample topic",
-            custom_instructions="Call the tool with 15 arguments for GROQ",
+            custom_instructions="Call the tool with 13 arguments for GROQ",
             agent_config=config,
         )
         res = agent.chat("What is the stock price?")
@@ -223,14 +298,14 @@ class TestToolsPackage(unittest.TestCase):
         agent = Agent(
             tools=[query_tool_1],
             topic="Sample topic",
-            custom_instructions="Call the tool with 15 arguments for ANTHROPIC",
+            custom_instructions="Call the tool with 13 arguments for ANTHROPIC",
             agent_config=config,
         )
         res = agent.chat("What is the stock price?")
         # ANTHROPIC does not have that 1024 limit
         self.assertIn("stock price", str(res))
 
-        # But using Compact_docstring=True, we can pass 15 arguments successfully.
+        # But using Compact_docstring=True, we can pass 13 arguments successfully.
         vec_factory = VectaraToolFactory(
             vectara_corpus_key, vectara_api_key, compact_docstring=True
         )
@@ -251,7 +326,9 @@ class TestToolsPackage(unittest.TestCase):
             agent_config=config,
         )
         res = agent.chat("What is the stock price?")
-        self.assertIn("stock price", str(res))
+        self.assertTrue(
+            any(sub in str(res) for sub in ["I don't know", "stock price"])
+        )
 
     def test_public_repo(self):
         vectara_corpus_key = "vectara-docs_1"
@@ -297,6 +374,31 @@ class TestToolsPackage(unittest.TestCase):
             "50",
         )
 
+    def test_vectara_tool_docstring(self):
+        class DummyArgs(BaseModel):
+            foo: int = Field(..., description="how many foos", examples=[1, 2, 3])
+            bar: str = Field(
+                "baz",
+                description="what bar to use",
+                examples=["x", "y"],
+            )
+
+        vec_factory = VectaraToolFactory(vectara_corpus_key, vectara_api_key)
+        dummy_tool = vec_factory.create_rag_tool(
+            tool_name="dummy_tool",
+            tool_description="A dummy tool.",
+            tool_args_schema=DummyArgs,
+        )
+
+        doc = dummy_tool.metadata.description
+        self.assertTrue(doc.startswith("dummy_tool(query: str, foo: int, bar: str) -> dict[str, Any]"))
+        self.assertIn("Args:", doc)
+        self.assertIn("query (str): The search query to perform, in the form of a question", doc)
+        self.assertIn("foo (int): how many foos (e.g., 1, 2, 3)", doc)
+        self.assertIn("bar (str, default='baz'): what bar to use (e.g., 'x', 'y')", doc)
+        self.assertIn("Returns:", doc)
+        self.assertIn("dict[str, Any]: A dictionary containing the result data.", doc)
+
 
 if __name__ == "__main__":
     unittest.main()

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/tests/test_vectara_llms.py

@@ -15,7 +15,10 @@ vectara_api_key = "zqt_UXrBcnI2UXINZkrv4g1tQPhzj02vfdtqYJIDiA"
 class TestLLMPackage(unittest.TestCase):
 
     def test_vectara_openai(self):
-        vec_factory = VectaraToolFactory(vectara_corpus_key, vectara_api_key)
+        vec_factory = VectaraToolFactory(
+            vectara_corpus_key=vectara_corpus_key,
+            vectara_api_key=vectara_api_key
+        )
 
         self.assertEqual(vectara_corpus_key, vec_factory.vectara_corpus_key)
         self.assertEqual(vectara_api_key, vec_factory.vectara_api_key)

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/vectara_agentic/_observability.py

@@ -12,28 +12,50 @@ def setup_observer(config: AgentConfig, verbose: bool) -> bool:
     '''
     Setup the observer.
     '''
-    import phoenix as px
-    from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
-    from phoenix.otel import register
-    if config.observer == ObserverType.ARIZE_PHOENIX:
-        phoenix_endpoint = os.getenv("PHOENIX_ENDPOINT", None)
-        if not phoenix_endpoint:
-            px.launch_app()
-            tracer_provider = register(endpoint='http://localhost:6006/v1/traces', project_name="vectara-agentic")
-        elif 'app.phoenix.arize.com' in phoenix_endpoint:   # hosted on Arize
-            phoenix_api_key = os.getenv("PHOENIX_API_KEY", None)
-            if not phoenix_api_key:
-                raise ValueError("Arize Phoenix API key not set. Please set PHOENIX_API_KEY environment variable.")
-            os.environ["PHOENIX_CLIENT_HEADERS"] = f"api_key={phoenix_api_key}"
-            os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com"
-            tracer_provider = register(endpoint=phoenix_endpoint, project_name="vectara-agentic")
-        else:       # Self hosted Phoenix
-            tracer_provider = register(endpoint=phoenix_endpoint, project_name="vectara-agentic")
-        LlamaIndexInstrumentor().instrument(tracer_provider=tracer_provider)
-        return True
+    if config.observer != ObserverType.ARIZE_PHOENIX:
+        if verbose:
+            print("No Phoenix observer set.")
+        return False
+
+    try:
+        import phoenix as px
+        from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
+        from phoenix.otel import register
+    except ImportError:
+        print(
+            (
+                "Phoenix libraries not found. Please install with"
+                "'pip install arize-phoenix openinference-instrumentation-llama-index'"
+            )
+        )
+        return False
+
+    phoenix_endpoint = os.getenv("PHOENIX_ENDPOINT", None)
+    if not phoenix_endpoint:
+        print("Phoenix endpoint not set. Attempting to launch local Phoenix UI...")
+        px.launch_app()
+        print("Local Phoenix UI launched. You can view traces at the UI address (usually http://localhost:6006).")
+
+    if phoenix_endpoint and 'app.phoenix.arize.com' in phoenix_endpoint:
+        phoenix_api_key = os.getenv("PHOENIX_API_KEY")
+        if not phoenix_api_key:
+            raise ValueError(
+                "Arize Phoenix API key not set. Please set PHOENIX_API_KEY."
+            )
+        os.environ["PHOENIX_CLIENT_HEADERS"] = f"api_key={phoenix_api_key}"
+        os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com"
+
+    reg_kwargs = {
+        "endpoint": phoenix_endpoint or 'http://localhost:6006/v1/traces',
+        "project_name": "vectara-agentic",
+        "batch": False,
+        "set_global_tracer_provider": False,
+    }
+    tracer_provider = register(**reg_kwargs)
+    LlamaIndexInstrumentor().instrument(tracer_provider=tracer_provider)
     if verbose:
-        print("No observer set.")
-    return False
+        print(f"Phoenix observer configured for project 'vectara-agentic' at endpoint: {reg_kwargs['endpoint']}")
+    return True
 
 
 def _extract_fcs_value(output: Union[str, dict]) -> Optional[float]:

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/vectara_agentic/_prompts.py

@@ -4,11 +4,13 @@ This file contains the prompt templates for the different types of agents.
 
 # General (shared) instructions
 GENERAL_INSTRUCTIONS = """
-- Use tools as your main source of information, do not respond without using a tool. Do not respond based on pre-trained knowledge.
-- Use the 'get_bad_topics' tool to determine the topics you are not allowed to discuss or respond to.
+- Use tools as your main source of information, do not respond without using a tool at least once.
+- Do not respond based on pre-trained knowledge, unless repeated calls to the tools fail or do not provide the information needed.
+- Use the 'get_bad_topics' (if it exists) tool to determine the topics you are not allowed to discuss or respond to.
 - Before responding to a user query that requires knowledge of the current date, call the 'get_current_date' tool to get the current date.
   Never rely on previous knowledge of the current date.
   Example queries that require the current date: "What is the revenue of Apple last october?" or "What was the stock price 5 days ago?".
+  Never call 'get_current_date' more than once for the same user query.
 - When using a tool with arguments, simplify the query as much as possible if you use the tool with arguments.
   For example, if the original query is "revenue for apple in 2021", you can use the tool with a query "revenue" with arguments year=2021 and company=apple.
 - If a tool responds with "I do not have enough information", try one or more of the following strategies:
@@ -24,7 +26,7 @@ GENERAL_INSTRUCTIONS = """
 - If a tool provides citations or references in markdown as part of its response, include the references in your response.
 - Ensure that every URL in your response includes descriptive anchor text that clearly explains what the user can expect from the linked content.
   Avoid using generic terms like “source” or “reference”, or the full URL, as the anchor text.
-- If a tool returns in the metadata a valid URL pointing to a PDF file, along with page number - then combine the URL and page number in the response.
+- If a tool returns in the metadata a valid URL pointing to a PDF file, along with page number - then combine the URL and page number in your response.
   For example, if the URL returned from the tool is "https://example.com/doc.pdf" and "page=5", then the combined URL would be "https://example.com/doc.pdf#page=5".
   If a tool returns in the metadata invalid URLs or an URL empty (e.g. "[[1]()]"), ignore it and do not include that citation or reference in your response.
 - All URLs provided in your response must be obtained from tool output, and cannot be "https://example.com" or empty strings, and should open in a new tab.

{vectara_agentic-0.2.14 → vectara_agentic-0.2.16}/vectara_agentic/_version.py

@@ -1,4 +1,4 @@
 """
 Define the version of the package.
 """
-__version__ = "0.2.14"
+__version__ = "0.2.16"