Ziya

Documentation

See the GitHub Repository

Overview

Ziya is a code assist tool for AWS Bedrock models. It can read your entire codebase and answer questions.

The current version only performs read operations. However, future versions will be able to:

Write and edit code.
Search the web for resources.
Run commands locally.
Iteratively continue to do 1,2,3 for a given objective.

Pre-requisites

Setup AWS credentials:

The easiest way is to set the env variables with access to AWS Bedrock claude models.

export AWS_ACCESS_KEY_ID=<YOUR-KEY>
export AWS_SECRET_ACCESS_KEY=<YOUR-SECRET>

Installation

pip install ziya

Run Ziya

ziya

Then navigate to http://localhost:6969 in your browser and start chatting with your codebase.

When you ask a question Ziya sends your entire codebase as context to the LLM, along with your question and any chat history.

> Entering new AgentExecutor chain...
Reading user's current codebase: /Users/vkrishnaprasad/personal_projects/ziya
ziya
    ├── .gitignore
    ├── DEVELOPMENT.md
    ├── LICENSE
    ├── README.md
    └── pyproject.toml
    app
        ├── __init__.py
        ├── main.py
        └── server.py
...

Options

--exclude: Comma-separated list of files or directories or file suffix patterns to exclude from the codebase. Eg: "--exclude 'tst,build,*.py'"

--profile: AWS profile to use for the Bedrock LLM.

--model: The AWS Bedrock Model to use, one of sonnet(default), sonnet3.5, haiku or opus.

--port: The port number for frontend app. Default is 6969.

ziya --include='app,src/mappers' --exclude='tst,build,*.py' --profile=ziya --model=sonnet --port=8080

Anibus/ziya