Class: OpenApiSDK::RagQuery

Inherits:
Object
  • Object
show all
Extended by:
T::Sig
Defined in:
lib/open_api_sdk/rag_query.rb

Instance Method Summary collapse

Constructor Details

#initialize(sdk_config) ⇒ RagQuery

Returns a new instance of RagQuery.



17
18
19
 
# File 'lib/open_api_sdk/rag_query.rb', line 17

def initialize(sdk_config)
  @sdk_configuration = sdk_config
end

Instance Method Details

#query(x_connection_token, query_body) ⇒ Object

Raises:

  • (StandardError) 


23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/open_api_sdk/rag_query.rb', line 23

def query(x_connection_token, query_body)
  # query - Query using RAG Search
  # Query across your connected data sources using RAG Search
  request = ::OpenApiSDK::Operations::QueryRequest.new(
    
    x_connection_token: x_connection_token,
    query_body: query_body
  )
  url, params = @sdk_configuration.get_server_details
  base_url = Utils.template_url(url, params)
  url = "#{base_url}/rag/query"
  headers = Utils.get_headers(request)
  req_content_type, data, form = Utils.serialize_request_body(request, :query_body, :json)
  headers['content-type'] = req_content_type
  raise StandardError, 'request body is required' if data.nil? && form.nil?
  headers['Accept'] = 'application/json'
  headers['user-agent'] = @sdk_configuration.user_agent

  r = @sdk_configuration.client.post(url) do |req|
    req.headers = headers
    Utils.configure_request_security(req, @sdk_configuration.security) if !@sdk_configuration.nil? && !@sdk_configuration.security.nil?
    if form
      req.body = Utils.encode_form(form)
    elsif Utils.match_content_type(req_content_type, 'application/x-www-form-urlencoded')
      req.body = URI.encode_www_form(data)
    else
      req.body = data
    end
  end

  content_type = r.headers.fetch('Content-Type', 'application/octet-stream')

  res = ::OpenApiSDK::Operations::QueryResponse.new(
    status_code: r.status, content_type: content_type, raw_response: r
  )
  if r.status == 201
    if Utils.match_content_type(content_type, 'application/json')
      out = Utils.unmarshal_complex(r.env.response_body, ::OpenApiSDK::Operations::QueryResponseBody)
      res.object = out
    end
  end
  res
end