https://github.com/harsh-vardhhan/ai-agent-flight-scanner
AI agent to search Google Flights data
https://github.com/harsh-vardhhan/ai-agent-flight-scanner
agentic-ai chromadb deepseek-r1 langchain llm rag sqlite
Last synced: 6 months ago
JSON representation
AI agent to search Google Flights data
- Host: GitHub
- URL: https://github.com/harsh-vardhhan/ai-agent-flight-scanner
- Owner: harsh-vardhhan
- Created: 2025-01-02T13:28:29.000Z (10 months ago)
- Default Branch: main
- Last Pushed: 2025-02-19T17:45:14.000Z (8 months ago)
- Last Synced: 2025-03-25T15:34:55.621Z (7 months ago)
- Topics: agentic-ai, chromadb, deepseek-r1, langchain, llm, rag, sqlite
- Language: Python
- Homepage:
- Size: 742 KB
- Stars: 46
- Watchers: 2
- Forks: 6
- Open Issues: 0
-
Metadata Files:
- Readme: readme.md
Awesome Lists containing this project
README
# RAG on Flight Data
## Frontend Repo
[https://github.com/harsh-vardhhan/ai-agent-flight-scanner-frontend](https://github.com/harsh-vardhhan/ai-agent-flight-scanner-frontend)
## Technical spec
| Spec | |
|----------------------------------------- |-----------|
| Platform to run large LLM | Groq |
| Platform to run small LLM | Ollama |
| LLM for SQL | deepseek-r1-distill-llama-70b |
| LLM for Vector Database | Llama 3.2:3B |
| AI agent framework | LangChain |
| SQL Database | SQLite |
| Vector Database | Chroma |
| REST framework | FastAPI |
## Application architecture
## Create `.env` file and set environment variables
```python
GROQ_API_KEY=your_groq_api_key_here
OPENAI_API_KEY=your_openai_api_key_here
DEEPSEEK_API_KEY=your_deepseek_api_key_here
```
## Running application
```
python3 app/main.py
```
## Prompt testing
### Basic Price Queries (India to Vietnam)
| Prompt |
|---------------------------------------------------------------------------------------------|
| What is the cheapest flight from New Delhi to Hanoi? |
| Find the lowest price flight from Mumbai to Ho Chi Minh City |
| Show me the cheapest flight from New Delhi to Da Nang |
| What is the lowest fare from Mumbai to Phu Quoc? |
### Basic Price Queries (Vietnam to India)
| Prompt |
|---------------------------------------------------------------------------------------------|
| What is the cheapest flight from Hanoi to New Delhi? |
| Find the lowest price flight from Ho Chi Minh City to Mumbai |
| Show me the cheapest flight from Da Nang to New Delhi |
| What is the lowest fare from Phu Quoc to Mumbai? |
### Price Range Queries (Generic)
| Prompt |
|---------------------------------------------------------------------------------------------|
| Show me flights from New Delhi to Hanoi ordered by price |
| List all flights from Ho Chi Minh City to Mumbai from lowest to highest price |
| What are the available flights from Mumbai to Da Nang sorted by fare? |
| Find flights from Phu Quoc to New Delhi ordered by cost |
### Flight Type Specific
| Prompt |
|---------------------------------------------------------------------------------------------|
| Show me all direct flights from New Delhi to Ho Chi Minh City |
| List connecting flights from Hanoi to Mumbai |
| What types of flights are available from New Delhi to Da Nang? |
| Find direct flights from Phu Quoc to Mumbai |
### Comparative Queries
| Prompt |
|---------------------------------------------------------------------------------------------|
| Compare prices of flights from New Delhi to all Vietnamese cities |
| Show me the cheapest routes from Mumbai to Vietnam |
| List all flight options from Hanoi to Indian cities |
| Compare fares from Ho Chi Minh City to Indian destinations |
### Round Trip Queries
| Prompt |
|---------------------------------------------------------------------------------------------|
| Find the cheapest round trip from New Delhi to Hanoi |
| Show me round trip options between Mumbai and Ho Chi Minh City |
| What are the most affordable round trip flights from New Delhi to Da Nang? |
| List round trip flights between Mumbai and Phu Quoc |
| List cheapest round trip flights between Mumbai and Phu Quoc |
| Find the cheapest return flight between New Delhi and Hanoi with at least 7 days gap |
| Show exactly one cheapest flight from New Delhi to Hanoi and exactly one from Hanoi to New Delhi, which should be at least 7 days later |
### Statistical Analysis
| Prompt |
|---------------------------------------------------------------------------------------------|
| What's the average price of flights from New Delhi to Vietnamese cities? |
| Compare fares between all India-Vietnam routes |
| Show me the price distribution of flights from Vietnamese cities to Mumbai |
| Which Vietnam-India route has the most varying fares? |
### Combination Queries
| Prompt |
|---------------------------------------------------------------------------------------------|
| Find the cheapest direct flight from New Delhi to any Vietnamese city |
| List the most affordable flights from Vietnamese cities to Mumbai |
| Show me the top 5 best-value routes between India and Vietnam |
| What are the most economical flights from Hanoi to Indian cities? |