Skip to content

Instantly share code, notes, and snippets.

@tmasjc
Created October 13, 2023 02:57
Show Gist options
  • Save tmasjc/863bc9c16d172b950c6d2cf1da7715d7 to your computer and use it in GitHub Desktop.
Save tmasjc/863bc9c16d172b950c6d2cf1da7715d7 to your computer and use it in GitHub Desktop.
#!/usr/bin/env python3
import argparse
import configparser
import sqlite3
import openai
import pinecone
config = configparser.ConfigParser()
config.read("config.ini")
def openai_embeddings(content: str, model: str = "text-embedding-ada-002"):
res = openai.Embedding.create(model=model, input=content)
vec = res["data"][0]["embedding"] # type: ignore
return vec
def main(query):
vec = openai_embeddings(query)
pinecone.init(api_key=config["PINECONE"]["api"], environment="gcp-starter")
index = pinecone.Index("tech-articles")
resp = index.query(vec, top_k=3)
matches = [item['id'] for item in resp['matches']]
DB: str = config["SOURCE"]["db"]
TABLE: str = config["SOURCE"]["table"]
conn = sqlite3.connect(DB)
cursor = conn.cursor()
placeholders = ', '.join(['?'] * len(matches))
sqlite_q = f"SELECT content FROM tech_articles WHERE sign IN ({placeholders})"
cursor.execute(sqlite_q, matches)
result = cursor.fetchall()
for row in result:
print(row)
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Fetch summaries based on query.")
parser.add_argument("query", type=str, help="Query to search for.")
args = parser.parse_args()
main(args.query)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment