Class: Langchain::LLM::AwsTitanResponse

Inherits:

BaseResponse

• Object
• BaseResponse
• Langchain::LLM::AwsTitanResponse

Defined in:

lib/langchain/llm/response/aws_titan_response.rb

Instance Attribute Summary

Attributes inherited from BaseResponse

#context, #model, #raw_response

Instance Method Summary

• #embedding ⇒ Object
• #embeddings ⇒ Object
• #prompt_tokens ⇒ Object

Methods inherited from BaseResponse

#chat_completion, #chat_completions, #completion, #completion_tokens, #completions, #created_at, #initialize, #total_tokens

Constructor Details

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

Instance Method Details

#embedding ⇒ Object

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

def embedding
  embeddings&.first
end

#embeddings ⇒ Object

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

def embeddings
  [raw_response.dig("embedding")]
end

#prompt_tokens ⇒ Object

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

def prompt_tokens
  raw_response.dig("inputTextTokenCount")
end