From 47f9a61cb5c47eab585b79598dfa3419e40fb3d4 Mon Sep 17 00:00:00 2001 From: Montana Low Date: Thu, 7 Sep 2023 20:04:04 -0700 Subject: [PATCH] fix extra semicolon --- pgml-apps/pgml-chat/pgml_chat/main.py | 10 ++++++---- pgml-sdks/pgml/src/queries.rs | 2 +- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/pgml-apps/pgml-chat/pgml_chat/main.py b/pgml-apps/pgml-chat/pgml_chat/main.py index 51bf7d8d1..4b731b8bc 100644 --- a/pgml-apps/pgml-chat/pgml_chat/main.py +++ b/pgml-apps/pgml-chat/pgml_chat/main.py @@ -1,5 +1,5 @@ import asyncio -from pgml import Collection, Model, Splitter, Pipeline +from pgml import Collection, Model, Splitter, Pipeline, migrate, init_logger import logging from rich.logging import RichHandler from rich.progress import track @@ -121,6 +121,7 @@ async def generate_response( messages, openai_api_key, temperature=0.7, max_tokens=256, top_p=0.9 ): openai.api_key = openai_api_key + log.debug("Generating response from OpenAI API: " + str(messages)) response = openai.ChatCompletion.create( model="gpt-3.5-turbo", messages=messages, @@ -170,7 +171,7 @@ async def chat_cli(): response = await generate_response( messages, openai_api_key, max_tokens=512, temperature=0.0 ) - print("PgBot: " + response) + log.info("PgBot: " + response) user_input = input("User (Ctrl-C to exit): ") except KeyboardInterrupt: @@ -241,8 +242,8 @@ async def run(): The `main` function connects to a database, ingests documents from a specified folder, generates chunks, and logs the total number of documents and chunks. """ - print("Starting pgml-chat.... ") - + log.info("Starting pgml-chat.... ") + # await migrate() if stage == "ingest": root_dir = args.root_dir await ingest_documents(root_dir) @@ -255,6 +256,7 @@ async def run(): def main(): + init_logger() if ( stage == "chat" and chat_interface == "discord" diff --git a/pgml-sdks/pgml/src/queries.rs b/pgml-sdks/pgml/src/queries.rs index b815a2f35..820461dfe 100644 --- a/pgml-sdks/pgml/src/queries.rs +++ b/pgml-sdks/pgml/src/queries.rs @@ -8,7 +8,7 @@ CREATE TABLE IF NOT EXISTS pgml.collections ( name text NOT NULL, active BOOLEAN DEFAULT TRUE, project_id int8 NOT NULL REFERENCES pgml.projects ON DELETE CASCADE ON UPDATE CASCADE DEFERRABLE INITIALLY DEFERRED, - sdk_version text; + sdk_version text UNIQUE (name) ); "#; pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.

Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy