Class: OmniAgent::Providers::Mock

Inherits:
Base
  • Object
show all
Defined in:
lib/omni_agent/providers/mock.rb

Constant Summary collapse

LOREM_IPSUM =
"Lorem ipsum dolor sit amet, consectetur adipiscing elit."

Instance Attribute Summary

Attributes inherited from Base

#model

Instance Method Summary collapse

Methods inherited from Base

#initialize, #validate_messages!

Constructor Details

This class inherits a constructor from OmniAgent::Providers::Base

Instance Method Details

#chat(messages:, tools: [], **_options) ⇒ Object



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# File 'lib/omni_agent/providers/mock.rb', line 6

def chat(messages:, tools: [], **_options)
  validate_messages!(messages, allowed_roles: %i[system user assistant tool])

  OmniAgent::Providers::Response.new(
    content: LOREM_IPSUM,
    raw_request: {
      model: model,
      messages: messages,
      tools: tools
    },
    raw_response: {
      "choices" => [
        {
          "message" => {
            "content" => LOREM_IPSUM
          }
        }
      ]
    },
    tool_calls: []
  )
end