Skip to content

Commit

Permalink
Added comment to define sql connection
Browse files Browse the repository at this point in the history
  • Loading branch information
jasona7 committed Apr 5, 2024
1 parent 06110ba commit f17a233
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion chat_cve.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

llm = OpenAI(openai_api_key=os.environ.get("OPENAI_API_KEY"))

# Create a SQLDatabaseToolkit connection to the App_Patrol Database
# Define the SQLDatabaseToolkit connection to the App_Patrol Database
db = SQLDatabase.from_uri("sqlite:////home/jalloway/ChatCVE/app_patrol.db")
toolkit = SQLDatabaseToolkit(db=db, llm=llm) # Now passing both db and llm to SQLDatabaseToolkit

Expand Down

0 comments on commit f17a233

Please sign in to comment.