Commit2d7a335

committed

Fixed limit bug + added pypika

1 parentedc5cf5 commit2d7a335Copy full SHA for 2d7a335

File tree

+15

-3

lines changed

+15

-3

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -766,7 +766,6 @@ def vector_search(`
`766`	`766`	`FROM {embeddings_table}`
`767`	`767`	`CROSS JOIN query_cte`
`768`	`768`	`ORDER BY score DESC`
`769`		`- LIMIT {top_k}`
`770`	`769`	`)`
`771`	`770`	`SELECT cte.score, chunks.chunk, documents.metadata`
`772`	`771`	`FROM cte`
`@@ -777,7 +776,6 @@ def vector_search(`
`777`	`776`	`query_text=query,`
`778`	`777`	`model_params=sql.Literal(json.dumps(query_parameters)).as_string(conn),`
`779`	`778`	`embeddings_table=embeddings_table,`
`780`		`-top_k=top_k,`
`781`	`779`	`chunks_table=self.chunks_table,`
`782`	`780`	`documents_table=self.documents_table,`
`783`	`781`	`)`
`@@ -794,6 +792,8 @@ def vector_search(`
`794`	`792`	`ifgeneric_filter:`
`795`	`793`	`cte_select_statement+=" AND "+generic_filter`
`796`	`794`
	`795`	`+cte_select_statement+=" LIMIT {top_k}".format(top_k=top_k)`
	`796`	`+`
`797`	`797`	`search_results=run_select_statement(`
`798`	`798`	`conn,cte_select_statement,order_by="score",ascending=False`
`799`	`799`	`)`

Lines changed: 12 additions & 1 deletion

Some generated files are not rendered by default. Learn more aboutcustomizing how changed files appear on GitHub.

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,7 @@ langchain = "^0.0.167"`
`20`	`20`	`ipywidgets ="^8.0.6"`
`21`	`21`	`datasets ="^2.12.0"`
`22`	`22`	`python-dotenv ="^1.0.0"`
	`23`	`+pypika ="^0.48.9"`
`23`	`24`
`24`	`25`
`25`	`26`	`[tool.poetry.group.dev.dependencies]`

Comments

(0)