Class: RagRuby::Source

Inherits:
Object
  • Object
show all
Defined in:
lib/rag_ruby/source.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(chunk:, score:) ⇒ Source

Returns a new instance of Source.



7
8
9
10
11
 
# File 'lib/rag_ruby/source.rb', line 7

def initialize(chunk:, score:)
  @chunk = chunk
  @score = score
  @document_source = chunk.document_source
end

Instance Attribute Details

#chunk ⇒ Object (readonly)

Returns the value of attribute chunk.



5
6
7
 
# File 'lib/rag_ruby/source.rb', line 5

def chunk
  @chunk
end

#document_source ⇒ Object (readonly)

Returns the value of attribute document_source.



5
6
7
 
# File 'lib/rag_ruby/source.rb', line 5

def document_source
  @document_source
end

#score ⇒ Object (readonly)

Returns the value of attribute score.



5
6
7
 
# File 'lib/rag_ruby/source.rb', line 5

def score
  @score
end

Instance Method Details

#metadata ⇒ Object 



17
18
19
 
# File 'lib/rag_ruby/source.rb', line 17

def metadata
  chunk.metadata
end

#text ⇒ Object 



13
14
15
 
# File 'lib/rag_ruby/source.rb', line 13

def text
  chunk.text
end

#to_h ⇒ Object 



21
22
23
24
25
26
27
28
 
# File 'lib/rag_ruby/source.rb', line 21

def to_h
  {
    text: text,
    score: score,
    document_source: document_source,
    metadata: metadata
  }
end