Added the agent genie app source code #576

nitinaggarwal-databricks · 2025-09-16T14:06:57Z

Added the agent genie app source code

github-actions · 2025-09-16T14:07:08Z

All commits in PR should be signed ('git commit -S ...'). See https://docs.github.com/en/authentication/managing-commit-signature-verification/signing-commits

Copilot

Pull Request Overview

This pull request adds the complete source code for an agent genie application, implementing a FastAPI-based Databricks Genie++ Omni-Analytics Platform. The application provides a conversational interface for data analytics, supporting features like PDF document analysis, user interaction tracking, and multi-agent AI capabilities.

Implements a full-stack web application with FastAPI backend and HTML/JavaScript frontend
Includes user interaction tracking and feedback collection system
Adds PDF document analysis functionality
Provides integration with Databricks Genie spaces and serving endpoints

Reviewed Changes

Copilot reviewed 25 out of 25 changed files in this pull request and generated 3 comments.

Show a summary per file

File	Description
agent_genie/tracking.py	User interaction tracking and feedback management system
agent_genie/templates/index.html	Complete frontend UI with chat interface and analytics features
agent_genie/table_extraction.py	Unity Catalog table and column metadata extraction utilities
agent_genie/helper.py	Core helper functions for Databricks Genie API integration
agent_genie/requirements.txt	Python dependencies for the application
agent_genie/manual_ai_content.py	Manual content storage for AI function documentation
agent_genie/databricks.yml	Databricks bundle configuration for deployment
agent_genie/app.yaml	Application configuration with environment variables
agent_genie/app.py	Flask application with chat endpoint and model integration
Various manifest and config files	Bundle metadata and development configuration

Comments suppressed due to low confidence (3)

_{Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.}

Copilot · 2025-09-16T14:08:57Z

agent_genie/app.yaml

+  - name: "SERVING_ENDPOINT_NAME"
+    value: "databricks-gpt-oss-120b"
+  - name: "TAVILY_API_KEY"
+    value: "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"


Hard-coded API key in configuration file. This should be moved to an environment variable or secure secret management system to prevent exposure in version control.

Suggested change

value: "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"

# value should be set securely via environment variable or secret manager at deployment time

Copilot · 2025-09-16T14:08:57Z

agent_genie/tracking.py

+
+        # Payload for the SQL statement
+        payload = {
+            "warehouse_id": "63bf73769cfb22e3",  # Using the same warehouse ID from the example


Hard-coded warehouse ID should be configurable through environment variables or configuration parameters rather than being embedded in the code.

Copilot · 2025-09-16T14:08:58Z

agent_genie/requirements.txt

+beautifulsoup4
+pandas
+python-dotenv
+openai


The 'openai' package is listed twice (lines 8 and 17). Remove the duplicate entry to clean up the requirements file.

Suggested change

openai

…pe manifest.yaml

alexott · 2025-09-20T08:51:36Z

first commit is still unsigned so we can't merge it

Copilot

Pull Request Overview

Copilot reviewed 27 out of 27 changed files in this pull request and generated 5 comments.

Comments suppressed due to low confidence (1)

agent_genie/app.yaml:1

API keys should not be hardcoded in configuration files. Use secure secret management or environment variables that reference secrets.

command: [

_{Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.}

Copilot · 2025-09-20T08:53:04Z

agent_genie/tracking.py

+
+        # Payload for the SQL statement
+        payload = {
+            "warehouse_id": "",  # Using the same warehouse ID from the example


Empty warehouse_id will likely cause SQL execution failures. Consider using a configurable warehouse ID from environment variables or requiring it as a parameter.

Copilot · 2025-09-20T08:53:04Z

agent_genie/tracking.py

+
+        # Payload for the SQL statement
+        payload = {
+            "warehouse_id": "63bf73769cfb22e3",  # Using the same warehouse ID from the example


Hardcoded warehouse ID should be made configurable through environment variables or function parameters to support different environments and deployments.

Copilot · 2025-09-20T08:53:04Z

agent_genie/tracking.py

+        payload = {
+            "warehouse_id": "63bf73769cfb22e3",  # Using the same warehouse ID from the example


Hardcoded warehouse ID should be made configurable through environment variables or function parameters to support different environments and deployments.

Suggested change

payload = {

"warehouse_id": "63bf73769cfb22e3", # Using the same warehouse ID from the example

warehouse_id = os.getenv("WAREHOUSE_ID")

if not warehouse_id:

return {

"success": False,

"error": "WAREHOUSE_ID environment variable is not set"

}

payload = {

"warehouse_id": warehouse_id,

Copilot · 2025-09-20T08:53:05Z

agent_genie/tracking.py

+        payload = {
+            "warehouse_id": "63bf73769cfb22e3",  # Using the same warehouse ID from the example


Hardcoded warehouse ID should be made configurable through environment variables or function parameters to support different environments and deployments.

Suggested change

payload = {

"warehouse_id": "63bf73769cfb22e3", # Using the same warehouse ID from the example

warehouse_id = os.getenv("WAREHOUSE_ID")

if not warehouse_id:

return {

"success": False,

"error": "WAREHOUSE_ID environment variable is not set"

}

payload = {

"warehouse_id": warehouse_id,

Copilot · 2025-09-20T08:53:05Z

agent_genie/requirements.txt

+beautifulsoup4
+pandas
+python-dotenv
+openai


The 'openai' dependency is listed twice (lines 8 and 17). Remove the duplicate entry.

Suggested change

openai

Added the agent genie app source code

dd4a5ba

nitinaggarwal-databricks requested a review from a team as a code owner September 16, 2025 14:06

nitinaggarwal-databricks requested a review from HariGS-DB September 16, 2025 14:06

alexott requested a review from Copilot September 16, 2025 14:07

Copilot AI reviewed Sep 16, 2025

View reviewed changes

nitinaggarwal-databricks added 6 commits September 16, 2025 17:20

Create manifest.yaml

f9c359d

Created README.md

72cced7

removed volumes spec and change Tavily key to optional and textbox ty…

0e5d157

…pe manifest.yaml

added git clone

e15e4d2

removed ++

dba06ba

commented tavily

6425f64

alexott requested a review from Copilot September 20, 2025 08:51

Copilot AI reviewed Sep 20, 2025

View reviewed changes

nitinaggarwal-databricks added 5 commits September 30, 2025 13:13

Update manifest.yaml

3deb177

Update UI

3d63d1f

Update Prompts for AI_Forecast

eb9c139

removed email

c5a75a8

Update app.yaml

a88bb36

	value: "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"
	# value should be set securely via environment variable or secret manager at deployment time

		payload = {
		"warehouse_id": "63bf73769cfb22e3", # Using the same warehouse ID from the example

-        payload = {
-            "warehouse_id": "63bf73769cfb22e3",  # Using the same warehouse ID from the example
+        warehouse_id = os.getenv("WAREHOUSE_ID")
+        if not warehouse_id:
+            return {
+                "success": False,
+                "error": "WAREHOUSE_ID environment variable is not set"
+            }
+        payload = {
+            "warehouse_id": warehouse_id,

Added the agent genie app source code #576

Are you sure you want to change the base?

Added the agent genie app source code #576

Uh oh!

Conversation

nitinaggarwal-databricks commented Sep 16, 2025

Uh oh!

github-actions bot commented Sep 16, 2025

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Reviewed Changes

Uh oh!

Copilot AI Sep 16, 2025

Choose a reason for hiding this comment

Uh oh!

Copilot AI Sep 16, 2025

Choose a reason for hiding this comment

Uh oh!

Copilot AI Sep 16, 2025

Choose a reason for hiding this comment

Uh oh!

alexott commented Sep 20, 2025

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Uh oh!

Copilot AI Sep 20, 2025

Choose a reason for hiding this comment

Uh oh!

Copilot AI Sep 20, 2025

Choose a reason for hiding this comment

Uh oh!

Copilot AI Sep 20, 2025

Choose a reason for hiding this comment

Uh oh!

Copilot AI Sep 20, 2025

Choose a reason for hiding this comment

Uh oh!

Copilot AI Sep 20, 2025

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants