Merge pull request #1028 from code0-tech/#1013-query-available-models

query velorum models

Author: Taucher2003

app/graphql/types/query_type.rb

@@ -43,6 +43,7 @@ class QueryType < Types::BaseObject
     field :users, Types::UserType.connection_type, null: false, description: 'Find users'
 
     field :global_runtimes, Types::RuntimeType.connection_type, null: false, description: 'Find runtimes'
+    field :velorum, Types::VelorumType, null: true, description: 'Get Velorum information'
 
     def application
       {}
@@ -86,6 +87,10 @@ def global_runtimes
       Runtime.where(namespace: nil)
     end
 
+    def velorum
+      {}
+    end
+
     def current_authentication
       super.authentication
     end

app/graphql/types/velorum_model_type.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Types
+  class VelorumModelType < Types::BaseObject
+    description 'Represents a model available through Velorum'
+
+    field :identifier, String, null: false, description: 'Unique model identifier'
+    field :name, String, null: false, description: 'Human-readable model name'
+    field :token_cost, Float, null: false, description: 'Token cost for using this model'
+    field :types, [Types::VelorumModelTypeEnum],
+          null: false,
+          description: 'Capabilities supported by this model',
+          method: :type
+  end
+end

app/graphql/types/velorum_model_type_enum.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Types
+  class VelorumModelTypeEnum < Types::BaseEnum
+    description 'Supported Velorum model capabilities'
+
+    value :UNKNOWN, 'Unknown model capability', value: :UNKNOWN
+    value :EXPLAIN, 'Model can explain flows', value: :EXPLAIN
+    value :GENERATE, 'Model can generate flows', value: :GENERATE
+  end
+end

app/graphql/types/velorum_type.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Types
+  class VelorumType < Types::BaseObject
+    description 'Represents Velorum integration information'
+
+    authorize :read_velorum_config
+    declarative_policy_subject { :global }
+
+    field :enabled, Boolean, null: false, description: 'Whether Velorum is enabled'
+    field :models, [Types::VelorumModelType], null: false, description: 'Find models available through Velorum'
+
+    def enabled
+      config[:enabled]
+    end
+
+    def models
+      ::Velorum::ModelsService.new(config: config).execute
+    end
+
+    private
+
+    def config
+      Sagittarius::Configuration.config[:velorum]
+    end
+  end
+end

app/policies/global_policy.rb

@@ -12,6 +12,7 @@ class GlobalPolicy < BasePolicy
     enable :read_flow_type
     enable :read_flow_type_setting
     enable :read_metadata
+    enable :read_velorum_config
   end
 
   rule { admin }.policy do

app/services/velorum/models_service.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Velorum
+  class ModelsService
+    def initialize(client: nil, config: Sagittarius::Configuration.config[:velorum])
+      @client = client
+      @config = config
+    end
+
+    def execute
+      return [] unless config[:enabled]
+
+      client.models.models
+    end
+
+    private
+
+    attr_reader :config
+
+    def client
+      @client ||= Sagittarius::Velorum::Client.new
+    end
+  end
+end

config/initializers/tucana.rb

@@ -2,4 +2,5 @@
 
 Rails.application.config.to_prepare do
   Tucana.load_protocol(:sagittarius)
+  Tucana.load_protocol(:velorum)
 end

config/sagittarius.example.yml

@@ -26,4 +26,10 @@ rails:
       key_derivation_salt: Z6zcLTgobXLYjXUslRsLMKxvXKq3j6DJ
   secret_key_base: MVMD6CtQwEWrQ28TdokQakbG2FG5abOn
 
+velorum:
+  enabled: false
+  host: 'localhost:50052'
+  jwt_secret:
+  jwt_ttl_minutes: 8
+
 application_setting_overrides: {}

docs/graphql/enum/velorummodeltype.md

@@ -0,0 +1,11 @@
+---
+title: VelorumModelType
+---
+
+Supported Velorum model capabilities
+
+| Value | Description |
+|-------|-------------|
+| `EXPLAIN` | Model can explain flows |
+| `GENERATE` | Model can generate flows |
+| `UNKNOWN` | Unknown model capability |

docs/graphql/object/query.md

@@ -14,6 +14,7 @@ Root Query type
 | `globalRuntimes` | [`RuntimeConnection!`](../object/runtimeconnection.md) | Find runtimes |
 | `organizations` | [`OrganizationConnection!`](../object/organizationconnection.md) | Find organizations |
 | `users` | [`UserConnection!`](../object/userconnection.md) | Find users |
+| `velorum` | [`Velorum`](../object/velorum.md) | Get Velorum information |
 
 ## Fields with arguments