Added - Support for GenAI Agents Tools API

Author: rashimahawar

examples/generative_ai_agent/agent.tf

@@ -9,11 +9,14 @@ resource "oci_generative_ai_agent_agent" "test_agent" {
   display_name                  = var.agent_display_name
   description                   = var.agent_description
   welcome_message               = var.agent_welcome_message
-  #defined_tags not tested - cannot test in home region        
+  llm_config {
+    routing_llm_customization {
+          #Optional
+      instruction = var.agent_llm_config_routing_llm_customization_instruction
+    }
+  }
+  #defined_tags not tested - cannot test in home region
   freeform_tags                = var.test_freeform_tags
-  knowledge_base_ids = [
-    oci_generative_ai_agent_knowledge_base.test_service_managed_knowledge_base.id
-  ]
 }
 
 data "oci_generative_ai_agent_agent" "test_agent" {

examples/generative_ai_agent/agent_endpoint.tf

@@ -1,20 +1,61 @@
 resource "oci_generative_ai_agent_agent_endpoint" "test_agent_endpoint" {
   #Required
   compartment_id                 = var.compartment_ocid
-  agent_id                       = data.oci_generative_ai_agent_agent.test_agent.id
+  agent_id                       = var.test_agent_id
 
   #Optional
   display_name                  = var.test_agent_endpoint_display_name
   description                   = var.test_agent_endpoint_description
-  #defined_tags not tested - cannot test in home region        
+  #defined_tags not tested - cannot test in home region
   freeform_tags                 = var.test_freeform_tags
   should_enable_citation        = var.should_enable_citation
   should_enable_session         = var.should_enable_session
   should_enable_trace           = var.should_enable_trace
+  should_enable_multi_language  = var.agent_endpoint_should_enable_multi_language
   content_moderation_config  {
     should_enable_on_input = var.should_enable_on_input
     should_enable_on_output = var.should_enable_on_output
   }
+  guardrail_config {
+
+    #Optional
+    content_moderation_config {
+
+      #Optional
+      input_guardrail_mode  = var.agent_endpoint_guardrail_config_content_moderation_config_input_guardrail_mode
+      output_guardrail_mode = var.agent_endpoint_guardrail_config_content_moderation_config_output_guardrail_mode
+    }
+    personally_identifiable_information_config {
+
+      #Optional
+      input_guardrail_mode  = var.agent_endpoint_guardrail_config_personally_identifiable_information_config_input_guardrail_mode
+      output_guardrail_mode = var.agent_endpoint_guardrail_config_personally_identifiable_information_config_output_guardrail_mode
+    }
+    prompt_injection_config {
+
+      #Optional
+      input_guardrail_mode = var.agent_endpoint_guardrail_config_prompt_injection_config_input_guardrail_mode
+    }
+  }
+  human_input_config {
+    #Required
+    should_enable_human_input = var.agent_endpoint_human_input_config_should_enable_human_input
+  }
+  output_config {
+    #Required
+    output_location {
+      #Required
+      bucket               = var.agent_endpoint_output_config_output_location_bucket
+      namespace            = var.agent_endpoint_output_config_output_location_namespace
+      output_location_type = var.agent_endpoint_output_config_output_location_output_location_type
+
+      #Optional
+      prefix = var.agent_endpoint_output_config_output_location_prefix
+    }
+
+    #Optional
+    retention_period_in_minutes = var.agent_endpoint_output_config_retention_period_in_minutes
+  }
   session_config              {
     idle_timeout_in_seconds = var.idle_timeout_in_seconds
   }

examples/generative_ai_agent/data_source.tf

@@ -1,20 +1,20 @@
 resource "oci_generative_ai_agent_data_source" "test_data_source" {
   #Required
   compartment_id                 = var.compartment_ocid
-  knowledge_base_id              = oci_generative_ai_agent_knowledge_base.test_service_managed_knowledge_base.id
+  knowledge_base_id              = var.test_knowledge_base_id
   data_source_config  {
-    data_source_config_type = "OCI_OBJECT_STORAGE"
+    data_source_config_type      = "OCI_OBJECT_STORAGE"
     object_storage_prefixes {
-      bucket = data.oci_objectstorage_bucket.bucket.name
-      namespace = data.oci_objectstorage_namespace.ns.namespace
-      prefix = var.test_data_source_prefix
+      bucket = var.test_bucket
+      namespace = var.test_namespace
+      prefix = var.test_prefix
     }
   }
 
   #Optional
   display_name                  = var.test_data_source_display_name
   description                   = var.test_data_source_description
-  #defined_tags not tested - cannot test in home region        
+  #defined_tags not tested - cannot test in home region
   freeform_tags                 = var.test_freeform_tags
 }
 
@@ -27,13 +27,3 @@ data "oci_generative_ai_agent_data_sources" "test_data_sources" {
   #Required
   compartment_id                = var.compartment_ocid
 }
-
-data "oci_objectstorage_namespace" "ns" {
-  #Optional
-  compartment_id = var.compartment_ocid
-}
-
-data "oci_objectstorage_bucket" "bucket" {
-  name                         = "oci-docs"
-  namespace                    = data.oci_objectstorage_namespace.ns.namespace
-}

examples/generative_ai_agent/tools.tf

@@ -0,0 +1,57 @@
+resource "oci_generative_ai_agent_tool" "rag_test_tool" {
+  agent_id       = var.test_agent_id
+  compartment_id = var.compartment_ocid
+  display_name   = var.test_agent_tool_display_name
+  description    = var.test_agent_tool_description
+  freeform_tags = {
+    Department = "Finance"
+  }
+  tool_config {
+    knowledge_base_configs {
+      knowledge_base_id = var.test_knowledge_base_id
+    }
+    tool_config_type = "RAG_TOOL_CONFIG"
+    generation_llm_customization {
+      instruction = "instruction"
+    }
+  }
+}
+
+
+resource "oci_generative_ai_agent_tool" "sql_test_tool" {
+  agent_id       = var.test_agent_id
+  compartment_id = var.compartment_ocid
+  display_name   = var.test_agent_tool_display_name
+  description    = var.test_agent_tool_description
+  tool_config {
+    database_schema {
+      content             = "CREATE TABLE example ();"
+      input_location_type = "INLINE"
+    }
+    dialect                       = "SQL_LITE"
+    model_size                    = "SMALL"
+    should_enable_self_correction = false
+    should_enable_sql_execution   = false
+    tool_config_type              = "SQL_TOOL_CONFIG"
+  }
+}
+
+resource "oci_generative_ai_agent_tool" "fc_test_tool" {
+  agent_id       = var.test_agent_id
+  compartment_id = var.compartment_ocid
+  display_name   = var.test_agent_tool_display_name
+  description    = var.test_agent_tool_description
+  freeform_tags = {
+    Department = "Finance"
+  }
+  tool_config {
+    function {
+      description = "description"
+      name        = "name"
+      parameters = {
+        "parameters" = "parameters"
+      }
+    }
+    tool_config_type = "FUNCTION_CALLING_TOOL_CONFIG"
+  }
+}

examples/generative_ai_agent/variables.tf

@@ -53,6 +53,14 @@ variable "test_agent_endpoint_description" {
   default = "test agent endpoint"
 }
 
+variable "test_agent_tool_display_name" {
+  default = "test_agent_tool"
+}
+
+variable "test_agent_tool_description" {
+  default = "test agent tool"
+}
+
 variable "test_knowledge_base_display_name" {
   default = "test_knowledge_base"
 }
@@ -85,4 +93,74 @@ variable "test_freeform_tags" {
   default = {
     "bar-key" = "value"
   }
+}
+
+variable "should_enable_multi_modality"{
+  default = "true"
+}
+
+variable "test_agent_id" {}
+
+variable "test_knowledge_base_id" {}
+
+variable "test_bucket" {}
+
+variable "test_namespace" {}
+
+variable "test_prefix" {}
+
+variable "agent_endpoint_guardrail_config_content_moderation_config_input_guardrail_mode" {
+  default = "DISABLE"
+}
+
+variable "agent_endpoint_guardrail_config_content_moderation_config_output_guardrail_mode" {
+  default = "DISABLE"
+}
+
+variable "agent_endpoint_guardrail_config_personally_identifiable_information_config_input_guardrail_mode" {
+  default = "DISABLE"
+}
+
+variable "agent_endpoint_guardrail_config_personally_identifiable_information_config_output_guardrail_mode" {
+  default = "DISABLE"
+}
+
+variable "agent_endpoint_guardrail_config_prompt_injection_config_input_guardrail_mode" {
+  default = "DISABLE"
+}
+
+variable "agent_endpoint_human_input_config_should_enable_human_input" {
+  default = false
+}
+
+variable "agent_endpoint_metadata" {
+  default = "metadata"
+}
+
+variable "agent_endpoint_output_config_output_location_bucket" {
+  default = "bucket"
+}
+
+variable "agent_endpoint_output_config_output_location_namespace" {
+  default = "namespace"
+}
+
+variable "agent_endpoint_output_config_output_location_output_location_type" {
+  default = "OBJECT_STORAGE_PREFIX"
+}
+
+variable "agent_endpoint_output_config_output_location_prefix" {
+  default = "prefix"
+}
+
+variable "agent_endpoint_output_config_retention_period_in_minutes" {
+  default = 10
+}
+
+variable "agent_endpoint_should_enable_multi_language" {
+  default = false
+}
+
+variable "agent_llm_config_routing_llm_customization_instruction" {
+  default = "instruction"
 }