solace-agent-mesh 0.0.1__py3-none-any.whl → 0.1.1__py3-none-any.whl

solace_agent_mesh/configs/orchestrator.yaml

@@ -0,0 +1,241 @@
+# This is the configuration file for the orchestrator.
+#
+# It fulfills a few functions:
+# 1. It receives agent registration messages to allow it to know all the agents
+# 2. It receives stimuli from Slack (or other UIs) over the Solace broker
+#    and uses an LLM to break it down into a series of events.
+# 3. It sends action requests to agents over the Solace broker.
+# 4. It received action responses from agents over the Solace broker and
+#    processes them. They may result in a conversation response or a new action request.
+# 5. It also manages the conversation history
+
+---
+log:
+  stdout_log_level: INFO
+  log_file_level: INFO
+  log_file: solace_ai_connector.log
+
+# trace:
+#   trace_file: orchestrator.trace
+
+shared_config:
+  - broker_config: &broker_connection
+      broker_connection_share: ${SOLACE_BROKER_URL}
+      dev_mode: ${SOLACE_DEV_MODE, false}
+      broker_url: ${SOLACE_BROKER_URL}
+      broker_username: ${SOLACE_BROKER_USERNAME}
+      broker_password: ${SOLACE_BROKER_PASSWORD}
+      broker_vpn: ${SOLACE_BROKER_VPN}
+      temporary_queue: ${USE_TEMPORARY_QUEUES, false}
+
+# Take from Slack and publish to Solace
+flows:
+  # Receive registration messages from the agents
+  - name: orchestrator_register
+    components:
+      # Input from a Solace broker
+      - component_name: solace_sw_broker
+        component_module: broker_input
+        component_config:
+          <<: *broker_connection
+          broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}orchestrator_register
+          broker_subscriptions:
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/register/>
+              qos: 1
+          payload_encoding: utf-8
+          payload_format: json
+
+      # Custom component to process the registration
+      - component_name: registration
+        component_base_path: .
+        component_module: src.orchestrator.components.orchestrator_register_component
+        component_config:
+          agent_ttl_ms: 60000000
+        component_input:
+          source_expression: input.payload
+
+  # Slack input processing
+  - name: orchestrator_stimulus_input
+    components:
+      # Input from a Solace broker
+      - component_name: solace_sw_broker
+        component_module: broker_input
+        component_config:
+          <<: *broker_connection
+          broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}orchestrator_stimulus_input
+          broker_subscriptions:
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/stimulus/orchestrator/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/stimulus/gateway/>
+              qos: 1
+          payload_encoding: utf-8
+          payload_format: json
+
+      # Process the stimulus with a custom component
+      - component_name: orchestrator_stimulus_processor
+        num_instances: {{ORCHESTRATOR_INSTANCE_COUNT}}
+        component_base_path: .
+        component_module: src.orchestrator.components.orchestrator_stimulus_processor_component
+        component_config:
+          llm_service_topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/planning
+          llm_mode: stream
+          stream_to_flow: streaming_output
+          set_response_uuid_in_user_properties: true
+
+        broker_request_response:
+          enabled: true
+          broker_config: *broker_connection
+          request_expiry_ms: 120000
+          payload_encoding: utf-8
+          payload_format: json
+          response_topic_prefix: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/response
+          response_queue_prefix: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/response
+
+        component_input:
+          source_expression: input.payload
+
+      # The previous component created a list of action requests
+      # Separate them into individual action requests
+      - component_name: action_requestsplitter
+        component_module: iterate
+
+      # Send the action requests to the agents
+      - component_name: send_action_request
+        component_module: broker_output
+        component_config:
+          <<: *broker_connection
+          payload_encoding: utf-8
+          payload_format: json
+          copy_user_properties: true
+        input_transforms:
+          - type: copy
+            source_expression: previous:payload
+            dest_expression: user_data.output:payload
+          - type: copy
+            source_expression: previous:topic
+            dest_expression: user_data.output:topic
+        component_input:
+          source_expression: user_data.output
+
+  # Flow to handle action responses from the agents
+  - name: orchestrator_action_response
+    components:
+      # Input from a Solace broker
+      - component_name: solace_sw_broker
+        component_module: broker_input
+        component_config:
+          <<: *broker_connection
+          broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}orchestrator_action_response
+          broker_subscriptions:
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/actionResponse/>
+              qos: 1
+          payload_encoding: utf-8
+          payload_format: json
+
+      # Process the action response with a custom component
+      - component_name: action_response_processor
+        component_base_path: .
+        component_module: src.orchestrator.components.orchestrator_action_response_component
+        input_transforms:
+          - type: copy
+            source_expression: input.payload
+            dest_expression: user_data.component_input
+          - type: copy
+            source_expression: input.user_properties:session_id
+            dest_expression: user_data.component_input:session_id
+        component_input:
+          source_expression: user_data.component_input
+
+      # Break the action response into individual responses
+      - component_name: split_action_response
+        component_module: iterate
+
+      # Send to the broker
+      - component_name: broker_output
+        component_module: broker_output
+        component_config:
+          <<: *broker_connection
+          payload_encoding: utf-8
+          payload_format: json
+          copy_user_properties: true
+        component_input:
+          source_expression: previous
+
+  # Asynchronous streaming messages to the UIs - this is called directly by other flows
+  - name: streaming_output
+    components:
+      # Receive streaming messages from the LLM - this will be the raw yaml response
+      - component_name: process_streaming_llm_output
+        component_base_path: .
+        component_module: src.orchestrator.components.orchestrator_streaming_output_component
+        component_input:
+          source_expression: input.payload
+
+      # Break the streaming response into individual responses
+      - component_name: split_action_response
+        component_module: iterate
+
+      # Output to a Solace broker
+      - component_name: broker_output
+        component_module: broker_output
+        component_config:
+          <<: *broker_connection
+          payload_encoding: utf-8
+          payload_format: json
+          copy_user_properties: true
+        input_transforms:
+          - type: copy
+            source_expression: previous
+            dest_expression: user_data.output:payload
+          - type: copy
+            source_expression: input.user_properties
+            dest_expression: user_data.output:user_properties
+          - type: copy
+            source_expression:
+              invoke:
+                module: invoke_functions
+                function: if_else
+                params:
+                  positional:
+                    - source_expression(input.payload:response_complete)
+                    - source_expression(template:${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/responseComplete/orchestrator/{{text://input.user_properties:gateway_id}})
+                    - invoke:
+                        module: invoke_functions
+                        function: if_else
+                        params:
+                          positional:
+                            - source_expression(input.payload:status_update)
+                            - source_expression(template:${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/responseStatus/orchestrator/{{text://input.user_properties:gateway_id}})
+                            - source_expression(template:${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/streamingResponse/orchestrator/{{text://input.user_properties:gateway_id}})
+            dest_expression: user_data.output:topic
+        component_input:
+          source_expression: user_data.output
+
+  # Timer flow - just generate a timer event every 10 seconds to check on some timers
+  - name: action_manager_timer
+    components:
+      # Timer component
+      - component_name: timer
+        component_module: timer_input
+        component_config:
+          interval_ms: 10000
+
+      # Custom component to process the timer event
+      - component_name: action_manager_timeout_handler
+        component_base_path: .
+        component_module: src.orchestrator.components.orchestrator_action_manager_timeout_component
+
+      # Break the action response into individual responses indicating a timeout
+      - component_name: split_action_response
+        component_module: iterate
+
+      # Send to the broker
+      - component_name: broker_output
+        component_module: broker_output
+        component_config:
+          <<: *broker_connection
+          payload_encoding: utf-8
+          payload_format: json
+          copy_user_properties: true
+        component_input:
+          source_expression: previous

solace_agent_mesh/configs/service_embedding.yaml

@@ -0,0 +1,81 @@
+# This is the configuration file for the Embedding service.
+
+# The Embedding service provides integration with embedding models.
+# It will subscribe to Embedding service request events and based on the exact
+# topic, will call the appropriate Embedding service to generate a vector.
+
+# The general topic format is:
+# - <namespace>/solace-agent-mesh/v1/embedding-service/<verb>/<input-type>/<stimulus-uuid>/<session-uuid>/<originator-id>
+#
+# The specific subscriptions are:
+# - <namespace>/solace-agent-mesh/v1/embedding-service/request/text/>
+# - <namespace>/solace-agent-mesh/v1/embedding-service/request/image/>
+
+---
+log:
+  stdout_log_level: INFO
+  log_file_level: INFO
+  log_file: solace_agent_mesh.log
+
+shared_config:
+  - broker_config: &broker_connection
+      broker_connection_share: ${SOLACE_BROKER_URL}
+      dev_mode: ${SOLACE_DEV_MODE, false}
+      broker_url: ${SOLACE_BROKER_URL}
+      broker_username: ${SOLACE_BROKER_USERNAME}
+      broker_password: ${SOLACE_BROKER_PASSWORD}
+      broker_vpn: ${SOLACE_BROKER_VPN}
+      payload_encoding: utf-8
+      payload_format: json
+      temporary_queue: ${USE_TEMPORARY_QUEUES, false}
+
+flows:
+  # Handling text/image requests
+  - name: embedding-service
+    components:
+      - component_name: broker_input
+        component_module: broker_input
+        component_config:
+          <<: *broker_connection
+          broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/embedding-service/embedding
+          broker_subscriptions:
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/embedding-service/request/text/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/embedding-service/request/image/>
+              qos: 1
+
+      - component_name: file_resolver
+        component_base_path: .
+        component_module: src.tools.components.file_resolver_component
+        component_config:
+          force_resolve: true
+        input_selection:
+          source_expression: input.payload
+
+      - component_name: embedding_service_model
+        num_instances: 1
+        component_module: litellm_embeddings
+        component_config:
+          load_balancer:
+            - model_name: ${EMBEDDING_SERVICE_MODEL_NAME} # model alias
+              litellm_params:
+                model: ${EMBEDDING_SERVICE_MODEL_NAME}
+                api_key: ${EMBEDDING_SERVICE_API_KEY}
+                api_base: ${EMBEDDING_SERVICE_ENDPOINT}
+        input_selection:
+          source_expression: previous
+
+      - component_name: broker_output
+        component_module: broker_output
+        component_config:
+          <<: *broker_connection
+          copy_user_properties: true
+        input_transforms:
+          - type: copy
+            source_expression: previous
+            dest_expression: user_data.output:payload
+          - type: copy
+            source_expression: input.user_properties:__solace_ai_connector_broker_request_response_topic__
+            dest_expression: user_data.output:topic
+        input_selection:
+          source_expression: user_data.output

solace_agent_mesh/configs/service_llm.yaml

@@ -0,0 +1,265 @@
+# This is the configuration file for the LLM service.
+
+# The LLM service provides multiple integrations with a variety of large language models.
+# It will subscribe to LLM service request events and based on the exact
+# topic, will call the appropriate LLM service to generate a response.
+
+# The general topic format is:
+# - <namespace>/solace-agent-mesh/v1/llm-service/<verb>/<model-id-or-group>/<stimulus-uuid>/<session-uuid>/<originator-id>
+#
+# The specific subscriptions are:
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/planning/>
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/reasoning-expensive/>
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/reasoning-normal/>
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/general-good/>
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/general-fast/>
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/writing/>
+# - <namespace>/solace-agent-mesh/v1/llm-service/request/coding/>
+
+---
+log:
+  stdout_log_level: INFO
+  log_file_level: INFO
+  log_file: solace_agent_mesh.log
+
+shared_config:
+  - broker_config: &broker_connection
+      broker_connection_share: ${SOLACE_BROKER_URL}
+      dev_mode: ${SOLACE_DEV_MODE, false}
+      broker_url: ${SOLACE_BROKER_URL}
+      broker_username: ${SOLACE_BROKER_USERNAME}
+      broker_password: ${SOLACE_BROKER_PASSWORD}
+      broker_vpn: ${SOLACE_BROKER_VPN}
+      payload_encoding: utf-8
+      payload_format: json
+      temporary_queue: ${USE_TEMPORARY_QUEUES, false}
+
+  - llm_config: &llm_config
+      llm_mode: none
+      set_response_uuid_in_user_properties: true
+      stream_to_next_component: true
+
+  - llm_auth: &llm_auth
+      api_key: ${LLM_SERVICE_API_KEY}
+      base_url: ${LLM_SERVICE_ENDPOINT}
+
+  - llm_input_transform_and_select: &llm_input_transforms_and_select
+      input_selection:
+        source_expression: previous
+
+flows:
+  # Handling planning requests
+  - name: llm-service-planning
+    components:
+      - component_name: broker_input
+        component_module: broker_input
+        component_config:
+          <<: *broker_connection
+          broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/planning
+          broker_subscriptions:
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/planning/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/reasoning-expensive/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/reasoning-normal/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/general-good/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/general-fast/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/writing/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/coding/>
+              qos: 1
+
+      - component_name: file_resolver
+        component_base_path: .
+        component_module: src.tools.components.file_resolver_component
+        component_config:
+          force_resolve: true
+        input_selection:
+          source_expression: input.payload
+
+      - component_name: llm_service_planning
+        num_instances: 1
+        component_module: litellm_chat_model
+        component_config:
+          <<: *llm_config
+          load_balancer:
+            - model_name: ${LLM_SERVICE_PLANNING_MODEL_NAME}
+              litellm_params:
+                <<: *llm_auth
+                model: ${LLM_SERVICE_PLANNING_MODEL_NAME}
+                temperature: 0.02
+        <<: *llm_input_transforms_and_select
+
+      - component_name: broker_output
+        component_module: broker_output
+        component_config:
+          <<: *broker_connection
+          copy_user_properties: true
+        input_transforms:
+          - type: copy
+            source_expression: previous
+            dest_expression: user_data.output:payload
+          - type: copy
+            source_expression: input.user_properties:__solace_ai_connector_broker_request_response_topic__
+            dest_expression: user_data.output:topic
+        input_selection:
+          source_expression: user_data.output
+
+  # Handling reasoning requests
+  # - name: llm-service-reasoning-expensive
+  #   components:
+  #     - component_name: broker_input
+  #       component_module: broker_input
+  #       component_config:
+  #         <<: *broker_connection
+  #         broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/reasoning-expensive
+  #         broker_subscriptions:
+  #           - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/reasoning-expensive/>
+  #             qos: 1
+
+  #     - component_name: llm_service_reasoning_expensive
+  #       num_instances: 5
+  #       component_module: openai_chat_model
+  #       <<: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *llm_config
+  #         model: ${LLM_SERVICE_REASONING_EXPENSIVE_MODEL_NAME}
+  #         temperature: 0.01
+
+  #     - component_name: broker_output
+  #       component_module: broker_output
+  #       component_config:
+  #         <<: *broker_connection
+
+  # - name: llm-service-reasoning-normal
+  #   components:
+  #     - component_name: broker_input
+  #       component_module: broker_input
+  #       component_config:
+  #         <<: *broker_connection
+  #         broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/reasoning-normal
+  #         broker_subscriptions:
+  #           - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/reasoning-normal/>
+  #             qos: 1
+
+  #     - component_name: llm_service_reasoning_normal
+  #       num_instances: 5
+  #       component_module: openai_chat_model
+  #       <<: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *llm_config
+  #         model: ${LLM_SERVICE_REASONING_NORMAL_MODEL_NAME}
+  #         temperature: 0.01
+
+  #     - component_name: broker_output
+  #       component_module: broker_output
+  #       component_config:
+  #         <<: *broker_connection
+
+  # # Handling general-good requests
+  # - name: llm-service-general-good
+  #   components:
+  #     - component_name: broker_input
+  #       component_module: broker_input
+  #       component_config:
+  #         <<: *broker_connection
+  #         broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/general-good
+  #         broker_subscriptions:
+  #           - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/general-good/>
+  #             qos: 1
+
+  #     - component_name: llm_service_general_good
+  #       num_instances: 5
+  #       component_module: openai_chat_model
+  #       <<: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *llm_config
+  #         model: ${LLM_SERVICE_GENERAL_GOOD_MODEL_NAME}
+  #         temperature: 0.01
+
+  #     - component_name: broker_output
+  #       component_module: broker_output
+  #       component_config:
+  #         <<: *broker_connection
+
+  # # Handling general-fast requests
+  # - name: llm-service-general-fast
+  #   components:
+  #     - component_name: broker_input
+  #       component_module: broker_input
+  #       llm_input_transform_and_select: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *broker_connection
+  #         broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/general-fast
+  #         broker_subscriptions:
+  #           - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/general-fast/>
+  #             qos: 1
+
+  #     - component_name: llm_service_general_fast
+  #       num_instances: 5
+  #       component_module: openai_chat_model
+  #       <<: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *llm_config
+  #         model: ${LLM_SERVICE_GENERAL_FAST_MODEL_NAME}
+  #         temperature: 0.01
+
+  #     - component_name: broker_output
+  #       component_module: broker_output
+  #       component_config:
+  #         <<: *broker_connection
+
+  # # Handling writing requests
+  # - name: llm-service-writing
+  #   components:
+  #     - component_name: broker_input
+  #       component_module: broker_input
+  #       component_config:
+  #         <<: *broker_connection
+  #         broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/writing
+  #         broker_subscriptions:
+  #           - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/writing/>
+  #             qos: 1
+
+  #     - component_name: llm_service_writing
+  #       num_instances: 5
+  #       component_module: openai_chat_model
+  #       <<: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *llm_config
+  #         model: ${LLM_SERVICE_WRITING_MODEL_NAME}
+  #         temperature: 0.7
+
+  #     - component_name: broker_output
+  #       component_module: broker_output
+  #       component_config:
+  #         <<: *broker_connection
+
+  # # Handling coding requests
+  # - name: llm-service-coding
+  #   components:
+  #     - component_name: broker_input
+  #       component_module: broker_input
+  #       component_config:
+  #         <<: *broker_connection
+  #         broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/coding
+  #         broker_subscriptions:
+  #           - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/llm-service/request/coding/>
+  #             qos: 1
+
+  #     - component_name: llm_service_coding
+  #       num_instances: 5
+  #       component_module: openai_chat_model
+  #       <<: *llm_input_transforms_and_select
+  #       component_config:
+  #         <<: *llm_config
+  #         model: ${LLM_SERVICE_CODING_MODEL_NAME}
+  #         temperature: 0.01
+
+  #     - component_name: broker_output
+  #       component_module: broker_output
+  #       component_config:
+  #         <<: *broker_connection

solace_agent_mesh/configs/visualize_websocket.yaml

@@ -0,0 +1,55 @@
+---
+log:
+  stdout_log_level: INFO
+  log_file_level: INFO
+  log_file: solace_ai_connector.log
+
+shared_config:
+  - broker_config: &broker_connection
+      broker_connection_share: ${SOLACE_BROKER_URL}
+      dev_mode: ${SOLACE_DEV_MODE, false}
+      broker_url: ${SOLACE_BROKER_URL}
+      broker_username: ${SOLACE_BROKER_USERNAME}
+      broker_password: ${SOLACE_BROKER_PASSWORD}
+      broker_vpn: ${SOLACE_BROKER_VPN}
+      temporary_queue: ${USE_TEMPORARY_QUEUES, false}
+
+flows:
+  - name: visualize_websocket
+    components:
+      - component_name: broker_input
+        component_module: broker_input
+        component_config: 
+          <<: *broker_connection
+          broker_queue_name: ${SOLACE_AGENT_MESH_NAMESPACE}visualize_websocket
+          broker_subscriptions:
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/register/>
+              qos: 1
+            - topic: ${SOLACE_AGENT_MESH_NAMESPACE}solace-agent-mesh/v1/>
+              qos: 1
+          payload_encoding: utf-8
+          payload_format: json
+
+      - component_name: websocket_output
+        component_module: websocket_output
+        component_config:
+          listen_port: ${VISUALIZER_WEBSOCKET_PORT, 5000}
+          serve_html: false
+          payload_encoding: none
+          payload_format: json
+        input_transforms:
+          - type: copy
+            source_expression: input.payload
+            dest_expression: user_data.output:payload.payload
+          - type: copy
+            source_expression: input.topic
+            dest_expression: user_data.output:payload.topic
+          - type: copy
+            source_expression: input.user_properties
+            dest_expression: user_data.output:payload.user_properties
+          - type: copy
+            source_value: "*"
+            dest_expression: user_data.output:socket_id
+        component_input:
+          source_expression: user_data.output
+

solace_agent_mesh/gateway/components/gateway_base.py

@@ -0,0 +1,41 @@
+from solace_ai_connector.components.component_base import ComponentBase
+from solace_ai_connector.common.log import log
+from ...services.history_service import HistoryService
+from ..identity.identity_provider import IdentityProvider
+from ...common.constants import DEFAULT_IDENTITY_KEY_FIELD
+
+
+class GatewayBase(ComponentBase):
+    def __init__(self, info, **kwargs):
+        super().__init__(info, **kwargs)
+        self.gateway_id = self.get_config("gateway_id", "default-change-me")
+        self.history_instance = self._initialize_history()
+
+    def _initialize_history(self):
+        self.use_history = self.get_config("retain_history", True)
+
+        if not self.use_history:
+            return None
+
+        history_config = self.get_config("history_config", {})
+
+        try:
+            return HistoryService(
+                history_config, identifier=self.gateway_id + "_history"
+            )
+        except (ImportError, AttributeError, ValueError) as e:
+            log.error("Failed to load history class: %s", e)
+            raise
+
+    def _initialize_identity_component(self):
+        identity_config = self.get_config("identity", {})
+        identity_key_field = self.get_config("identity_key_field", DEFAULT_IDENTITY_KEY_FIELD)
+        
+        if "key_field" not in identity_config:
+            identity_config["key_field"] = identity_key_field
+
+        try:
+            return IdentityProvider(identity_config)
+        except (ImportError, AttributeError, ValueError) as e:
+            log.error("Failed to load identity class: %s", e)
+            raise