Retrieve from Knowledge Base
client.Retrieve.Documents(ctx, knowledgeBaseID, body) (*RetrieveDocumentsResponse, error)
/{knowledgeBaseId}/retrieve
Retrieve relevant documents from a knowledge base using semantic search.
This endpoint:
- Authenticates the request using the provided bearer token
- Generates embeddings for the query using the knowledge base's configured model
- Performs vector similarity search in the knowledge base
- Returns the most relevant document chunks
The search supports hybrid search combining:
- Vector similarity (semantic search)
- Keyword matching (BM25)
- Custom metadata filters
Parameters
knowledgeBaseID string
formatuuid
Returns
Retrieve from Knowledge Base
package main
import (
"context"
"fmt"
"github.com/stainless-sdks/-go"
)
func main() {
client := gradient.NewClient(
)
response, err := client.Retrieve.Documents(
context.TODO(),
"550e8400-e29b-41d4-a716-446655440000",
gradient.RetrieveDocumentsParams{
NumResults: gradient.F(int64(5)),
Query: gradient.F("What are the best practices for deploying machine learning models?"),
},
)
if err != nil {
panic(err.Error())
}
fmt.Printf("%+v\n", response.Results)
}
{
"results": [
{
"metadata": {
"source": "bar",
"page": "bar",
"category": "bar",
"timestamp": "bar"
},
"text_content": "Machine learning models should be deployed with proper monitoring and versioning..."
}
],
"total_results": 5
}Returns Examples
{
"results": [
{
"metadata": {
"source": "bar",
"page": "bar",
"category": "bar",
"timestamp": "bar"
},
"text_content": "Machine learning models should be deployed with proper monitoring and versioning..."
}
],
"total_results": 5
}