diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index eb6495f6b..df1821626 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -1,3 +1,4 @@ +bar import asyncio import base64 import copy @@ -44,6 +45,7 @@ from langchain.pydantic_v1 import BaseModel, Extra, root_validator from langchain.schema import LLMResult from langchain.utils import get_from_dict_or_env +from langchain_google_genai import ChatGoogleGenerativeAI class EnvAuthStrategy(BaseModel): diff --git a/packages/jupyter-ai-magics/pyproject.toml b/packages/jupyter-ai-magics/pyproject.toml index 05e11a10d..1abf1bc1c 100644 --- a/packages/jupyter-ai-magics/pyproject.toml +++ b/packages/jupyter-ai-magics/pyproject.toml @@ -27,6 +27,7 @@ dependencies = [ "typing_extensions>=4.5.0", "click~=8.0", "jsonpath-ng>=1.5.3,<2", + "langchain_google_genai", ] [project.optional-dependencies] diff --git a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/binder-on-pr.yml b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/binder-on-pr.yml index 5d7944bb6..a4ebd926a 100644 --- a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/binder-on-pr.yml +++ b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/binder-on-pr.yml @@ -3,6 +3,7 @@ name: Binder Badge on: pull_request_target: types: [opened] + workflow_dispatch: permissions: pull-requests: write diff --git a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/build.yml b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/build.yml index 0706fd9a5..eec18dad3 100644 --- a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/build.yml +++ b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/build.yml @@ -5,6 +5,7 @@ on: branches: main pull_request: branches: '*' + workflow_dispatch: jobs: build: diff --git a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/check-release.yml b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/check-release.yml index ad19ee9da..83a2a26ac 100644 --- a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/check-release.yml +++ b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/check-release.yml @@ -4,6 +4,7 @@ on: branches: ["main"] pull_request: branches: ["*"] + workflow_dispatch: jobs: check_release: diff --git a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/update-integration-tests.yml b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/update-integration-tests.yml index df49004e6..fcf1d738d 100644 --- a/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/update-integration-tests.yml +++ b/packages/jupyter-ai-module-cookiecutter/{{cookiecutter.labextension_name}}/.github/workflows/update-integration-tests.yml @@ -3,6 +3,7 @@ name: Update Playwright Snapshots on: issue_comment: types: [created, edited] + workflow_dispatch: permissions: contents: write