vectara-agentic 0.3.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

vectara_agentic/types.py

@@ -3,13 +3,13 @@ This module contains the types used in the Vectara Agentic.
 """
 
 from enum import Enum
-from typing import Protocol, Any
+from typing import Any, Dict, Callable, AsyncIterator, Protocol, cast
+from dataclasses import dataclass, field
 
 from llama_index.core.schema import Document as LI_Document
 from llama_index.core.tools.types import ToolOutput as LI_ToolOutput
-from llama_index.core.chat_engine.types import AgentChatResponse as LI_AgentChatResponse
 from llama_index.core.chat_engine.types import (
-    StreamingAgentChatResponse as LI_StreamingAgentChatResponse,
+    AgentChatResponse as LI_AgentChatResponse,
 )
 
 
@@ -17,7 +17,6 @@ class AgentType(Enum):
     """Enumeration for different types of agents."""
 
     REACT = "REACT"
-    OPENAI = "OPENAI"
     FUNCTION_CALLING = "FUNCTION_CALLING"
     LLMCOMPILER = "LLMCOMPILER"
     LATS = "LATS"
@@ -37,7 +36,6 @@ class ModelProvider(Enum):
     ANTHROPIC = "ANTHROPIC"
     TOGETHER = "TOGETHER"
     GROQ = "GROQ"
-    FIREWORKS = "FIREWORKS"
     COHERE = "COHERE"
     GEMINI = "GEMINI"
     BEDROCK = "BEDROCK"
@@ -84,8 +82,102 @@ class HumanReadableOutput(Protocol):
         """Get the raw output data."""
 
 
+class _StreamProto(Protocol):
+    """What we actually use from the LI streaming response."""
+
+    response: str | None
+    response_id: str | None
+
+    def get_response(self) -> LI_AgentChatResponse:
+        """Get the agent chat response."""
+
+    def to_response(self) -> LI_AgentChatResponse:
+        """Convert to agent chat response."""
+
+    def get_final_response(self) -> LI_AgentChatResponse:
+        """Get the final agent chat response."""
+
+    def __iter__(self):
+        """Return an iterator over the stream."""
+
+    async def async_response_gen(self) -> AsyncIterator[str]:
+        """Async generator that yields response strings."""
+
+
 # classes for Agent responses
 ToolOutput = LI_ToolOutput
 AgentResponse = LI_AgentChatResponse
-AgentStreamingResponse = LI_StreamingAgentChatResponse
 Document = LI_Document
+
+
+@dataclass
+class AgentStreamingResponse:
+    """Our stream wrapper with writable metadata and a typed get_response()."""
+
+    base: _StreamProto
+    metadata: Dict[str, Any] = field(default_factory=dict)
+
+    def __getattr__(self, name: str):
+        return getattr(self.base, name)
+
+    def get_response(self) -> AgentResponse:
+        """Get the response from the base stream, merging metadata."""
+        # choose whichever method the base actually has
+        if hasattr(self.base, "get_response"):
+            resp = cast(AgentResponse, self.base.get_response())
+        elif hasattr(self.base, "to_response"):
+            resp = cast(AgentResponse, self.base.to_response())
+        else:
+            resp = cast(AgentResponse, self.base.get_final_response())
+
+        resp.metadata = (resp.metadata or {}) | self.metadata
+        return resp
+
+    async def aget_response(self) -> AgentResponse:
+        """Get the response from the base stream, merging metadata (async version)."""
+        # prefer async version if available
+        if hasattr(self.base, "aget_response"):
+            resp = cast(AgentResponse, await self.base.aget_response())
+        elif hasattr(self.base, "get_response"):
+            resp = cast(AgentResponse, self.base.get_response())
+        elif hasattr(self.base, "to_response"):
+            resp = cast(AgentResponse, self.base.to_response())
+        else:
+            resp = cast(AgentResponse, self.base.get_final_response())
+
+        resp.metadata = (resp.metadata or {}) | self.metadata
+        return resp
+
+    @property
+    def async_response_gen(self) -> Callable[[], AsyncIterator[str]]:
+        """Get the async response generator from the base stream."""
+        return self.base.async_response_gen
+
+    @async_response_gen.setter
+    def async_response_gen(self, fn: Callable[[], AsyncIterator[str]]):
+        """Set the async response generator for the base stream."""
+        self.base.async_response_gen = fn
+
+    @classmethod
+    def from_error(cls, msg: str) -> "AgentStreamingResponse":
+        """Create an AgentStreamingResponse from an error message."""
+
+        async def _empty_gen():
+            if False:  # pylint: disable=using-constant-test
+                yield ""
+
+        class _ErrStream:
+            def __init__(self, msg: str):
+                self.response = msg
+                self.response_id = None
+
+            async def async_response_gen(self):
+                """Async generator that yields an error message."""
+                if False:  # pylint: disable=using-constant-test
+                    yield ""
+
+            def get_response(self) -> AgentResponse:
+                """Return an AgentResponse with the error message."""
+                return AgentResponse(response=self.response, metadata={})
+
+        return cls(base=_ErrStream(msg), metadata={})

{vectara_agentic-0.3.2.dist-info → vectara_agentic-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectara_agentic
-Version: 0.3.2
+Version: 0.4.0
 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,26 +16,26 @@ 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.37
-Requires-Dist: llama-index-core==0.12.37
-Requires-Dist: llama-index-cli==0.4.1
+Requires-Dist: llama-index==0.12.49
+Requires-Dist: llama-index-core==0.12.49
+Requires-Dist: llama-index-workflow==1.0.1
+Requires-Dist: llama-index-cli==0.4.4
 Requires-Dist: llama-index-indices-managed-vectara==0.4.5
-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.44
-Requires-Dist: llama-index-llms-openai-like==0.3.5
-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-agent-llm-compiler==0.3.2
+Requires-Dist: llama-index-agent-lats==0.3.2
+Requires-Dist: llama-index-agent-openai==0.4.12
+Requires-Dist: llama-index-llms-openai==0.4.7
+Requires-Dist: llama-index-llms-openai-like==0.4.0
+Requires-Dist: llama-index-llms-anthropic==0.7.6
+Requires-Dist: llama-index-llms-together==0.3.2
+Requires-Dist: llama-index-llms-groq==0.3.2
 Requires-Dist: llama-index-llms-cohere==0.5.0
-Requires-Dist: llama-index-llms-google-genai==0.2.1
-Requires-Dist: llama-index-llms-bedrock-converse==0.7.1
+Requires-Dist: llama-index-llms-google-genai==0.2.5
+Requires-Dist: llama-index-llms-bedrock-converse==0.7.6
 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
-Requires-Dist: llama-index-tools-google==0.3.1
+Requires-Dist: llama-index-tools-google==0.5.0
 Requires-Dist: llama-index-tools-tavily_research==0.3.0
 Requires-Dist: llama_index.tools.brave_search==0.3.0
 Requires-Dist: llama-index-tools-neo4j==0.3.0
@@ -44,11 +44,11 @@ Requires-Dist: llama-index-graph-stores-kuzu==0.7.0
 Requires-Dist: llama-index-tools-salesforce==0.3.0
 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-wikipedia==0.3.1
 Requires-Dist: llama-index-tools-bing-search==0.3.0
-Requires-Dist: openai>=1.82.1
-Requires-Dist: tavily-python==0.7.3
-Requires-Dist: exa-py==1.13.1
+Requires-Dist: openai>=1.96.1
+Requires-Dist: tavily-python>=0.7.9
+Requires-Dist: exa-py>=1.14.8
 Requires-Dist: openinference-instrumentation-llama-index==4.3.1
 Requires-Dist: opentelemetry-proto>=1.31.0
 Requires-Dist: arize-phoenix==10.9.1
@@ -56,9 +56,9 @@ Requires-Dist: arize-phoenix-otel==0.10.3
 Requires-Dist: protobuf==5.29.3
 Requires-Dist: tokenizers>=0.20
 Requires-Dist: pydantic==2.11.5
+Requires-Dist: pandas==2.2.3
 Requires-Dist: retrying==1.3.4
 Requires-Dist: python-dotenv==1.0.1
-Requires-Dist: tiktoken==0.9.0
 Requires-Dist: cloudpickle>=3.1.1
 Requires-Dist: httpx==0.28.1
 Requires-Dist: commonmark==0.9.1
@@ -108,6 +108,7 @@ Dynamic: summary
 - [Using Tools](#using-tools)
 - [Advanced Usage: Workflows](#advanced-usage-workflows)
 - [Configuration](#️-configuration)
+- [Migrating from v0.3.x](#-migrating-from-v03x)
 - [Contributing](#-contributing)
 - [License](#-license)
 
@@ -124,11 +125,11 @@ Dynamic: summary
 - **Rapid Tool Creation:**  
   Build Vectara RAG tools or search tools with a single line of code.
 - **Agent Flexibility:**  
-  Supports multiple agent types including `ReAct`, `OpenAIAgent`, `LATS`, and `LLMCompiler`.
+  Supports multiple agent types including `ReAct`, `Function Calling`, `LATS`, and `LLMCompiler`.
 - **Pre-Built Domain Tools:**  
   Tools tailored for finance, legal, and other verticals.
 - **Multi-LLM Integration:**  
-  Seamless integration with OpenAI, Anthropic, Gemini, GROQ, Together.AI, Cohere, Bedrock, and Fireworks.
+  Seamless integration with OpenAI, Anthropic, Gemini, GROQ, Together.AI, Cohere, and Bedrock.
 - **Observability:**  
   Built-in support with Arize Phoenix for monitoring and feedback.
 - **Workflow Support:**  
@@ -148,7 +149,7 @@ 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, Cohere, GEMINI or GROQ, if you choose to use them).
+- OpenAI API key (or API keys for Anthropic, TOGETHER.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.
@@ -200,7 +201,8 @@ ask_finance = vec_factory.create_rag_tool(
     tool_description="Query financial reports for a company and year",
     tool_args_schema=QueryFinancialReportsArgs,
     lambda_val=0.005,
-    summary_num_results=7, 
+    summary_num_results=7,
+    vhc_eligible=True,  # RAG tools participate in VHC by default
     # Additional Vectara query arguments...
 )
 ```
@@ -480,6 +482,30 @@ def mult_func(x, y):
 mult_tool = ToolsFactory().create_tool(mult_func)
 ```
 
+#### VHC Eligibility
+
+When creating tools, you can control whether they participate in Vectara Hallucination Correction, by using the `vhc_eligible` parameter:
+
+```python
+# Tool that provides factual data - should participate in VHC
+data_tool = ToolsFactory().create_tool(get_company_data, vhc_eligible=True)
+
+# Utility tool that doesn't provide context - should not participate in VHC  
+summary_tool = ToolsFactory().create_tool(summarize_text, vhc_eligible=False)
+```
+
+**VHC-eligible tools** (default: `True`) are those that provide factual context for responses, such as:
+- Data retrieval tools
+- Search tools  
+- API calls that return factual information
+
+**Non-VHC-eligible tools** (`vhc_eligible=False`) are utility tools that don't contribute factual context:
+- Text summarization tools
+- Text rephrasing tools
+- Formatting or processing tools
+
+Built-in utility tools like `summarize_text`, `rephrase_text`, and `get_bad_topics` are automatically marked as non-VHC-eligible.
+
 #### Human-Readable Tool Output
 
 Tools can provide both raw data and human-readable formatted output using the `create_human_readable_output` utility:
@@ -504,7 +530,7 @@ Built-in formatters include `format_as_table`, `format_as_json`, and `format_as_
 > and not as nested functions. Nested functions are not supported if you use serialization 
 > (dumps/loads or from_dict/to_dict).
 
-The human-readable format, if available, is used when computing the factual consistency score.
+The human-readable format, if available, is used when using Vectara Hallucination Correction.
 
 ### Tool Validation
 
@@ -719,12 +745,13 @@ agent = Agent(
 ```
 
 The `AgentConfig` object may include the following items:
-- `agent_type`: the agent type. Valid values are `REACT`, `LLMCOMPILER`, `LATS` or `OPENAI` (default: `OPENAI`).
-- `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` or `FIREWORKS` (default: `OPENAI`).
-- `main_llm_model_name` and `tool_llm_model_name`: agent model name for agent and tools (default depends on provider).
+- `agent_type`: the agent type. Valid values are `REACT`, `LLMCOMPILER`, `LATS` or `FUNCTION_CALLING` (default: `FUNCTION_CALLING`).
+- `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, Gemini uses gemini-2.5-flash).
 - `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`)
-- `max_reasoning_steps`: the maximum number of reasoning steps (iterations for React and function calls for OpenAI agent, respectively). Defaults to 50.
 
 If any of these are not provided, `AgentConfig` first tries to read the values from the OS environment.
 
@@ -759,3 +786,15 @@ agent = Agent(
 )
 ```
 
+## 🚀 Migrating from v0.3.x
+
+If you're upgrading from v0.3.x, please note the following breaking changes in v0.4.0:
+
+- **Fireworks LLM removed**: Migrate to OpenAI, Anthropic, Together.AI, GROQ, Cohere, Bedrock, or Gemini
+- **OPENAI AgentType removed**: Use the FUNCTION_CALLING AgentType instead, when using OpenAI for main_llm_provider
+- **StructuredPlanning deprecated**: Use standard Agent workflows or create custom workflows
+- **Token counting and compact_docstring removed**: Remove these from your configuration
+- **update_func removed**: This functionality is no longer available
+
+For detailed migration instructions, see [CHANGELOG.md](CHANGELOG.md).
+

vectara_agentic-0.4.0.dist-info/RECORD

@@ -0,0 +1,50 @@
+tests/__init__.py,sha256=vXhQJCyD1Uhx2NP8b8vIUG3RUhkXyvn7oOir2bmctQU,175
+tests/conftest.py,sha256=WHK_SxlhU2EN55w8wXUMMHhks8yriOab5nK1y8HXe3g,9276
+tests/endpoint.py,sha256=0URgtz8uydhP_rtpGn_59P1LiWkd3idNlI85LzXnlUE,2744
+tests/run_tests.py,sha256=eZd50pV4FIbr8riDaqXvDheoW3mOcO3ZRGloGUNusAM,3197
+tests/test_agent.py,sha256=V5r7Cqe0iqV8VmeDJdE2lvB5tBLQUqcX182HIXTNYQQ,5721
+tests/test_agent_type.py,sha256=WNUwyUxC431BTtQPSfKpG42IxsPnexdbXQTM3P6itBk,5085
+tests/test_api_endpoint.py,sha256=I2UDamPMSLLkgw0pZ5QMM0o_8vVga9-F6ql-S3zlMBs,5136
+tests/test_bedrock.py,sha256=lXIHGtH0otjOqCkCSnanEUM6HavSkbail1900drfJiU,1358
+tests/test_fallback.py,sha256=6wkyiyAvsibIdr33aXdsuU9nzDeJt0XSz5yiyuisUEQ,2963
+tests/test_gemini.py,sha256=ksHd8JA7ZMuzs8W40Fb4RBoen1rniXDghSfQImw_3nk,3016
+tests/test_groq.py,sha256=m6HpJEeqmDQqYCQwg9_bCyGJ3ek1tK-aLBktxgRGGJ0,1346
+tests/test_private_llm.py,sha256=P6sldeAWcHg29u_Nu4FdHVUyNaRe5ULE-hjNJz6WKHc,2620
+tests/test_return_direct.py,sha256=QsCw-ZGp06cutLkyrLh1U1rggoH7iBiFz4SQ9MIx-Xk,1521
+tests/test_serialization.py,sha256=CsW7qEXgGE24oEqo85c-GEbzn_mZjbF3er_juL9JbF8,4896
+tests/test_streaming.py,sha256=e_XztBLCWf39HgfN1zsUz_vFNblzmMC2zfYHB8JM-zQ,2795
+tests/test_tools.py,sha256=869Fl54kmLc44ijykO2QpfcXyAWLDqJ9Niq3XNzhzv8,13621
+tests/test_vectara_llms.py,sha256=H1M9OaDvD8_GCFRBm6IdvWejYKn-zm3-Rzt_noCBbiQ,2496
+tests/test_vhc.py,sha256=oDZzx1AdtDO0K5MHpzrCegw7wfc3h9E0V7boVFoMWXs,1945
+tests/test_workflow.py,sha256=FXUM4hKh-La9FRJD0ir2sOiXsvkDFe2kI0r1faRAlMc,3873
+vectara_agentic/__init__.py,sha256=CfS3QR4drKygcTcyH5zUUDuXXQ3WZtTCytz8W4-loeE,1077
+vectara_agentic/_callback.py,sha256=ueckIfLNa9ykmmEyLqrrZwfDNWrEfyZzJeWktpnkwJQ,12970
+vectara_agentic/_observability.py,sha256=c_1WuP8rS9sPuMH6twzcz6CGLqfTT5y4fyOHvDVdxsg,4423
+vectara_agentic/_version.py,sha256=tSVqctgqLCPSvb0zkh8BNhEaR1d92yEJqmdCvwJzdKQ,65
+vectara_agentic/agent.py,sha256=ga-8Flc01EX6xUhEx-MHArjb8bUeAYlRK_cKioihxdk,38667
+vectara_agentic/agent_config.py,sha256=njqEX2qHJjAp2KpNuJglgZhyWXPK74wjIjBPACD6w7w,4074
+vectara_agentic/agent_endpoint.py,sha256=E_AF-YwxaKqd1-p43X62e1e4ugwOWKIyNq4RWOfsO7A,7402
+vectara_agentic/db_tools.py,sha256=nVZkpGdG63ooGngjX9g7YWyBZRtYMDpvzNasbO696nM,11498
+vectara_agentic/llm_utils.py,sha256=s0g04lqQkX27njAKPAM-H0ZFEmohaC0VO7hs_ByaGaQ,7460
+vectara_agentic/sub_query_workflow.py,sha256=wm2Lb2wbKrYx5bSq-npb3XbaxWzTcvK5BkW3NZ9vuIc,12968
+vectara_agentic/tool_utils.py,sha256=whnQlk9coeIt01sqUnKnzUorefgn96yWqhtRfHxNL84,25921
+vectara_agentic/tools.py,sha256=8gmC6UnHFTUr_hWWbuMyRNMMLkeY5Sb1FTgCsb7Hx1w,35689
+vectara_agentic/tools_catalog.py,sha256=p6eRram-diJyMz5dZI703auSAm97FfW5wLAMyz_2sB0,4634
+vectara_agentic/types.py,sha256=V04L8Man79qI9SmnKhlR3verJjm7yhcEE1RHPq9ADpc,5580
+vectara_agentic/utils.py,sha256=R9HitEG5K3Q_p2M_teosT181OUxkhs1-hnj98qDYGbE,2545
+vectara_agentic/agent_core/__init__.py,sha256=R3KGbSOiY21FOjbeQ_GyIi6uR9Rz7PTfudO9RjSuEZQ,722
+vectara_agentic/agent_core/factory.py,sha256=yIoA-GumyuoUu-tmfAp79v2kAujUj7D7a7d5vx3_kj8,17697
+vectara_agentic/agent_core/prompts.py,sha256=JGyAyZyLd__hTuEeBBuCHFdIS1nTIQJZJPGbxRpxY7A,9414
+vectara_agentic/agent_core/serialization.py,sha256=Kxa7irtaeOhw2NbPpPkT3R7rKe-imx13XCL1V63eRqI,11634
+vectara_agentic/agent_core/streaming.py,sha256=0mN5qpDP9evXOG_vj65GINhmUkbSQsWmGUsVDkNVPFE,18134
+vectara_agentic/agent_core/utils/__init__.py,sha256=kLdT0Idw0xhT1zOJIhx13T4qsWh01O3taNC7aN2SEI4,958
+vectara_agentic/agent_core/utils/hallucination.py,sha256=KG-ELY9ZzCwBjj4KMyncPkgvEg190Pw2D612O9fHE-Q,7037
+vectara_agentic/agent_core/utils/logging.py,sha256=-Ll8iUelml92WuhNWScuY6H-RheyZOTBHNxXQ1UGy0M,1701
+vectara_agentic/agent_core/utils/prompt_formatting.py,sha256=C0WqmSHZ-r_asRi2mkMsFOlqrOVrmADqNudidS6CU3s,1801
+vectara_agentic/agent_core/utils/schemas.py,sha256=e7xhJBevgK7IM8cRT5hoO67T-Ep_FhNGp72Zo0OC_Jo,2853
+vectara_agentic/agent_core/utils/tools.py,sha256=k9Gm-UUQ3ZeGxrkjyrjmjcGxOkvnpylcm_Krnr-0fsY,4748
+vectara_agentic-0.4.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vectara_agentic-0.4.0.dist-info/METADATA,sha256=BAILIhLZFOZAo71UTsWlzDPnUGfmq8nhJDRfAwzEgEc,33857
+vectara_agentic-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vectara_agentic-0.4.0.dist-info/top_level.txt,sha256=Y7TQTFdOYGYodQRltUGRieZKIYuzeZj2kHqAUpfCUfg,22
+vectara_agentic-0.4.0.dist-info/RECORD,,

tests/test_agent_planning.py

@@ -1,64 +0,0 @@
-import unittest
-
-from vectara_agentic.agent_config import AgentConfig
-from vectara_agentic.agent import Agent
-from vectara_agentic.tools import VectaraToolFactory
-
-# 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'
-
-vec_factory = VectaraToolFactory(vectara_api_key=vectara_api_key,
-                                 vectara_corpus_key=vectara_corpus_key)
-summarizer = 'vectara-summary-table-md-query-ext-jan-2025-gpt-4o'
-ask_vectara = vec_factory.create_rag_tool(
-    tool_name = "ask_vectara",
-    tool_description = "This tool can respond to questions about Vectara.",
-    reranker = "multilingual_reranker_v1", rerank_k = 100, rerank_cutoff = 0.1,
-    n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
-    summary_num_results = 10,
-    vectara_summarizer = summarizer,
-    include_citations = True,
-    verbose=False,
-)
-
-class TestAgentPlanningPackage(unittest.TestCase):
-
-    def test_no_planning(self):
-        tools = [ask_vectara]
-        topic = "vectara"
-        instructions = "Answer user queries about Vectara."
-
-        query = "What is Vectara and what demos are available of the Vectara platform?"
-        agent = Agent(
-            tools=tools,
-            topic=topic,
-            custom_instructions=instructions,
-            agent_config=AgentConfig(),
-        )
-        res = agent.chat(query)
-        self.assertIn("demos", res.response)
-        self.assertIn("Vectara", res.response)
-
-    def test_structured_planning(self):
-        tools = [ask_vectara]
-        topic = "vectara"
-        instructions = "Answer user queries about Vectara."
-
-        query = "What is Vectara and what demos are available of the Vectara platform?"
-        agent = Agent(
-            tools=tools,
-            topic=topic,
-            custom_instructions=instructions,
-            agent_config=AgentConfig(),
-            use_structured_planning=True,
-        )
-
-        res = agent.chat(query)
-        self.assertIn("demos", res.response)
-        self.assertIn("Vectara", res.response)
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_hhem.py

@@ -1,100 +0,0 @@
-import unittest
-
-from vectara_agentic.agent import Agent, AgentType
-from vectara_agentic.agent_config import AgentConfig
-from vectara_agentic.tools import ToolsFactory, VectaraToolFactory
-from vectara_agentic.types import ModelProvider
-
-import nest_asyncio
-
-nest_asyncio.apply()
-
-vectara_corpus_key = "vectara-docs_1"
-vectara_api_key = 'zqt_UXrBcnI2UXINZkrv4g1tQPhzj02vfdtqYJIDiA'
-
-vec_factory = VectaraToolFactory(vectara_api_key=vectara_api_key,
-                                 vectara_corpus_key=vectara_corpus_key)
-summarizer = 'vectara-summary-table-md-query-ext-jan-2025-gpt-4o'
-ask_vectara = vec_factory.create_rag_tool(
-    tool_name = "ask_vectara",
-    tool_description = "This tool can respond to questions about Vectara.",
-    reranker = "multilingual_reranker_v1", rerank_k = 100, rerank_cutoff = 0.1,
-    n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
-    summary_num_results = 10,
-    vectara_summarizer = summarizer,
-    include_citations = True,
-    verbose=False,
-)
-
-
-statements = [
-    "The sky is blue.",
-    "Cats are better than dogs.",
-    "Python is a great programming language.",
-    "The Earth revolves around the Sun.",
-    "Chocolate is the best ice cream flavor.",
-]
-st_inx = 0
-def get_statement() -> str:
-    "Generate next statement"
-    global st_inx
-    st = statements[st_inx]
-    st_inx += 1
-    return st
-
-
-fc_config = AgentConfig(
-    agent_type=AgentType.FUNCTION_CALLING,
-    main_llm_provider=ModelProvider.ANTHROPIC,
-    tool_llm_provider=ModelProvider.ANTHROPIC,
-)
-
-
-class TestFCS(unittest.TestCase):
-
-    def test_fcs(self):
-        tools = [ToolsFactory().create_tool(get_statement)]
-        topic = "statements"
-        instructions = (
-            f"Call the get_statement tool multiple times to get all {len(statements)} statements."
-            f"Respond to the user question based exclusively on the statements you receive - do not use any other knowledge or information."
-        )
-
-        agent = Agent(
-            tools=tools,
-            topic=topic,
-            agent_config=fc_config,
-            custom_instructions=instructions,
-            vectara_api_key=vectara_api_key,
-        )
-
-        res = agent.chat("Are cats better than dogs?")
-        fcs = res.metadata.get("fcs", None)
-        self.assertIsNotNone(fcs, "FCS score should not be None")
-        self.assertIsInstance(fcs, float, "FCS score should be a float")
-        self.assertGreater(
-            fcs, 0.5, "FCS score should be higher than 0.5 for this question"
-        )
-
-    def test_vectara_corpus(self):
-        tools = [ask_vectara]
-        topic = "vectara"
-        instructions = "Answer user queries about Vectara."
-
-        query = "What is Vectara and what API endpoints are available of the Vectara platform?"
-        agent = Agent(
-            tools=tools,
-            topic=topic,
-            custom_instructions=instructions,
-            agent_config=AgentConfig(),
-            vectara_api_key=vectara_api_key,
-        )
-
-        res = agent.chat(query)
-        fcs = res.metadata.get("fcs", None)
-        self.assertIn("Vectara", res.response)
-        self.assertGreater(fcs, 0.5, "FCS score should be higher than 0.5 for this question")
-
-
-if __name__ == "__main__":
-    unittest.main()

vectara_agentic/hhem.py

@@ -1,82 +0,0 @@
-"""Vectara HHEM (Hypothesis Hypothetical Evaluation Model) client."""
-
-import requests
-from commonmark import Parser
-
-
-def markdown_to_text(md: str) -> str:
-    """
-    Convert a Markdown-formatted string into plain text.
-    """
-    parser = Parser()
-    ast = parser.parse(md)
-    out: list[str] = []
-
-    def recurse(node):
-        if node.t in ("text", "code", "html_inline"):
-            out.append(node.literal or "")
-        elif node.t == "softbreak":
-            out.append(" ")
-        elif node.t == "linebreak":
-            out.append("\n")
-        child = getattr(node, "first_child", None)
-        while child is not None:
-            recurse(child)
-            child = getattr(child, "next", None)
-
-    recurse(ast)
-    text = "".join(out)
-    # collapse runs of spaces but preserve newlines
-    lines = [" ".join(line.split()) for line in text.splitlines()]
-    return "\n".join(line if line.strip() else "" for line in lines)
-
-
-class HHEM:
-    """Vectara HHEM (Hypothesis Hypothetical Evaluation Model) client."""
-
-    def __init__(self, vectara_api_key: str):
-        self._vectara_api_key = vectara_api_key
-
-    def compute(self, context: str, hypothesis: str) -> float:
-        """
-        Calls the Vectara HHEM endpoint to evaluate the factual consistency of a hypothesis against a given context.
-
-        Parameters:
-            context (str): The source text against which the hypothesis will be evaluated.
-            hypothesis (str): The generated text to be evaluated for factual consistency.
-
-        Returns:
-            float: The factual consistency score rounded to four decimal places.
-
-        Raises:
-            requests.exceptions.RequestException: If there is a network-related error or the API call fails.
-        """
-
-        # clean response from any markdown or other formatting.
-        try:
-            clean_hypothesis = markdown_to_text(hypothesis)
-        except Exception as e:
-            # If markdown parsing fails, use the original text
-            raise ValueError(f"Markdown parsing of hypothesis failed: {e}") from e
-
-        # compute HHEM with Vectara endpoint
-        payload = {
-            "model_parameters": {"model_name": "hhem_v2.3"},
-            "generated_text": clean_hypothesis,
-            "source_texts": [context],
-        }
-        headers = {
-            "Content-Type": "application/json",
-            "Accept": "application/json",
-            "x-api-key": self._vectara_api_key,
-        }
-
-        response = requests.post(
-            "https://api.vectara.io/v2/evaluate_factual_consistency",
-            json=payload,
-            headers=headers,
-            timeout=30,
-        )
-        response.raise_for_status()
-        data = response.json()
-        return round(data.get("score", 0.0), 4)

vectara_agentic-0.3.2.dist-info/RECORD

@@ -1,39 +0,0 @@
-tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/endpoint.py,sha256=frnpdZQpnuQNNKNYgAn2rFTarNG8MCJaNA77Bw_W22A,1420
-tests/test_agent.py,sha256=4aRDhEVM1zARpSOevPR8PFGTkIaSw5g5WvoQDx-4sns,5496
-tests/test_agent_planning.py,sha256=JwEebGooROAvsQ9JZoaH6KEcrSyv1F0lL4TD4FjP8a8,2213
-tests/test_agent_type.py,sha256=9ZImIDw7Qz5EV2tX8bDD0tT1IFYghS3-SShpAezHO7s,7381
-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=5yTlOLwpzGRmiBPExAuulK6SEO9O13cMJWVSSHX9CsE,1212
-tests/test_hhem.py,sha256=XFUX3x0xOa9QsafzYNP8_ZsvzIiLnWee4hdsowzHft8,3250
-tests/test_private_llm.py,sha256=-bQBI69Z-SwhhQJGzyC9GGM4kz5c6T5kAbXLbqn_G7E,2248
-tests/test_return_direct.py,sha256=mgj1xJnvOX28aaidhZfH2DaI-kM0yfQ0P1PNV0iyLcw,1405
-tests/test_serialization.py,sha256=Ed23GN2zhSJNdPFrVK4aqLkOhJKviczR_o0t-r9TuRI,4762
-tests/test_tools.py,sha256=bhhq49I3eEJyuNf5FU7M9IxPT0XtwRbRTs3gwoBHihk,17128
-tests/test_vectara_llms.py,sha256=gw5KQ4XT3L-_A6yj2jKqj-QomCjOb9VESKBtBH2Xb8s,2362
-tests/test_workflow.py,sha256=TmNBxBqSW5owk_Nz9LLtHvqryVNsFPkf-M1G_uFSsAM,3739
-vectara_agentic/__init__.py,sha256=2GLDS3U6KckK-dBRl9v_x1kSV507gEhjOfuMmmu0Qxg,850
-vectara_agentic/_callback.py,sha256=c3848EMSpaQWXtuwdqRGbhgbZhiDwgGnemJkgm9yWAc,13238
-vectara_agentic/_observability.py,sha256=iZlByeQTyx6g3Y8aBYcdGcxdRkoYrfxHdcrTEKO26UE,4485
-vectara_agentic/_prompts.py,sha256=9s8VEjaaLuRgNK1xQYWj4bnjM4asJP1Z5zCihUMRonk,10768
-vectara_agentic/_version.py,sha256=5evj7VxbzqoTrhhHqk9AvX1nIb07P-5iiJ7QJ_zRV8A,65
-vectara_agentic/agent.py,sha256=zu7nMxhKin3rLuV8y4F_OcssU3R8bJOjMixKMC_P2k0,58857
-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=GUsQTZfRbT9F5K_e5HNaKXUkU6x8RErUyjDVKlZi1IA,11196
-vectara_agentic/hhem.py,sha256=j4euBX24PSCQ8P_MhhsKKnm1kv6nHKAbduHsTwtQuR0,2774
-vectara_agentic/llm_utils.py,sha256=TX01e4QY8qb5O5D6ZrlkLZEZFHJ4LbDL6g-l52lTB40,7561
-vectara_agentic/sub_query_workflow.py,sha256=JYwN0wK4QzHjTaFDsSCAQvMx9GD4g6CnqxZCnzi6xb4,13086
-vectara_agentic/tool_utils.py,sha256=9xoqVPB97CIDXOxuFIw4yZ2RlXvdayCEGPUaUPC2Tbc,24168
-vectara_agentic/tools.py,sha256=bj8Zn3Lv63vWxu7N6_kkvOk9Vr2ZtuiiBetXUCzsK0w,34860
-vectara_agentic/tools_catalog.py,sha256=cAN_kDOWZUoW4GNFwY5GdS6ImMUQNnF2sggx9OGK9Cg,4906
-vectara_agentic/types.py,sha256=3mrtshHiy-d5JHVxl-4tJk5DRspvYKwAYiI5LvKO1Bw,2226
-vectara_agentic/utils.py,sha256=R9HitEG5K3Q_p2M_teosT181OUxkhs1-hnj98qDYGbE,2545
-vectara_agentic-0.3.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vectara_agentic-0.3.2.dist-info/METADATA,sha256=BpKTuP41lQct4SaRL9kWCwRqg5zAn75ffLAhJ7enVpc,32079
-vectara_agentic-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vectara_agentic-0.3.2.dist-info/top_level.txt,sha256=Y7TQTFdOYGYodQRltUGRieZKIYuzeZj2kHqAUpfCUfg,22
-vectara_agentic-0.3.2.dist-info/RECORD,,