Module: StubOllamaServer

Defined in:
spec/spec_helper.rb

Overview

A module that provides functionality for stubbing Ollama server responses.

The StubOllamaServer module enables developers to simulate Ollama API interactions in test environments by intercepting requests and returning predefined responses. This allows for faster, more reliable testing without requiring external service calls.

Instance Method Summary collapse

Instance Method Details

#connect_to_ollama_server(instantiate: true) ⇒ Object

The connect_to_ollama_server method establishes a connection to an Ollama server.

This method sets up stubbed HTTP requests to simulate responses from an Ollama server, including API tags, show, and version endpoints. It can optionally instantiate a chat session after setting up the stubs.

Parameters:

  • instantiate (Boolean) (defaults to: true) —

    whether to instantiate a chat session after setting up stubs



104
105
106
107
108
109
110
111
112
113
114
115
 
# File 'spec/spec_helper.rb', line 104

def connect_to_ollama_server(instantiate: true)
  before do
    stub_request(:get, %r(/api/tags\z)).
      to_return(status: 200, body: asset_json('api_tags.json'))
    stub_request(:post, %r(/api/show\z)).
      to_return(status: 200, body: asset_json('api_show.json'))
    stub_request(:get, %r(/api/version\z)).
      to_return(status: 200, body: asset_json('api_version.json'))
    allow_any_instance_of(OllamaChat::Chat).to receive(:connect_message)
    instantiate and chat
  end
end