This commit is contained in:
2025-05-27 00:21:32 -04:00
parent 1b8213a92e
commit af918aab19
4 changed files with 38 additions and 7 deletions

View File

@@ -36,7 +36,7 @@ def embed(text, input_type):
return resp.embeddings.float[0]
def search():
query = 'caves locations in bath county'
query = 'tazwell county caves'
query_embedding = embed(query, 'search_query')
rows = conn.execute('SELECT * FROM embeddings ORDER BY embedding <=> %s::vector LIMIT 5', (query_embedding,)).fetchall()