Sep 8, 2023 · Sep 8, 2023
diff --git a/pgml-apps/pgml-chat/pgml_chat/main.py b/pgml-apps/pgml-chat/pgml_chat/main.py
 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
    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,
            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:
    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)


 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
  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)
 );
 "#;
Original file line number	Diff line number	Diff line change
		@@ -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
Expand DownExpand Up		@@ -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,
Expand DownExpand Up		@@ -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:
Expand DownExpand Up		@@ -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)
Expand All		@@ -255,6 +256,7 @@ async def run():


		def main():
		init_logger()
		if (
		stage == "chat"
		and chat_interface == "discord"
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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)
		);
		"#;
Expand Down