Class: Langchain::LLM::CohereResponse

Inherits:

BaseResponse

• Object
• BaseResponse
• Langchain::LLM::CohereResponse

Defined in:

lib/langchain/llm/response/cohere_response.rb

Instance Attribute Summary

Attributes inherited from BaseResponse

#context, #model, #raw_response

Instance Method Summary

• #completion ⇒ Object
• #completions ⇒ Object
• #embedding ⇒ Object
• #embeddings ⇒ Object

Methods inherited from BaseResponse

#chat_completion, #chat_completions, #completion_tokens, #created_at, #initialize, #prompt_tokens, #total_tokens

Constructor Details

This class inherits a constructor from Langchain::LLM::BaseResponse

Instance Method Details

#completion ⇒ Object

# File 'lib/langchain/llm/response/cohere_response.rb', line 17

def completion
  completions&.dig(0, "text")
end

#completions ⇒ Object

# File 'lib/langchain/llm/response/cohere_response.rb', line 13

def completions
  raw_response.dig("generations")
end

#embedding ⇒ Object

# File 'lib/langchain/llm/response/cohere_response.rb', line 5

def embedding
  embeddings.first
end

#embeddings ⇒ Object

# File 'lib/langchain/llm/response/cohere_response.rb', line 9

def embeddings
  raw_response.dig("embeddings")
end