Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: add embedding based table search support #1314

Merged
merged 8 commits into from
Sep 2, 2023

Conversation

jczhong84
Copy link
Collaborator

  1. Add vector store plugin and necessary support functions
  2. Try to find tables first when doing text2sql if tables are not provided
  3. log table & query executions accordingly when table gets updated or a successful execution is done
  4. refactored useStream hook
  5. show table popovers

containers/bundled_querybook_config.yaml Outdated Show resolved Hide resolved
querybook/server/datasources_socketio/ai_assistant.py Outdated Show resolved Hide resolved
querybook/server/lib/ai_assistant/ai_socket.py Outdated Show resolved Hide resolved
querybook/server/lib/ai_assistant/tools/table_schema.py Outdated Show resolved Hide resolved
querybook/webapp/components/AIAssistant/AutoFixButton.tsx Outdated Show resolved Hide resolved
querybook/webapp/hooks/useAISocket.ts Outdated Show resolved Hide resolved
@czgu
Copy link
Collaborator

czgu commented Aug 30, 2023

where would init_vector_store.py be called?

@jczhong84 jczhong84 force-pushed the vectordb branch 3 times, most recently from a52094d to 5c85caf Compare September 1, 2023 04:56
@czgu czgu merged commit 839a96b into pinterest:ai-assistant Sep 2, 2023
4 checks passed
czgu pushed a commit that referenced this pull request Sep 15, 2023
* feat: use websocket for ai assistant (#1311)

* feat: use websocket for ai assistant

* fix node test

* comments

* remove memory and add keep button

* fix linter

* feat: add embedding based table search support (#1314)

* feat: add embedding based table search support

* update

* build fail

* linter

* test failure

* comments

* nodetest

* opensearch volumne path

* docs: ai assistant plugin (#1323)

* feat: add vector table search (#1322)

* feat: add vector table search

* fix linter

* remove realtime record query cell

* handle table select exceptions

* add public config

* comments

* remove unused config

* update var name metatore_id as metastoreId

* comments
aidenprice pushed a commit to arrowtail-precision/querybook that referenced this pull request Jan 3, 2024
…terest#1325)

* feat: use websocket for ai assistant (pinterest#1311)

* feat: use websocket for ai assistant

* fix node test

* comments

* remove memory and add keep button

* fix linter

* feat: add embedding based table search support (pinterest#1314)

* feat: add embedding based table search support

* update

* build fail

* linter

* test failure

* comments

* nodetest

* opensearch volumne path

* docs: ai assistant plugin (pinterest#1323)

* feat: add vector table search (pinterest#1322)

* feat: add vector table search

* fix linter

* remove realtime record query cell

* handle table select exceptions

* add public config

* comments

* remove unused config

* update var name metatore_id as metastoreId

* comments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants