spec/ollama/commands/generate_spec.rb in ollama-ruby-0.0.0 vs spec/ollama/commands/generate_spec.rb in ollama-ruby-0.0.1

@@ -16,11 +16,11 @@
     )
   end
 
   it 'can perform' do
     generate = described_class.new(model: 'llama3.1', prompt: 'Hello World', stream: true)
-    generate.client = client = double('client')
-    expect(client).to receive(:request).
+    generate.client = ollama = double('Ollama::Client')
+    expect(ollama).to receive(:request).
       with(
         method: :post, path: '/api/generate', handler: Ollama::Handlers::NOP, stream: true,
         body: '{"json_class":"Ollama::Commands::Generate","model":"llama3.1","prompt":"Hello World","stream":true}'
       )
     generate.perform(Ollama::Handlers::NOP)