Skip to main content

MCP based Pocket Agent in vertor db as storage

Project description

Pocket Assistant MCP Server

A Model Context Protocol (MCP) server that provides pocket assistance capabilities with ChromaDB vector storage. This server enables AI assistants to save, retrieve, and manage research content efficiently using vector embeddings.

Features

  • Vector Storage: Uses ChromaDB for efficient storage and retrieval
  • Topic Organization: Organize research content by topics
  • Deduplication: Automatic content deduplication using hashing
  • Semantic Search: Query research content using natural language
  • Multiple Topics: Manage multiple research topics simultaneously
  • OpenAI Embeddings: Uses OpenAI's text-embedding-3-small model

Installation

Using uvx (Recommended)

uvx pocket-agent-mcp

Using uv

uv pip install pocket-agent-mcp

Using pip

pip install pocket-agent-mcp

From Source

git clone https://github.com/VikashS/pocket_agent_mcp.git
cd pocket-agent-mcp
uv pip install -e .

Configuration

Environment Variables

Required:

  • OPENAI_API_KEY - Your OpenAI API key for embeddings
  • RESEARCH_DB_PATH - Base path for storing research databases
    • A pocket_chroma_dbs directory will be created inside this path
    • Example: /path/to/data (will create /path/to/data/pocket_chroma_dbs)
    • Example: ~/.pocket-agent-mcp (will create ~/.pocket-agent-mcp/pocket_chroma_dbs)

Create a .env file with your configuration:

OPENAI_API_KEY=your-api-key-here
RESEARCH_DB_PATH=/path/to/data

Claude Desktop Configuration

MacOS: ~/Library/Application Support/Claude/claude_desktop_config.json Windows: %APPDATA%\Claude\claude_desktop_config.json

{
  "mcpServers": {
    "research-assistant": {
      "command": "uvx",
      "args": ["pocket-agent-mcp"],
      "env": {
        "OPENAI_API_KEY": "your-api-key-here",
        "POCKET_DB_PATH": "/path/to/data"
      }
    }
  }
}

Note: Both OPENAI_API_KEY and POCKET_DB_PATH are required. The database will be stored in POCKET_DB_PATH/pocket_chroma_dbs/.

Available Tools

1. save_research_data

Save research content to vector database for future retrieval.

Parameters:

  • content (List[str]): List of text content to save
  • topic (str): Topic name for organizing the data (creates separate DB)

Example:

Save these research findings about AI to the "artificial-intelligence" topic

2. query_research_data

Query saved research content using natural language.

Parameters:

  • query (str): Natural language query
  • topic (str): Topic to search in (default: "default")
  • k (int): Number of results to return (default: 5)

Example:

Query the "artificial-intelligence" topic for information about transformers

3. list_topics

List all available research topics and their document counts.

Example:

List all available research topics

4. delete_topic

Delete a research topic and all its associated data.

Parameters:

  • topic (str): Topic name to delete

Example:

Delete the "old-research" topic

5. get_topic_info

Get detailed information about a specific topic.

Parameters:

  • topic (str): Topic name

Example:

Get information about the "artificial-intelligence" topic

Usage Examples

Once configured with Claude Desktop or another MCP client, you can:

  • "Save this article about machine learning to my 'ml-research' topic"
  • "Query my 'ml-research' for information about neural networks"
  • "List all my research topics"
  • "Get information about the 'quantum-computing' topic"
  • "Delete the 'old-notes' topic"

Technical Details

  • Protocol: Model Context Protocol (MCP)
  • Transport: stdio
  • Vector Database: ChromaDB
  • Embeddings: OpenAI text-embedding-3-small
  • Storage: Local filesystem at POCKET_DB_PATH/pocket_chroma_dbs/

Requirements

  • Python 3.11 or higher
  • OpenAI API key
  • Dependencies: chromadb, langchain, fastmcp, openai

Development

Setup Development Environment

# Clone the repository
git clone https://github.com/VikashS/pocket_agent_mcp.git
cd pocket_agent_mcp

# Install with development dependencies
uv pip install -e .

License

This project is licensed under the MIT License - see the LICENSE file for details.

Author

Vikash Singh

Acknowledgments

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pocket_agent_mcp-0.1.0.tar.gz (11.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pocket_agent_mcp-0.1.0-py3-none-any.whl (9.7 kB view details)

Uploaded Python 3

File details

Details for the file pocket_agent_mcp-0.1.0.tar.gz.

File metadata

  • Download URL: pocket_agent_mcp-0.1.0.tar.gz
  • Upload date:
  • Size: 11.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.4

File hashes

Hashes for pocket_agent_mcp-0.1.0.tar.gz
Algorithm Hash digest
SHA256 5e56be077d5101c36608e4fcc168aa172fc767730f84f9f5c54a65d05c189110
MD5 fcec78f40ae8a90cdb9ac0c30722624c
BLAKE2b-256 e80c78233a7db569b90220f8fdbb8a0db29ecefcabbee25ba8d31a419a5c3d1c

See more details on using hashes here.

File details

Details for the file pocket_agent_mcp-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for pocket_agent_mcp-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 3b5f367411a196e9be05d3e34618812b0e963807530f9b944cf212613eca2184
MD5 300f43968935d8911c9e594904a2f9f6
BLAKE2b-256 ad71d0d7f43e6662bb95928cea70a24f31584d6472190eaed340895308069954

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page