ibm-watsonx-orchestrate 1.11.0b0__py3-none-any.whl → 1.11.0b1__py3-none-any.whl

ibm_watsonx_orchestrate/__init__.py

@@ -5,7 +5,8 @@
 
 pkg_name = "ibm-watsonx-orchestrate"
 
-__version__ = "1.11.0b0"
+
+__version__ = "1.11.0b1"
 
 
 from ibm_watsonx_orchestrate.utils.logging.logger import setup_logging

ibm_watsonx_orchestrate/agent_builder/agents/types.py

@@ -223,6 +223,11 @@ class ExternalAgentSpec(BaseAgentSpec):
 
     @model_validator(mode="before")
     def validate_fields_for_external(cls, values):
+        # The get api responds with a flat object with no config
+        if values.get("config") is None:
+            values["config"] = {}
+            values["config"]["enable_cot"] = values.get("enable_cot", None)
+            values["config"]["hidden"] = values.get("hidden", None)
         return validate_external_agent_fields(values)
 
     @model_validator(mode="after")
@@ -276,6 +281,14 @@ class AssistantAgentSpec(BaseAgentSpec):
 
     @model_validator(mode="before")
     def validate_fields_for_external(cls, values):
+        if values.get("config") is None:
+            values["config"] = {}
+            values["config"]["api_version"] = values.get("api_version", None)
+            values["config"]["assistant_id"] = values.get("assistant_id", None)
+            values["config"]["crn"] = values.get("crn", None)
+            values["config"]["service_instance_url"] = values.get("service_instance_url", None)
+            values["config"]["environment_id"] = values.get("environment_id", None)
+            values["config"]["authorization_url"] = values.get("authorization_url", None)
         return validate_assistant_agent_fields(values)
 
     @model_validator(mode="after")

ibm_watsonx_orchestrate/agent_builder/connections/connections.py

@@ -1,5 +1,6 @@
 import os
 import logging
+from copy import deepcopy
 from typing import List
 from ibm_watsonx_orchestrate.agent_builder.connections.types import (
     BasicAuthCredentials,
@@ -67,7 +68,8 @@ def _clean_env_vars(vars: dict[str:str], requirements: List[str], app_id: str) -
     return required_env_vars
 
 def _build_credentials_model(credentials_type: type[CREDENTIALS], vars: dict[str,str], base_prefix: str) -> type[CREDENTIALS]:
-    requirements = connection_type_requirements_mapping[credentials_type]
+    requirements_lut = deepcopy(connection_type_requirements_mapping)
+    requirements = requirements_lut[credentials_type]
 
     if requirements:
         requirements.append("url")
@@ -101,7 +103,8 @@ def _get_credentials_model(connection_type: ConnectionSecurityScheme, app_id: st
     
     credentials_type = CONNECTION_TYPE_CREDENTIAL_MAPPING[connection_type]
 
-    requirements = connection_type_requirements_mapping.get(credentials_type)
+    requirements_lut = deepcopy(connection_type_requirements_mapping)
+    requirements = requirements_lut.get(credentials_type)
     if requirements:
         variables = _clean_env_vars(vars=variables, requirements=requirements, app_id=app_id)
 

ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py

@@ -3,7 +3,8 @@ from datetime import datetime
 from uuid import UUID
 from enum import Enum
 
-from pydantic import BaseModel
+from pydantic import BaseModel, model_validator
+
 
 class SpecVersion(str, Enum):
     V1 = "v1"
@@ -104,6 +105,7 @@ class GenerationConfiguration(BaseModel):
     display_text_no_results_found: Optional[str] = None
     display_text_connectivity_issue: Optional[str] = None
     idk_message: Optional[str] = None
+    enabled: bool = True
 
 class FieldMapping(BaseModel):
     """
@@ -224,15 +226,14 @@ class AstraDBConnection(BaseModel):
     api_endpoint: str
     port: Optional[str] = None
     server_cert: Optional[str] = None
-    keyspace: Optional[str]
+    keyspace: Optional[str] = None
     data_type: str
-    collection: Optional[str]
-    table: Optional[str]
-    index_column: Optional[str]
+    collection: Optional[str] = None
+    table: Optional[str] = None
+    index_column: Optional[str] = None
     embedding_mode: str
-    embedding_model_id: Optional[str]
-    credentials: dict
-    search_mode: str
+    embedding_model_id: Optional[str] = None
+    search_mode: Optional[str] = None
     limit: Optional[int] = 5
     filter: Optional[str] = None
     field_mapping: Optional[FieldMapping] = None
@@ -243,7 +244,13 @@ class IndexConnection(BaseModel):
     elastic_search: Optional[ElasticSearchConnection] = None
     custom_search: Optional[CustomSearchConnection] = None
     astradb: Optional[AstraDBConnection] = None
-    
+
+
+class QuerySource(str, Enum):
+    SessionHistory = "SessionHistory"
+    Agent = "Agent"
+
+
 class ConversationalSearchConfig(BaseModel):
     language: Optional[str] = None
     index_config: list[IndexConnection] = None
@@ -252,6 +259,14 @@ class ConversationalSearchConfig(BaseModel):
     citations: Optional[CitationsConfig] = None
     hap_filtering: Optional[HAPFiltering] = None
     confidence_thresholds: Optional[ConfidenceThresholds] = None
+    query_source: QuerySource = QuerySource.SessionHistory
+    agent_query_description: str = "The query to search for in the knowledge base"
+
+    @model_validator(mode="after")
+    def validate_agent_query_description(self) -> 'ConversationalSearchConfig':
+        if self.query_source == QuerySource.Agent and len(self.agent_query_description) == 0:
+            raise ValueError("Provide a non-empty agent_query_description when query source is `Agent`")
+        return self
 
 class KnowledgeBaseBuiltInVectorIndexConfig(BaseModel):
     embeddings_model_name: Optional[str] = None

ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py

@@ -1,5 +1,5 @@
 import typer
-from typing_extensions import Annotated, List
+from typing_extensions import Annotated, List, Optional
 from ibm_watsonx_orchestrate.cli.commands.agents.agents_controller import AgentsController
 from ibm_watsonx_orchestrate.agent_builder.agents.types import DEFAULT_LLM, AgentKind, AgentStyle, ExternalAgentAuthScheme, AgentProvider
 import json
@@ -14,7 +14,7 @@ def agent_import(
         typer.Option("--file", "-f", help="YAML file with agent definition"),
     ],
     app_id: Annotated[
-        str, typer.Option(
+        Optional[str], typer.Option(
             '--app-id', '-a',
             help='The app id of the connection to associate with this external agent. An application connection represents the server authentication credentials needed to connection to this agent (for example Api Keys, Basic, Bearer or OAuth credentials).'
         )
@@ -46,6 +46,13 @@ def agent_create(
         AgentKind,
         typer.Option("--kind", "-k", help="The kind of agent you wish to create"),
     ] = AgentKind.NATIVE,
+    instructions: Annotated[
+        str,
+        typer.Option(
+            "--instructions",
+            help="A set of instructions for how the agent should preform actions.",
+        ),
+    ] = None,
     api_url: Annotated[
         str,
         typer.Option("--api", "-a", help="External Api url your Agent will use"),
@@ -175,6 +182,7 @@ def agent_create(
         kind=kind,
         description=description,
         title=title,
+        instructions=instructions,
         api_url=api_url,
         auth_scheme=auth_scheme,
         auth_config=auth_config_dict,