mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 07:38:29 +00:00
1a1bb0ca3d
* docs(cli): document device-code login and config reset guidance; renumber sections * docs(cli): fix duplicate numbering (renumber Login/API Keys/Configuration sections) * docs: Fix webhook documentation to include meta dict in all webhook payloads - Add note explaining that meta objects from kickoff requests are included in all webhook payloads - Update webhook examples to show proper payload structure including meta field - Fix webhook examples to match actual API implementation - Apply changes to English, Korean, and Portuguese documentation Resolves the documentation gap where meta dict passing to webhooks was not documented despite being implemented in the API. * WIP: CrewAI docs theme, changelog, GEO, localization * docs(cli): fix merge markers; ensure mode: "wide"; convert ASCII tables to Markdown (en/pt-BR/ko) * docs: add group icons across locales; split Automation/Integrations; update tools overviews and links
83 lines
3.3 KiB
Plaintext
83 lines
3.3 KiB
Plaintext
---
|
|
title: PG RAG Search
|
|
description: The `PGSearchTool` is designed to search PostgreSQL databases and return the most relevant results.
|
|
icon: elephant
|
|
mode: "wide"
|
|
---
|
|
|
|
## Overview
|
|
|
|
<Note>
|
|
The PGSearchTool is currently under development. This document outlines the intended functionality and interface.
|
|
As development progresses, please be aware that some features may not be available or could change.
|
|
</Note>
|
|
|
|
## Description
|
|
|
|
The PGSearchTool is envisioned as a powerful tool for facilitating semantic searches within PostgreSQL database tables. By leveraging advanced Retrieve and Generate (RAG) technology,
|
|
it aims to provide an efficient means for querying database table content, specifically tailored for PostgreSQL databases.
|
|
The tool's goal is to simplify the process of finding relevant data through semantic search queries, offering a valuable resource for users needing to conduct advanced queries on
|
|
extensive datasets within a PostgreSQL environment.
|
|
|
|
## Installation
|
|
|
|
The `crewai_tools` package, which will include the PGSearchTool upon its release, can be installed using the following command:
|
|
|
|
```shell
|
|
pip install 'crewai[tools]'
|
|
```
|
|
|
|
<Note>
|
|
The PGSearchTool is not yet available in the current version of the `crewai_tools` package. This installation command will be updated once the tool is released.
|
|
</Note>
|
|
|
|
## Example Usage
|
|
|
|
Below is a proposed example showcasing how to use the PGSearchTool for conducting a semantic search on a table within a PostgreSQL database:
|
|
|
|
```python Code
|
|
from crewai_tools import PGSearchTool
|
|
|
|
# Initialize the tool with the database URI and the target table name
|
|
tool = PGSearchTool(
|
|
db_uri='postgresql://user:password@localhost:5432/mydatabase',
|
|
table_name='employees'
|
|
)
|
|
```
|
|
|
|
## Arguments
|
|
|
|
The PGSearchTool is designed to require the following arguments for its operation:
|
|
|
|
| Argument | Type | Description |
|
|
|:---------------|:---------|:-------------------------------------------------------------------------------------------------------------------------------------|
|
|
| **db_uri** | `string` | **Mandatory**. A string representing the URI of the PostgreSQL database to be queried. This argument will be mandatory and must include the necessary authentication details and the location of the database. |
|
|
| **table_name** | `string` | **Mandatory**. A string specifying the name of the table within the database on which the semantic search will be performed. This argument will also be mandatory. |
|
|
|
|
## Custom Model and Embeddings
|
|
|
|
The tool intends to use OpenAI for both embeddings and summarization by default. Users will have the option to customize the model using a config dictionary as follows:
|
|
|
|
```python Code
|
|
tool = PGSearchTool(
|
|
config=dict(
|
|
llm=dict(
|
|
provider="ollama", # or google, openai, anthropic, llama2, ...
|
|
config=dict(
|
|
model="llama2",
|
|
# temperature=0.5,
|
|
# top_p=1,
|
|
# stream=true,
|
|
),
|
|
),
|
|
embedder=dict(
|
|
provider="google", # or openai, ollama, ...
|
|
config=dict(
|
|
model="models/embedding-001",
|
|
task_type="retrieval_document",
|
|
# title="Embeddings",
|
|
),
|
|
),
|
|
)
|
|
)
|
|
``` |