bump 305 (#11224)

Enviroment -> Environment

<!-- Thank you for contributing to LangChain!

Replace this entire comment with:
  - **Description:** a description of the change, 
  - **Issue:** the issue # it fixes (if applicable),
  - **Dependencies:** any dependencies required for this change,
- **Tag maintainer:** for a quicker response, tag the relevant
maintainer (see below),
- **Twitter handle:** we announce bigger features on Twitter. If your PR
gets announced, and you'd like a mention, we'll gladly shout you out!

Please make sure your PR is passing linting and testing before
submitting. Run `make format`, `make lint` and `make test` to check this
locally.

See contribution guidelines for more information on how to write/run
tests, lint, etc:

https://github.com/hwchase17/langchain/blob/master/.github/CONTRIBUTING.md

If you're adding a new integration, please include:
1. a test for the integration, preferably unit tests that do not rely on
network access,
2. an example notebook showing its use. It lives in `docs/extras`
directory.

If no one reviews your PR within a few days, please @-mention one of
@baskaryan, @eyurtsev, @hwchase17.
 -->

.github/CONTRIBUTING.md

@@ -14,8 +14,8 @@ Please do not try to push directly to this repo unless you are a maintainer.
 Please follow the checked-in pull request template when opening pull requests. Note related issues and tag relevant
 maintainers.
 
-Pull requests cannot land without passing the formatting, linting and testing checks first. See
-[Common Tasks](#-common-tasks) for how to run these checks locally.
+Pull requests cannot land without passing the formatting, linting and testing checks first. See [Testing](#testing) and
+[Formatting and Linting](#formatting-and-linting) for how to run these checks locally.
 
 It's essential that we maintain great documentation and testing. If you:
 - Fix a bug
@@ -59,43 +59,85 @@ we do not want these to get in the way of getting good code into the codebase.
 
 ## 🚀 Quick Start
 
-> **Note:** You can run this repository locally (which is described below) or in a [development container](https://containers.dev/) (which is described in the [.devcontainer folder](https://github.com/hwchase17/langchain/tree/master/.devcontainer)).
+This quick start describes running the repository locally.
+For a [development container](https://containers.dev/), see the [.devcontainer folder](https://github.com/hwchase17/langchain/tree/master/.devcontainer).
 
-This project uses [Poetry](https://python-poetry.org/) v1.5.1 as a dependency manager. Check out Poetry's [documentation on how to install it](https://python-poetry.org/docs/#installation) on your system before proceeding.
+### Dependency Management: Poetry and other env/dependency managers
 
-❗Note: If you use `Conda` or `Pyenv` as your environment/package manager, avoid dependency conflicts by doing the following first:
-1. *Before installing Poetry*, create and activate a new Conda env (e.g. `conda create -n langchain python=3.9`)
-2. Install Poetry v1.5.1 (see above)
-3. Tell Poetry to use the virtualenv python environment (`poetry config virtualenvs.prefer-active-python true`)
-4. Continue with the following steps.
+This project uses [Poetry](https://python-poetry.org/) v1.5.1+ as a dependency manager.
+
+❗Note: *Before installing Poetry*, if you use `Conda`, create and activate a new Conda env (e.g. `conda create -n langchain python=3.9`)
+
+Install Poetry: **[documentation on how to install it](https://python-poetry.org/docs/#installation)**.
+
+❗Note: If you use `Conda` or `Pyenv` as your environment/package manager, after installing Poetry,
+tell Poetry to use the virtualenv python environment (`poetry config virtualenvs.prefer-active-python true`)
+
+### Core vs. Experimental
 
 There are two separate projects in this repository:
 - `langchain`: core langchain code, abstractions, and use cases
-- `langchain.experimental`: more experimental code
+- `langchain.experimental`: see the [Experimental README](../libs/experimental/README.md) for more information.
 
-Each of these has their OWN development environment.
-In order to run any of the commands below, please move into their respective directories.
-For example, to contribute to `langchain` run `cd libs/langchain` before getting started with the below.
+Each of these has their own development environment. Docs are run from the top-level makefile, but development
+is split across separate test & release flows.
 
-To install requirements:
+For this quickstart, start with langchain core:
+
+```bash
+cd libs/langchain
+```
+
+### Local Development Dependencies
+
+Install langchain development requirements (for running langchain, running examples, linting, formatting, tests, and coverage):
 
 ```bash
 poetry install --with test
 ```
 
-This will install all requirements for running the package, examples, linting, formatting, tests, and coverage.
+Then verify dependency installation:
 
-❗Note: If during installation you receive a `WheelFileValidationError` for `debugpy`, please make sure you are running Poetry v1.5.1. This bug was present in older versions of Poetry (e.g. 1.4.1) and has been resolved in newer releases. If you are still seeing this bug on v1.5.1, you may also try disabling "modern installation" (`poetry config installer.modern-installation false`) and re-installing requirements. See [this `debugpy` issue](https://github.com/microsoft/debugpy/issues/1246) for more details.
+```bash
+make test
+```
 
-Now assuming `make` and `pytest` are installed, you should be able to run the common tasks in the following section. To double check, run `make test` under `libs/langchain`, all tests should pass. If they don't, you may need to pip install additional dependencies, such as `numexpr` and `openapi_schema_pydantic`.
+If the tests don't pass, you may need to pip install additional dependencies, such as `numexpr` and `openapi_schema_pydantic`.
 
-## ✅ Common Tasks
+If during installation you receive a `WheelFileValidationError` for `debugpy`, please make sure you are running
+Poetry v1.5.1+. This bug was present in older versions of Poetry (e.g. 1.4.1) and has been resolved in newer releases.
+If you are still seeing this bug on v1.5.1, you may also try disabling "modern installation"
+(`poetry config installer.modern-installation false`) and re-installing requirements.
+See [this `debugpy` issue](https://github.com/microsoft/debugpy/issues/1246) for more details.
 
-Type `make` for a list of common tasks.
+### Testing
 
-### Code Formatting
+_some test dependencies are optional; see section about optional dependencies_.
 
-Formatting for this project is done via a combination of [Black](https://black.readthedocs.io/en/stable/) and [isort](https://pycqa.github.io/isort/).
+Unit tests cover modular logic that does not require calls to outside APIs.
+If you add new logic, please add a unit test.
+
+To run unit tests:
+
+```bash
+make test
+```
+
+To run unit tests in Docker:
+
+```bash
+make docker_tests
+```
+
+There are also [integration tests and code-coverage](../libs/langchain/tests/README.md) available.
+
+### Formatting and Linting
+
+Run these locally before submitting a PR; the CI system will check also.
+
+#### Code Formatting
+
+Formatting for this project is done via a combination of [Black](https://black.readthedocs.io/en/stable/) and [ruff](https://docs.astral.sh/ruff/rules/).
 
 To run formatting for this project:
 
@@ -111,9 +153,9 @@ make format_diff
 
 This is especially useful when you have made changes to a subset of the project and want to ensure your changes are properly formatted without affecting the rest of the codebase.
 
-### Linting
+#### Linting
 
-Linting for this project is done via a combination of [Black](https://black.readthedocs.io/en/stable/), [isort](https://pycqa.github.io/isort/), [flake8](https://flake8.pycqa.org/en/latest/), and [mypy](http://mypy-lang.org/).
+Linting for this project is done via a combination of [Black](https://black.readthedocs.io/en/stable/), [ruff](https://docs.astral.sh/ruff/rules/), and [mypy](http://mypy-lang.org/).
 
 To run linting for this project:
 
@@ -131,7 +173,7 @@ This can be very helpful when you've made changes to only certain parts of the p
 
 We recognize linting can be annoying - if you do not want to do it, please contact a project maintainer, and they can help you with it. We do not want this to be a blocker for good code getting contributed.
 
-### Spellcheck
+#### Spellcheck
 
 Spellchecking for this project is done via [codespell](https://github.com/codespell-project/codespell).
 Note that `codespell` finds common typos, so it could have false-positive (correctly spelled but rarely used) and false-negatives (not finding misspelled) words.
@@ -157,17 +199,7 @@ If codespell is incorrectly flagging a word, you can skip spellcheck for that wo
 ignore-words-list = 'momento,collison,ned,foor,reworkd,parth,whats,aapply,mysogyny,unsecure'
 ```
 
-### Coverage
-
-Code coverage (i.e. the amount of code that is covered by unit tests) helps identify areas of the code that are potentially more or less brittle.
-
-To get a report of current coverage, run the following:
-
-```bash
-make coverage
-```
-
-### Working with Optional Dependencies
+## Working with Optional Dependencies
 
 Langchain relies heavily on optional dependencies to keep the Langchain package lightweight.
 
@@ -192,51 +224,7 @@ To introduce the dependency to the pyproject.toml file correctly, please do the
 test makes use of lightweight fixtures to test the logic of the code.
 5. Please use the `@pytest.mark.requires(package_name)` decorator for any tests that require the dependency.
 
-### Testing
-
-See section about optional dependencies.
-
-#### Unit Tests
-
-Unit tests cover modular logic that does not require calls to outside APIs.
-
-To run unit tests:
-
-```bash
-make test
-```
-
-To run unit tests in Docker:
-
-```bash
-make docker_tests
-```
-
-If you add new logic, please add a unit test.
-
-
-
-#### Integration Tests
-
-Integration tests cover logic that requires making calls to outside APIs (often integration with other services).
-
-**warning** Almost no tests should be integration tests.
-
-  Tests that require making network connections make it difficult for other
-  developers to test the code.
-
-  Instead favor relying on `responses` library and/or mock.patch to mock
-  requests using small fixtures.
-
-To run integration tests:
-
-```bash
-make integration_tests
-```
-
-If you add support for a new external API, please add a new integration test.
-
-### Adding a Jupyter Notebook
+## Adding a Jupyter Notebook
 
 If you are adding a Jupyter Notebook example, you'll want to install the optional `dev` dependencies.
 
@@ -259,6 +247,12 @@ When you run `poetry install`, the `langchain` package is installed as editable
 While the code is split between `langchain` and `langchain.experimental`, the documentation is one holistic thing.
 This covers how to get started contributing to documentation.
 
+From the top-level of this repo, install documentation dependencies:
+
+```bash
+poetry install
+```
+
 ### Contribute Documentation
 
 The docs directory contains Documentation and API Reference.

.github/workflows/doc_lint.yml

@@ -19,4 +19,4 @@ jobs:
       run: |
         # We should not encourage imports directly from main init file
         # Expect for hub
-        git grep 'from langchain import' docs | grep -vE 'from langchain import (hub)' && exit 1 || exit 0
+        git grep 'from langchain import' docs/{extras,docs_skeleton,snippets} | grep -vE 'from langchain import (hub)' && exit 1 || exit 0

.github/workflows/langserve_ci.yml

@@ -0,0 +1,82 @@
+---
+name: libs/langserve CI
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+    paths:
+      - '.github/actions/poetry_setup/action.yml'
+      - '.github/tools/**'
+      - '.github/workflows/_lint.yml'
+      - '.github/workflows/_test.yml'
+      - '.github/workflows/langserve_ci.yml'
+      - 'libs/langserve/**'
+  workflow_dispatch:  # Allows to trigger the workflow manually in GitHub UI
+
+# If another push to the same PR or branch happens while this workflow is still running,
+# cancel the earlier run in favor of the next run.
+#
+# There's no point in testing an outdated version of the code. GitHub only allows
+# a limited number of job runners to be active at the same time, so it's better to cancel
+# pointless jobs early so that more useful jobs can run sooner.
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+env:
+  POETRY_VERSION: "1.5.1"
+  WORKDIR: "libs/langserve"
+
+jobs:
+  lint:
+    uses:
+      ./.github/workflows/_lint.yml
+    with:
+      working-directory: libs/langserve
+    secrets: inherit
+
+  test:
+    runs-on: ubuntu-latest
+    defaults:
+      run:
+        working-directory: ${{ env.WORKDIR }}
+    strategy:
+      matrix:
+        python-version:
+          - "3.8"
+          - "3.9"
+          - "3.10"
+          - "3.11"
+    name: Python ${{ matrix.python-version }} extended tests
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Set up Python ${{ matrix.python-version }} + Poetry ${{ env.POETRY_VERSION }}
+        uses: "./.github/actions/poetry_setup"
+        with:
+          python-version: ${{ matrix.python-version }}
+          poetry-version: ${{ env.POETRY_VERSION }}
+          working-directory: libs/langserve
+          cache-key: langserve-all
+
+      - name: Install dependencies
+        shell: bash
+        run: |
+          echo "Running extended tests, installing dependencies with poetry..."
+          poetry install --with test,lint --extras all
+
+      - name: Run tests
+        run: make test
+
+      - name: Ensure the tests did not create any additional files
+        shell: bash
+        run: |
+          set -eu
+
+          STATUS="$(git status)"
+          echo "$STATUS"
+
+          # grep will exit non-zero if the target message isn't found,
+          # and `set -e` above will cause the step to fail.
+          echo "$STATUS" | grep 'nothing to commit, working tree clean'

.github/workflows/langserve_release.yml

@@ -0,0 +1,13 @@
+---
+name: libs/langserve Release
+
+on:
+  workflow_dispatch:  # Allows to trigger the workflow manually in GitHub UI
+
+jobs:
+  release:
+    uses:
+      ./.github/workflows/_release.yml
+    with:
+      working-directory: libs/langserve
+    secrets: inherit

.github/workflows/scheduled_test.yml

@@ -34,28 +34,24 @@ jobs:
           working-directory: libs/langchain
           cache-key: scheduled
 
+      - name: 'Authenticate to Google Cloud'
+        id: 'auth'
+        uses: 'google-github-actions/auth@v1'
+        with:
+          credentials_json: '${{ secrets.GOOGLE_CREDENTIALS }}'
+
       - name: Install dependencies
         working-directory: libs/langchain
         shell: bash
         run: |
           echo "Running scheduled tests, installing dependencies with poetry..."
           poetry install --with=test_integration
+          poetry run pip install google-cloud-aiplatform
 
       - name: Run tests
         shell: bash
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
         run: |
           make scheduled_tests
-
-      - name: Ensure the tests did not create any additional files
-        shell: bash
-        run: |
-          set -eu
-
-          STATUS="$(git status)"
-          echo "$STATUS"
-
-          # grep will exit non-zero if the target message isn't found,
-          # and `set -e` above will cause the step to fail.
-          echo "$STATUS" | grep 'nothing to commit, working tree clean'

Makefile

@@ -42,7 +42,8 @@ spell_fix:
 ######################
 
 help:
-	@echo '----'
+	@echo '===================='
+	@echo '-- DOCUMENTATION --'
 	@echo 'clean                        - run docs_clean and api_docs_clean'
 	@echo 'docs_build                   - build the documentation'
 	@echo 'docs_clean                   - clean the documentation build artifacts'
@@ -51,4 +52,5 @@ help:
 	@echo 'api_docs_clean               - clean the API Reference documentation build artifacts'
 	@echo 'api_docs_linkcheck           - run linkchecker on the API Reference documentation'
 	@echo 'spell_check               	- run codespell on the project'
-	@echo 'spell_fix               		- run codespell on the project and fix the errors'
+	@echo 'spell_fix               		- run codespell on the project and fix the errors'
+	@echo '-- TEST and LINT tasks are within libs/*/ per-package --'

docs/_scripts/model_feat_table.py

@@ -0,0 +1,150 @@
+import os
+from pathlib import Path
+
+from langchain import chat_models, llms
+from langchain.chat_models.base import BaseChatModel, SimpleChatModel
+from langchain.llms.base import BaseLLM, LLM
+
+INTEGRATIONS_DIR = (
+    Path(os.path.abspath(__file__)).parents[1] / "extras" / "integrations"
+)
+LLM_IGNORE = ("FakeListLLM", "OpenAIChat", "PromptLayerOpenAIChat")
+LLM_FEAT_TABLE_CORRECTION = {
+    "TextGen": {"_astream": False, "_agenerate": False},
+    "Ollama": {
+        "_stream": False,
+    },
+    "PromptLayerOpenAI": {"batch_generate": False, "batch_agenerate": False},
+}
+CHAT_MODEL_IGNORE = ("FakeListChatModel", "HumanInputChatModel")
+CHAT_MODEL_FEAT_TABLE_CORRECTION = {
+    "ChatMLflowAIGateway": {"_agenerate": False},
+    "PromptLayerChatOpenAI": {"_stream": False, "_astream": False},
+    "ChatKonko": {"_astream": False, "_agenerate": False},
+}
+
+LLM_TEMPLATE = """\
+---
+sidebar_position: 0
+sidebar_class_name: hidden
+---
+
+# LLMs
+
+import DocCardList from "@theme/DocCardList";
+
+## Features (natively supported)
+All LLMs implement the Runnable interface, which comes with default implementations of all methods, ie. `ainvoke`, `batch`, `abatch`, `stream`, `astream`. This gives all LLMs basic support for async, streaming and batch, which by default is implemented as below:
+- *Async* support defaults to calling the respective sync method in asyncio's default thread pool executor. This lets other async functions in your application make progress while the LLM is being executed, by moving this call to a background thread.
+- *Streaming* support defaults to returning an `Iterator` (or `AsyncIterator` in the case of async streaming) of a single value, the final result returned by the underlying LLM provider. This obviously doesn't give you token-by-token streaming, which requires native support from the LLM provider, but ensures your code that expects an iterator of tokens can work for any of our LLM integrations.
+- *Batch* support defaults to calling the underlying LLM in parallel for each input by making use of a thread pool executor (in the sync batch case) or `asyncio.gather` (in the async batch case). The concurrency can be controlled with the `max_concurrency` key in `RunnableConfig`.
+
+Each LLM integration can optionally provide native implementations for async, streaming or batch, which, for providers that support it, can be more efficient. The table shows, for each integration, which features have been implemented with native support.
+
+{table}
+
+<DocCardList />
+"""
+
+CHAT_MODEL_TEMPLATE = """\
+---
+sidebar_position: 1
+sidebar_class_name: hidden
+---
+
+# Chat models
+
+import DocCardList from "@theme/DocCardList";
+
+## Features (natively supported)
+All ChatModels implement the Runnable interface, which comes with default implementations of all methods, ie. `ainvoke`, `batch`, `abatch`, `stream`, `astream`. This gives all ChatModels basic support for async, streaming and batch, which by default is implemented as below:
+- *Async* support defaults to calling the respective sync method in asyncio's default thread pool executor. This lets other async functions in your application make progress while the ChatModel is being executed, by moving this call to a background thread.
+- *Streaming* support defaults to returning an `Iterator` (or `AsyncIterator` in the case of async streaming) of a single value, the final result returned by the underlying ChatModel provider. This obviously doesn't give you token-by-token streaming, which requires native support from the ChatModel provider, but ensures your code that expects an iterator of tokens can work for any of our ChatModel integrations.
+- *Batch* support defaults to calling the underlying ChatModel in parallel for each input by making use of a thread pool executor (in the sync batch case) or `asyncio.gather` (in the async batch case). The concurrency can be controlled with the `max_concurrency` key in `RunnableConfig`.
+
+Each ChatModel integration can optionally provide native implementations to truly enable async or streaming.
+The table shows, for each integration, which features have been implemented with native support.
+
+{table}
+
+<DocCardList />
+"""
+
+
+def get_llm_table():
+    llm_feat_table = {}
+    for cm in llms.__all__:
+        llm_feat_table[cm] = {}
+        cls = getattr(llms, cm)
+        if issubclass(cls, LLM):
+            for feat in ("_stream", "_astream", ("_acall", "_agenerate")):
+                if isinstance(feat, tuple):
+                    feat, name = feat
+                else:
+                    feat, name = feat, feat
+                llm_feat_table[cm][name] = getattr(cls, feat) != getattr(LLM, feat)
+        else:
+            for feat in [
+                "_stream",
+                "_astream",
+                ("_generate", "batch_generate"),
+                "_agenerate",
+                ("_agenerate", "batch_agenerate"),
+            ]:
+                if isinstance(feat, tuple):
+                    feat, name = feat
+                else:
+                    feat, name = feat, feat
+                llm_feat_table[cm][name] = getattr(cls, feat) != getattr(BaseLLM, feat)
+    final_feats = {
+        k: v
+        for k, v in {**llm_feat_table, **LLM_FEAT_TABLE_CORRECTION}.items()
+        if k not in LLM_IGNORE
+    }
+
+    header = [
+        "model",
+        "_agenerate",
+        "_stream",
+        "_astream",
+        "batch_generate",
+        "batch_agenerate",
+    ]
+    title = ["Model", "Invoke", "Async invoke", "Stream", "Async stream", "Batch", "Async batch"]
+    rows = [title, [":-"] + [":-:"] * (len(title) - 1)]
+    for llm, feats in sorted(final_feats.items()):
+        rows += [[llm, "✅"] + ["✅" if feats.get(h) else "❌" for h in header[1:]]]
+    return "\n".join(["|".join(row) for row in rows])
+
+
+def get_chat_model_table():
+    feat_table = {}
+    for cm in chat_models.__all__:
+        feat_table[cm] = {}
+        cls = getattr(chat_models, cm)
+        if issubclass(cls, SimpleChatModel):
+            comparison_cls = SimpleChatModel
+        else:
+            comparison_cls = BaseChatModel
+        for feat in ("_stream", "_astream", "_agenerate"):
+            feat_table[cm][feat] = getattr(cls, feat) != getattr(comparison_cls, feat)
+    final_feats = {
+        k: v
+        for k, v in {**feat_table, **CHAT_MODEL_FEAT_TABLE_CORRECTION}.items()
+        if k not in CHAT_MODEL_IGNORE
+    }
+    header = ["model", "_agenerate", "_stream", "_astream"]
+    title = ["Model", "Invoke", "Async invoke", "Stream", "Async stream"]
+    rows = [title, [":-"] + [":-:"] * (len(title) - 1)]
+    for llm, feats in sorted(final_feats.items()):
+        rows += [[llm, "✅"] + ["✅" if feats.get(h) else "❌" for h in header[1:]]]
+    return "\n".join(["|".join(row) for row in rows])
+
+
+if __name__ == "__main__":
+    llm_page = LLM_TEMPLATE.format(table=get_llm_table())
+    with open(INTEGRATIONS_DIR / "llms" / "index.mdx", "w") as f:
+        f.write(llm_page)
+    chat_model_page = CHAT_MODEL_TEMPLATE.format(table=get_chat_model_table())
+    with open(INTEGRATIONS_DIR / "chat" / "index.mdx", "w") as f:
+        f.write(chat_model_page)

docs/api_reference/create_api_rst.py

@@ -1,7 +1,6 @@
 """Script for auto-generating api_reference.rst."""
 import importlib
 import inspect
-import os
 import typing
 from pathlib import Path
 from typing import TypedDict, Sequence, List, Dict, Literal, Union, Optional
@@ -284,9 +283,12 @@ Functions
 def main() -> None:
     """Generate the reference.rst file for each package."""
     lc_members = _load_package_modules(PKG_DIR)
-    # Put tools.render at the top level
+    # Put some packages at top level
     tools = _load_package_modules(PKG_DIR, "tools")
     lc_members['tools.render'] = tools['render']
+    agents = _load_package_modules(PKG_DIR, "agents")
+    lc_members['agents.output_parsers'] = agents['output_parsers']
+    lc_members['agents.format_scratchpad'] = agents['format_scratchpad']
     lc_doc = ".. _api_reference:\n\n" + _construct_doc("langchain", lc_members)
     with open(WRITE_FILE, "w") as f:
         f.write(lc_doc)

docs/docs_skeleton/docs/expression_language/index.mdx

@@ -5,7 +5,23 @@ sidebar_class_name: hidden
 # LangChain Expression Language (LCEL)
 
 LangChain Expression Language or LCEL is a declarative way to easily compose chains together.
-Any chain constructed this way will automatically have full sync, async, and streaming support.
+There are several benefits to writing chains in this manner (as opposed to writing normal code):
+
+**Async, Batch, and Streaming Support**
+Any chain constructed this way will automatically have full sync, async, batch, and streaming support.
+This makes it easy to prototype a chain in a Jupyter notebook using the sync interface, and then expose it as an async streaming interface.
+
+**Fallbacks**
+The non-determinism of LLMs makes it important to be able to handle errors gracefully.
+With LCEL you can easily attach fallbacks to any chain.
+
+**Parallelism**
+Since LLM applications involve (sometimes long) API calls, it often becomes important to run things in parallel.
+With LCEL syntax, any components that can be run in parallel automatically are.
+
+**Seamless LangSmith Tracing Integration**
+As your chains get more and more complex, it becomes increasingly important to understand what exactly is happening at every step.
+With LCEL, **all** steps are automatically logged to [LangSmith](https://smith.langchain.com) for maximal observability and debuggability.
 
 #### [Interface](/docs/expression_language/interface)
 The base interface shared by all LCEL objects

docs/docs_skeleton/docs/guides/evaluation/comparison/index.mdx

@@ -16,6 +16,10 @@ Here's a summary of the key methods and properties of a comparison evaluator:
 - `requires_input`: This property indicates whether this evaluator requires an input string.
 - `requires_reference`: This property specifies whether this evaluator requires a reference label.
 
+:::note LangSmith Support
+The [run_on_dataset](https://api.python.langchain.com/en/latest/api_reference.html#module-langchain.smith) evaluation method is designed to evaluate only a single model at a time, and thus, doesn't support these evaluators.
+:::
+
 Detailed information about creating custom evaluators and the available built-in comparison evaluators is provided in the following sections.
 
 import DocCardList from "@theme/DocCardList";

docs/docs_skeleton/docs/modules/agents/agent_types/chat_conversation_agent.mdx

@@ -1,13 +0,0 @@
-# Conversational
-
-This walkthrough demonstrates how to use an agent optimized for conversation. Other agents are often optimized for using tools to figure out the best response, which is not ideal in a conversational setting where you may want the agent to be able to chat with the user as well.
-
-import Example from "@snippets/modules/agents/agent_types/conversational_agent.mdx"
-
-<Example/>
-
-import ChatExample from "@snippets/modules/agents/agent_types/chat_conversation_agent.mdx"
-
-## Using a chat model
-
-<ChatExample/>

docs/docs_skeleton/docs/modules/agents/agent_types/index.mdx

@@ -2,15 +2,13 @@
 sidebar_position: 0
 ---
 
-# Agent types
-
-## Action agents
+# Agent Types
 
 Agents use an LLM to determine which actions to take and in what order.
 An action can either be using a tool and observing its output, or returning a response to the user.
 Here are the agents available in LangChain.
 
-### [Zero-shot ReAct](/docs/modules/agents/agent_types/react.html)
+## [Zero-shot ReAct](/docs/modules/agents/agent_types/react.html)
 
 This agent uses the [ReAct](https://arxiv.org/pdf/2210.03629) framework to determine which tool to use
 based solely on the tool's description. Any number of tools can be provided.
@@ -18,33 +16,33 @@ This agent requires that a description is provided for each tool.
 
 **Note**: This is the most general purpose action agent.
 
-### [Structured input ReAct](/docs/modules/agents/agent_types/structured_chat.html)
+## [Structured input ReAct](/docs/modules/agents/agent_types/structured_chat.html)
 
 The structured tool chat agent is capable of using multi-input tools.
 Older agents are configured to specify an action input as a single string, but this agent can use a tools' argument
 schema to create a structured action input. This is useful for more complex tool usage, like precisely
 navigating around a browser.
 
-### [OpenAI Functions](/docs/modules/agents/agent_types/openai_functions_agent.html)
+## [OpenAI Functions](/docs/modules/agents/agent_types/openai_functions_agent.html)
 
 Certain OpenAI models (like gpt-3.5-turbo-0613 and gpt-4-0613) have been explicitly fine-tuned to detect when a
 function should be called and respond with the inputs that should be passed to the function.
 The OpenAI Functions Agent is designed to work with these models.
 
-### [Conversational](/docs/modules/agents/agent_types/chat_conversation_agent.html)
+## [Conversational](/docs/modules/agents/agent_types/chat_conversation_agent.html)
 
 This agent is designed to be used in conversational settings.
 The prompt is designed to make the agent helpful and conversational.
 It uses the ReAct framework to decide which tool to use, and uses memory to remember the previous conversation interactions.
 
-### [Self-ask with search](/docs/modules/agents/agent_types/self_ask_with_search.html)
+## [Self-ask with search](/docs/modules/agents/agent_types/self_ask_with_search.html)
 
 This agent utilizes a single tool that should be named `Intermediate Answer`.
 This tool should be able to lookup factual answers to questions. This agent
 is equivalent to the original [self-ask with search paper](https://ofir.io/self-ask.pdf),
 where a Google search API was provided as the tool.
 
-### [ReAct document store](/docs/modules/agents/agent_types/react_docstore.html)
+## [ReAct document store](/docs/modules/agents/agent_types/react_docstore.html)
 
 This agent uses the ReAct framework to interact with a docstore. Two tools must
 be provided: a `Search` tool and a `Lookup` tool (they must be named exactly as so).
@@ -52,6 +50,3 @@ The `Search` tool should search for a document, while the `Lookup` tool should l
 a term in the most recently found document.
 This agent is equivalent to the
 original [ReAct paper](https://arxiv.org/pdf/2210.03629.pdf), specifically the Wikipedia example.
-
-## [Plan-and-execute agents](/docs/modules/agents/agent_types/plan_and_execute.html)
-Plan-and-execute agents accomplish an objective by first planning what to do, then executing the sub tasks. This idea is largely inspired by [BabyAGI](https://github.com/yoheinakajima/babyagi) and then the ["Plan-and-Solve" paper](https://arxiv.org/abs/2305.04091).

docs/docs_skeleton/docs/modules/agents/agent_types/openai_functions_agent.mdx

@@ -1,11 +0,0 @@
-# OpenAI functions
-
-Certain OpenAI models (like gpt-3.5-turbo-0613 and gpt-4-0613) have been fine-tuned to detect when a function should be called and respond with the inputs that should be passed to the function.
-In an API call, you can describe functions and have the model intelligently choose to output a JSON object containing arguments to call those functions.
-The goal of the OpenAI Function APIs is to more reliably return valid and useful function calls than a generic text completion or chat API.
-
-The OpenAI Functions Agent is designed to work with these models.
-
-import Example from "@snippets/modules/agents/agent_types/openai_functions_agent.mdx";
-
-<Example/>

docs/docs_skeleton/docs/modules/agents/agent_types/plan_and_execute.mdx

@@ -1,11 +0,0 @@
-# Plan-and-execute
-
-Plan-and-execute agents accomplish an objective by first planning what to do, then executing the sub tasks. This idea is largely inspired by [BabyAGI](https://github.com/yoheinakajima/babyagi) and then the ["Plan-and-Solve" paper](https://arxiv.org/abs/2305.04091).
-
-The planning is almost always done by an LLM.
-
-The execution is usually done by a separate agent (equipped with tools).
-
-import Example from "@snippets/modules/agents/agent_types/plan_and_execute.mdx"
-
-<Example/>

docs/docs_skeleton/docs/modules/agents/agent_types/react.mdx

@@ -1,15 +0,0 @@
-# ReAct
-
-This walkthrough showcases using an agent to implement the [ReAct](https://react-lm.github.io/) logic.
-
-import Example from "@snippets/modules/agents/agent_types/react.mdx"
-
-<Example/>
-
-## Using chat models
-
-You can also create ReAct agents that use chat models instead of LLMs as the agent driver.
-
-import ChatExample from "@snippets/modules/agents/agent_types/react_chat.mdx"
-
-<ChatExample/>

docs/docs_skeleton/docs/modules/agents/agent_types/structured_chat.mdx

@@ -1,10 +0,0 @@
-# Structured tool chat
-
-The structured tool chat agent is capable of using multi-input tools.
-
-Older agents are configured to specify an action input as a single string, but this agent can use the provided tools' `args_schema` to populate the action input.
-
-
-import Example from "@snippets/modules/agents/agent_types/structured_chat.mdx"
-
-<Example/>

docs/docs_skeleton/docs/modules/agents/index.mdx

@@ -7,20 +7,27 @@ The core idea of agents is to use an LLM to choose a sequence of actions to take
 In chains, a sequence of actions is hardcoded (in code).
 In agents, a language model is used as a reasoning engine to determine which actions to take and in which order.
 
+Some important terminology (and schema) to know:
+
+1. `AgentAction`: This is a dataclass that represents the action an agent should take. It has a `tool` property (which is the name of the tool that should be invoked) and a `tool_input` property (the input to that tool)
+2. `AgentFinish`: This is a dataclass that signifies that the agent has finished and should return to the user. It has a `return_values` parameter, which is a dictionary to return. It often only has one key - `output` - that is a string, and so often it is just this key that is returned.
+3. `intermediate_steps`: These represent previous agent actions and corresponding outputs that are passed around. These are important to pass to future iteration so the agent knows what work it has already done. This is typed as a `List[Tuple[AgentAction, Any]]`. Note that observation is currently left as type `Any` to be maximally flexible. In practice, this is often a string.
+
 There are several key components here:
 
 ## Agent
 
-This is the class responsible for deciding what step to take next.
+This is the chain responsible for deciding what step to take next.
 This is powered by a language model and a prompt.
-This prompt can include things like:
+The inputs to this chain are:
 
-1. The personality of the agent (useful for having it respond in a certain way)
-2. Background context for the agent (useful for giving it more context on the types of tasks it's being asked to do)
-3. Prompting strategies to invoke better reasoning (the most famous/widely used being [ReAct](https://arxiv.org/abs/2210.03629))
+1. List of available tools
+2. User input
+3. Any previously executed steps (`intermediate_steps`)
 
-LangChain provides a few different types of agents to get started.
-Even then, you will likely want to customize those agents with parts (1) and (2).
+This chain then returns either the next action to take or the final response to send to the user (`AgentAction` or `AgentFinish`).
+
+Different agents have different prompting styles for reasoning, different ways of encoding input, and different ways of parsing the output.
 For a full list of agent types see [agent types](/docs/modules/agents/agent_types/)
 
 ## Tools
@@ -74,12 +81,22 @@ The `AgentExecutor` class is the main agent runtime supported by LangChain.
 However, there are other, more experimental runtimes we also support.
 These include:
 
-- [Plan-and-execute Agent](/docs/modules/agents/agent_types/plan_and_execute.html)
-- [Baby AGI](/docs/use_cases/autonomous_agents/baby_agi.html)
-- [Auto GPT](/docs/use_cases/autonomous_agents/autogpt.html)
+- [Plan-and-execute Agent](/docs/use_cases/more/agents/autonomous_agents/plan_and_execute)
+- [Baby AGI](/docs/use_cases/more/agents/autonomous_agents/baby_agi)
+- [Auto GPT](/docs/use_cases/more/agents/autonomous_agents/autogpt)
 
 ## Get started
 
 import GetStarted from "@snippets/modules/agents/get_started.mdx"
 
 <GetStarted/>
+
+## Next Steps
+
+Awesome! You've now run your first end-to-end agent.
+To dive deeper, you can:
+
+- Check out all the different [agent types](/docs/modules/agents/agent_types/) supported
+- Learn all the controls for [AgentExecutor](/docs/modules/agents/how_to/)
+- See a full list of all the off-the-shelf [toolkits](/docs/modules/agents/toolkits/) we provide
+- Explore all the individual [tools](/docs/modules/agents/tools/) supported

docs/docs_skeleton/docusaurus.config.js

@@ -71,9 +71,9 @@ const config = {
               test: /\.ipynb$/,
               loader: "raw-loader",
               resolve: {
-                fullySpecified: false
-              }
-            }
+                fullySpecified: false,
+              },
+            },
           ],
         },
       }),
@@ -158,22 +158,32 @@ const config = {
             position: "left",
           },
           {
-            type: 'docSidebar',
-            position: 'left',
-            sidebarId: 'use_cases',
-            label: 'Use cases',
+            type: "docSidebar",
+            position: "left",
+            sidebarId: "use_cases",
+            label: "Use cases",
           },
           {
-            type: 'docSidebar',
-            position: 'left',
-            sidebarId: 'integrations',
-            label: 'Integrations',
+            type: "docSidebar",
+            position: "left",
+            sidebarId: "integrations",
+            label: "Integrations",
           },
           {
-            href: "https://api.python.langchain.com",
+            to: "https://api.python.langchain.com",
             label: "API",
             position: "left",
           },
+          {
+            to: "/docs/community",
+            label: "Community",
+            position: "left",
+          },
+          {
+            to: "https://chat.langchain.com",
+            label: "Chat our docs",
+            position: "right",
+          },
           {
             to: "https://smith.langchain.com",
             label: "LangSmith",
@@ -187,9 +197,9 @@ const config = {
           // Please keep GitHub link to the right for consistency.
           {
             href: "https://github.com/hwchase17/langchain",
-            position: 'right',
-            className: 'header-github-link',
-            'aria-label': 'GitHub repository',
+            position: "right",
+            className: "header-github-link",
+            "aria-label": "GitHub repository",
           },
         ],
       },
@@ -239,6 +249,14 @@ const config = {
         copyright: `Copyright © ${new Date().getFullYear()} LangChain, Inc.`,
       },
     }),
+
+  scripts: [
+    "/js/google_analytics.js",
+    {
+      src: "https://www.googletagmanager.com/gtag/js?id=G-9B66JQQH2F",
+      async: true,
+    },
+  ],
 };
 
 module.exports = config;

docs/docs_skeleton/package-lock.json

@@ -12,7 +12,7 @@
         "@docusaurus/preset-classic": "2.4.0",
         "@docusaurus/remark-plugin-npm2yarn": "^2.4.0",
         "@mdx-js/react": "^1.6.22",
-        "@mendable/search": "^0.0.150",
+        "@mendable/search": "^0.0.160",
         "clsx": "^1.2.1",
         "json-loader": "^0.5.7",
         "process": "^0.11.10",
@@ -3212,9 +3212,9 @@
       }
     },
     "node_modules/@mendable/search": {
-      "version": "0.0.150",
-      "resolved": "https://registry.npmjs.org/@mendable/search/-/search-0.0.150.tgz",
-      "integrity": "sha512-Eb5SeAWlMxzEim/8eJ/Ysn01Pyh39xlPBzRBw/5OyOBhti0HVLXk4wd1Fq2TKgJC2ppQIvhEKO98PUcj9dNDFw==",
+      "version": "0.0.160",
+      "resolved": "https://registry.npmjs.org/@mendable/search/-/search-0.0.160.tgz",
+      "integrity": "sha512-Lq9Cy176iVeUlSS9PALyc0KPgMWv9MELgsDKXKLhyoPS85yQXs0uEpC2Zgf9i+R4jar5PibKZPh2Hj2xIm/Ajg==",
       "dependencies": {
         "html-react-parser": "^4.2.0",
         "posthog-js": "^1.45.1"

docs/docs_skeleton/package.json

@@ -23,7 +23,7 @@
     "@docusaurus/preset-classic": "2.4.0",
     "@docusaurus/remark-plugin-npm2yarn": "^2.4.0",
     "@mdx-js/react": "^1.6.22",
-    "@mendable/search": "^0.0.150",
+    "@mendable/search": "^0.0.160",
     "clsx": "^1.2.1",
     "json-loader": "^0.5.7",
     "process": "^0.11.10",

docs/docs_skeleton/sidebars.js

@@ -67,7 +67,7 @@ module.exports = {
     },
     {
       type: "category",
-      label: "Additional resources",
+      label: "More",
       collapsed: true,
       items: [
         { type: "autogenerated", dirName: "additional_resources" },
@@ -77,8 +77,7 @@ module.exports = {
         type: 'generated-index',
         slug: "additional_resources",
       },
-    },
-    'community'
+    }
   ],
   integrations: [
     {
@@ -99,8 +98,8 @@ module.exports = {
       label: "Components",
       collapsible: false,
       items: [
-        { type: "category", label: "LLMs", collapsed: true, items: [{type:"autogenerated", dirName: "integrations/llms" }], link: {type: "generated-index", slug: "integrations/llms" }},
-        { type: "category", label: "Chat models", collapsed: true, items: [{type:"autogenerated", dirName: "integrations/chat" }], link: {type: "generated-index", slug: "integrations/chat" }},
+        { type: "category", label: "LLMs", collapsed: true, items: [{type:"autogenerated", dirName: "integrations/llms" }], link: { type: 'doc', id: "integrations/llms/index"}},
+        { type: "category", label: "Chat models", collapsed: true, items: [{type:"autogenerated", dirName: "integrations/chat" }], link: { type: 'doc', id: "integrations/chat/index"}},
         { type: "category", label: "Document loaders", collapsed: true, items: [{type:"autogenerated", dirName: "integrations/document_loaders" }], link: {type: "generated-index", slug: "integrations/document_loaders" }},
         { type: "category", label: "Document transformers", collapsed: true, items: [{type: "autogenerated", dirName: "integrations/document_transformers" }], link: {type: "generated-index", slug: "integrations/document_transformers" }},
         { type: "category", label: "Text embedding models", collapsed: true, items: [{type: "autogenerated", dirName: "integrations/text_embedding" }], link: {type: "generated-index", slug: "integrations/text_embedding" }},

docs/docs_skeleton/src/css/custom.css

@@ -36,13 +36,11 @@
   --ifm-color-primary-lightest: #4fddbf;
 }
 
-/* Reduce width on mobile for Mendable Search */
-@media (max-width: 767px) {
-  .mendable-search {
-    width: 200px;
-  }
+.mendable-search {
+  width: 175px;
 }
 
+/* Reduce width on mobile for Mendable Search */
 @media (max-width: 500px) {
   .mendable-search {
     width: 150px;
@@ -157,4 +155,6 @@
 [data-theme='dark'] .header-github-link::before {
   background: url("data:image/svg+xml,%3Csvg viewBox='0 0 24 24' xmlns='http://www.w3.org/2000/svg'%3E%3Cpath fill='white' d='M12 .297c-6.63 0-12 5.373-12 12 0 5.303 3.438 9.8 8.205 11.385.6.113.82-.258.82-.577 0-.285-.01-1.04-.015-2.04-3.338.724-4.042-1.61-4.042-1.61C4.422 18.07 3.633 17.7 3.633 17.7c-1.087-.744.084-.729.084-.729 1.205.084 1.838 1.236 1.838 1.236 1.07 1.835 2.809 1.305 3.495.998.108-.776.417-1.305.76-1.605-2.665-.3-5.466-1.332-5.466-5.93 0-1.31.465-2.38 1.235-3.22-.135-.303-.54-1.523.105-3.176 0 0 1.005-.322 3.3 1.23.96-.267 1.98-.399 3-.405 1.02.006 2.04.138 3 .405 2.28-1.552 3.285-1.23 3.285-1.23.645 1.653.24 2.873.12 3.176.765.84 1.23 1.91 1.23 3.22 0 4.61-2.805 5.625-5.475 5.92.42.36.81 1.096.81 2.22 0 1.606-.015 2.896-.015 3.286 0 .315.21.69.825.57C20.565 22.092 24 17.592 24 12.297c0-6.627-5.373-12-12-12'/%3E%3C/svg%3E")
     no-repeat;
-}
+}
+
+

docs/docs_skeleton/src/theme/SearchBar.js

@@ -19,9 +19,14 @@ export default function SearchBarWrapper() {
       <MendableSearchBar
         anon_key={customFields.mendableAnonKey}
         style={{ accentColor: "#4F956C", darkMode: false }}
-        placeholder="Search..."
+        placeholder="Search"
         dialogPlaceholder="How do I use a LLM Chain?"
         messageSettings={{ openSourcesInNewTab: false, prettySources: true }}
+        searchBarStyle={{
+          borderColor: "#9d9ea1",
+          color:"#9d9ea1"
+        }}
+        askAIText="Ask Mendable AI"
         isPinnable
         showSimpleSearch
       />

docs/docs_skeleton/static/js/google_analytics.js

@@ -0,0 +1,7 @@
+window.dataLayer = window.dataLayer || [];
+function gtag() {
+  dataLayer.push(arguments);
+}
+gtag("js", new Date());
+
+gtag("config", "G-9B66JQQH2F");

docs/docs_skeleton/vercel.json

@@ -1,72 +1,92 @@
 {
   "redirects": [
+    {
+      "source": "/docs/modules/agents/agents/examples/mrkl_chat(.html?)",
+      "destination": "/docs/modules/agents/"
+    },
+    {
+      "source": "/docs/use_cases(/?)",
+      "destination": "/docs/use_cases/question_answering/"
+    },
+    {
+      "source": "/docs/integrations(/?)",
+      "destination": "/docs/integrations/providers/"
+    },
+    {
+      "source": "/docs/integrations/platforms(/?)",
+      "destination": "/docs/integrations/providers/"
+    },
+    {
+      "source": "/docs/integrations/platforms(/?)",
+      "destination": "/docs/integrations/providers/"
+    },
     {
       "source": "/docs/expression_language/cookbook/routing",
       "destination": "/docs/expression_language/how_to/routing"
     },
     {
       "source":  "/docs/integrations/providers/amazon_api_gateway",
-      "destination": "/docs/integrations/platform/aws"
+      "destination": "/docs/integrations/platforms/aws"
     },
     {
       "source":  "/docs/integrations/providers/azure_blob_storage",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/google_vertexai_matchingengine",
-      "destination": "/docs/integrations/platform/google"
+      "destination": "/docs/integrations/platforms/google"
     },
     {
       "source":  "/docs/integrations/providers/aws_s3",
-      "destination": "/docs/integrations/platform/aws"
+      "destination": "/docs/integrations/platforms/aws"
     },
     {
       "source":  "/docs/integrations/providers/azure_openai",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/azure_blob_storage",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/azure_cognitive_search_",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/bedrock",
-      "destination": "/docs/integrations/platform/aws"
+      "destination": "/docs/integrations/platforms/aws"
     },
     {
       "source":  "/docs/integrations/providers/google_bigquery",
-      "destination": "/docs/integrations/platform/google"
+      "destination": "/docs/integrations/platforms/google"
     },
     {
       "source":  "/docs/integrations/providers/google_cloud_storage",
-      "destination": "/docs/integrations/platform/google"
+      "destination": "/docs/integrations/platforms/google"
     },
     {
       "source":  "/docs/integrations/providers/google_drive",
-      "destination": "/docs/integrations/platform/google"
+      "destination": "/docs/integrations/platforms/google"
     },
     {
       "source":  "/docs/integrations/providers/google_search",
-      "destination": "/docs/integrations/platform/google"
+      "destination": "/docs/integrations/platforms/google"
     },
     {
       "source":  "/docs/integrations/providers/microsoft_onedrive",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/microsoft_powerpoint",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/microsoft_word",
-      "destination": "/docs/integrations/platform/microsoft"
+      "destination": "/docs/integrations/platforms/microsoft"
     },
     {
       "source":  "/docs/integrations/providers/sagemaker_endpoint",
-      "destination": "/docs/integrations/platform/aws"
+      "destination": "/docs/integrations/platforms/aws"
     },
     {
       "source":  "/docs/integrations/providers/sagemaker_tracking",
@@ -74,7 +94,7 @@
     },
     {
       "source":  "/docs/integrations/providers/openai",
-      "destination": "/docs/integrations/callbacks/openai"
+      "destination": "/docs/integrations/platforms/openai"
     },
     {
       "source": "/docs/modules/data_connection/caching_embeddings(/?)",
@@ -438,7 +458,7 @@
     },
     {
       "source": "/docs/integrations/openai",
-      "destination": "/docs/integrations/providers/openai"
+      "destination": "/docs/integrations/platforms/openai"
     },
     {
       "source": "/docs/integrations/opensearch",
@@ -1952,6 +1972,18 @@
       "source": "/docs/modules/data_connection/document_loaders/integrations/youtube_transcript",
       "destination": "/docs/integrations/document_loaders/youtube_transcript"
     },
+    {
+      "source": "/docs/integrations/document_loaders/Etherscan",
+      "destination": "/docs/integrations/document_loaders/etherscan"
+    },
+    {
+      "source": "/docs/integrations/document_loaders/merge_doc_loader",
+      "destination": "/docs/integrations/document_loaders/merge_doc"
+    },
+    {
+      "source": "/docs/integrations/document_loaders/recursive_url_loader",
+      "destination": "/docs/integrations/document_loaders/recursive_url"
+    },
     {
       "source": "/en/latest/modules/indexes/text_splitters/examples/markdown_header_metadata.html",
       "destination": "/docs/modules/data_connection/document_transformers/text_splitters/markdown_header_metadata"

docs/extras/expression_language/cookbook/multiple_chains.ipynb

@@ -95,7 +95,7 @@
     }
    ],
    "source": [
-    "question_generator.invoke({\"warm\"})"
+    "question_generator.invoke(\"warm\")"
    ]
   },
   {
@@ -116,7 +116,7 @@
     }
    ],
    "source": [
-    "prompt = question_generator.invoke({\"warm\"})\n",
+    "prompt = question_generator.invoke(\"warm\")\n",
     "model.invoke(prompt)"
    ]
   },

docs/extras/expression_language/how_to/_category_.yml

@@ -1,2 +0,0 @@
-label: 'How to'
-position: 1

docs/extras/expression_language/how_to/fallbacks.ipynb

@@ -277,7 +277,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.1"
+   "version": "3.10.1"
   }
  },
  "nbformat": 4,

docs/extras/expression_language/how_to/functions.ipynb

@@ -14,12 +14,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 77,
+   "execution_count": 4,
    "id": "6bb221b3",
    "metadata": {},
    "outputs": [],
    "source": [
     "from langchain.schema.runnable import RunnableLambda\n",
+    "from langchain.prompts import ChatPromptTemplate\n",
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from operator import itemgetter\n",
     "\n",
     "def length_function(text):\n",
     "    return len(text)\n",
@@ -31,6 +34,7 @@
     "    return _multiple_length_function(_dict[\"text1\"], _dict[\"text2\"])\n",
     "\n",
     "prompt = ChatPromptTemplate.from_template(\"what is {a} + {b}\")\n",
+    "model = ChatOpenAI()\n",
     "\n",
     "chain1 = prompt | model\n",
     "\n",
@@ -42,7 +46,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 78,
+   "execution_count": 5,
    "id": "5488ec85",
    "metadata": {},
    "outputs": [
@@ -52,7 +56,7 @@
        "AIMessage(content='3 + 9 equals 12.', additional_kwargs={}, example=False)"
       ]
      },
-     "execution_count": 78,
+     "execution_count": 5,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -73,17 +77,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 139,
+   "execution_count": 9,
    "id": "80b3b5f6-5d58-44b9-807e-cce9a46bf49f",
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.schema.runnable import RunnableConfig"
+    "from langchain.schema.runnable import RunnableConfig\n",
+    "from langchain.schema.output_parser import StrOutputParser"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 149,
+   "execution_count": 10,
    "id": "ff0daf0c-49dd-4d21-9772-e5fa133c5f36",
    "metadata": {},
    "outputs": [],
@@ -109,7 +114,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 152,
+   "execution_count": 12,
    "id": "1a5e709e-9d75-48c7-bb9c-503251990505",
    "metadata": {},
    "outputs": [
@@ -132,6 +137,14 @@
     "    RunnableLambda(parse_or_fix).invoke(\"{foo: bar}\", {\"tags\": [\"my-tag\"], \"callbacks\": [cb]})\n",
     "    print(cb)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "29f55c38",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {
@@ -150,7 +163,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.1"
+   "version": "3.10.1"
   }
  },
  "nbformat": 4,

docs/extras/expression_language/how_to/index.mdx

@@ -0,0 +1,9 @@
+---
+sidebar_position: 1
+---
+
+# How to
+
+import DocCardList from "@theme/DocCardList";
+
+<DocCardList />

docs/extras/expression_language/how_to/map.ipynb

@@ -12,18 +12,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 2,
    "id": "7e1873d6-d4b6-43ac-96a1-edcf178201e0",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "{'joke': AIMessage(content=\"Why don't bears wear shoes? \\nBecause they have bear feet!\", additional_kwargs={}, example=False),\n",
-       " 'poem': AIMessage(content=\"In twilight's embrace, a bear's gentle lumber,\\nSilent strength, nature's awe, a humble slumber.\", additional_kwargs={}, example=False)}"
+       "{'joke': AIMessage(content=\"Why don't bears wear shoes? \\n\\nBecause they have bear feet!\", additional_kwargs={}, example=False),\n",
+       " 'poem': AIMessage(content=\"In woodland depths, bear prowls with might,\\nSilent strength, nature's sovereign, day and night.\", additional_kwargs={}, example=False)}"
       ]
      },
-     "execution_count": 5,
+     "execution_count": 2,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -38,7 +38,7 @@
     "joke_chain = ChatPromptTemplate.from_template(\"tell me a joke about {topic}\") | model\n",
     "poem_chain = ChatPromptTemplate.from_template(\"write a 2-line poem about {topic}\") | model\n",
     "\n",
-    "map_chain = RunnableMap({\"joke\": chain1, \"poem\": chain2,})\n",
+    "map_chain = RunnableMap({\"joke\": joke_chain, \"poem\": poem_chain,})\n",
     "\n",
     "map_chain.invoke({\"topic\": \"bear\"})"
    ]
@@ -54,7 +54,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 3,
    "id": "267d1460-53c1-4fdb-b2c3-b6a1eb7fccff",
    "metadata": {},
    "outputs": [
@@ -64,7 +64,7 @@
        "'Harrison worked at Kensho.'"
       ]
      },
-     "execution_count": 4,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -191,7 +191,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.1"
+   "version": "3.10.1"
   }
  },
  "nbformat": 4,

docs/extras/guides/deployments/template_repos.mdx

@@ -47,13 +47,13 @@ A minimal example on how to deploy LangChain to [Kinsta](https://kinsta.com) usi
 
 A minimal example of how to deploy LangChain to [Fly.io](https://fly.io/) using Flask.
 
-## [Digitalocean App Platform](https://github.com/homanp/digitalocean-langchain)
+## [DigitalOcean App Platform](https://github.com/homanp/digitalocean-langchain)
 
 A minimal example of how to deploy LangChain to DigitalOcean App Platform.
 
 ## [CI/CD Google Cloud Build + Dockerfile + Serverless Google Cloud Run](https://github.com/g-emarco/github-assistant)
 
-Boilerplate LangChain project on how to deploy to Google Cloud Run using Docker with Cloud Build CI/CD pipeline
+Boilerplate LangChain project on how to deploy to Google Cloud Run using Docker with Cloud Build CI/CD pipeline.
 
 ## [Google Cloud Run](https://github.com/homanp/gcp-langchain)
 

docs/extras/guides/evaluation/comparison/custom.ipynb

@@ -1,280 +1,281 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "657d2c8c-54b4-42a3-9f02-bdefa0ed6728",
-   "metadata": {},
-   "source": [
-    "# Custom Pairwise Evaluator\n",
-    "\n",
-    "You can make your own pairwise string evaluators by inheriting from `PairwiseStringEvaluator` class and overwriting the `_evaluate_string_pairs` method (and the `_aevaluate_string_pairs` method if you want to use the evaluator asynchronously).\n",
-    "\n",
-    "In this example, you will make a simple custom evaluator that just returns whether the first prediction has more whitespace tokenized 'words' than the second.\n",
-    "\n",
-    "You can check out the reference docs for the [PairwiseStringEvaluator interface](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.schema.PairwiseStringEvaluator.html#langchain.evaluation.schema.PairwiseStringEvaluator) for more info.\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "93f3a653-d198-4291-973c-8d1adba338b2",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from typing import Optional, Any\n",
-    "from langchain.evaluation import PairwiseStringEvaluator\n",
-    "\n",
-    "\n",
-    "class LengthComparisonPairwiseEvalutor(PairwiseStringEvaluator):\n",
-    "    \"\"\"\n",
-    "    Custom evaluator to compare two strings.\n",
-    "    \"\"\"\n",
-    "\n",
-    "    def _evaluate_string_pairs(\n",
-    "        self,\n",
-    "        *,\n",
-    "        prediction: str,\n",
-    "        prediction_b: str,\n",
-    "        reference: Optional[str] = None,\n",
-    "        input: Optional[str] = None,\n",
-    "        **kwargs: Any,\n",
-    "    ) -> dict:\n",
-    "        score = int(len(prediction.split()) > len(prediction_b.split()))\n",
-    "        return {\"score\": score}"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "7d4a77c3-07a7-4076-8e7f-f9bca0d6c290",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 1}"
-      ]
-     },
-     "execution_count": 2,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator = LengthComparisonPairwiseEvalutor()\n",
-    "\n",
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"The quick brown fox jumped over the lazy dog.\",\n",
-    "    prediction_b=\"The quick brown fox jumped over the dog.\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "d90f128f-6f49-42a1-b05a-3aea568ee03b",
-   "metadata": {},
-   "source": [
-    "## LLM-Based Example\n",
-    "\n",
-    "That example was simple to illustrate the API, but it wasn't very useful in practice. Below, use an LLM with some custom instructions to form a simple preference scorer similar to the built-in [PairwiseStringEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain). We will use `ChatAnthropic` for the evaluator chain."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "b4b43098-4d96-417b-a8a9-b3e75779cfe8",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# %pip install anthropic\n",
-    "# %env ANTHROPIC_API_KEY=YOUR_API_KEY"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "b6e978ab-48f1-47ff-9506-e13b1a50be6e",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from typing import Optional, Any\n",
-    "from langchain.evaluation import PairwiseStringEvaluator\n",
-    "from langchain.chat_models import ChatAnthropic\n",
-    "from langchain.chains import LLMChain\n",
-    "\n",
-    "\n",
-    "class CustomPreferenceEvaluator(PairwiseStringEvaluator):\n",
-    "    \"\"\"\n",
-    "    Custom evaluator to compare two strings using a custom LLMChain.\n",
-    "    \"\"\"\n",
-    "\n",
-    "    def __init__(self) -> None:\n",
-    "        llm = ChatAnthropic(model=\"claude-2\", temperature=0)\n",
-    "        self.eval_chain = LLMChain.from_string(\n",
-    "            llm,\n",
-    "            \"\"\"Which option is preferred? Do not take order into account. Evaluate based on accuracy and helpfulness. If neither is preferred, respond with C. Provide your reasoning, then finish with Preference: A/B/C\n",
-    "\n",
-    "Input: How do I get the path of the parent directory in python 3.8?\n",
-    "Option A: You can use the following code:\n",
-    "```python\n",
-    "import os\n",
-    "\n",
-    "os.path.dirname(os.path.dirname(os.path.abspath(__file__)))\n",
-    "```\n",
-    "Option B: You can use the following code:\n",
-    "```python\n",
-    "from pathlib import Path\n",
-    "Path(__file__).absolute().parent\n",
-    "```\n",
-    "Reasoning: Both options return the same result. However, since option B is more concise and easily understand, it is preferred.\n",
-    "Preference: B\n",
-    "\n",
-    "Which option is preferred? Do not take order into account. Evaluate based on accuracy and helpfulness. If neither is preferred, respond with C. Provide your reasoning, then finish with Preference: A/B/C\n",
-    "Input: {input}\n",
-    "Option A: {prediction}\n",
-    "Option B: {prediction_b}\n",
-    "Reasoning:\"\"\",\n",
-    "        )\n",
-    "\n",
-    "    @property\n",
-    "    def requires_input(self) -> bool:\n",
-    "        return True\n",
-    "\n",
-    "    @property\n",
-    "    def requires_reference(self) -> bool:\n",
-    "        return False\n",
-    "\n",
-    "    def _evaluate_string_pairs(\n",
-    "        self,\n",
-    "        *,\n",
-    "        prediction: str,\n",
-    "        prediction_b: str,\n",
-    "        reference: Optional[str] = None,\n",
-    "        input: Optional[str] = None,\n",
-    "        **kwargs: Any,\n",
-    "    ) -> dict:\n",
-    "        result = self.eval_chain(\n",
-    "            {\n",
-    "                \"input\": input,\n",
-    "                \"prediction\": prediction,\n",
-    "                \"prediction_b\": prediction_b,\n",
-    "                \"stop\": [\"Which option is preferred?\"],\n",
-    "            },\n",
-    "            **kwargs,\n",
-    "        )\n",
-    "\n",
-    "        response_text = result[\"text\"]\n",
-    "        reasoning, preference = response_text.split(\"Preference:\", maxsplit=1)\n",
-    "        preference = preference.strip()\n",
-    "        score = 1.0 if preference == \"A\" else (0.0 if preference == \"B\" else None)\n",
-    "        return {\"reasoning\": reasoning.strip(), \"value\": preference, \"score\": score}"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "5cbd8b1d-2cb0-4f05-b435-a1a00074d94a",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "evaluator = CustomPreferenceEvaluator()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "2c0a7fb7-b976-4443-9f0e-e707a6dfbdf7",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'reasoning': 'Option B is preferred over option A for importing from a relative directory, because it is more straightforward and concise.\\n\\nOption A uses the importlib module, which allows importing a module by specifying the full name as a string. While this works, it is less clear compared to option B.\\n\\nOption B directly imports from the relative path using dot notation, which clearly shows that it is a relative import. This is the recommended way to do relative imports in Python.\\n\\nIn summary, option B is more accurate and helpful as it uses the standard Python relative import syntax.',\n",
-       " 'value': 'B',\n",
-       " 'score': 0.0}"
-      ]
-     },
-     "execution_count": 7,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    input=\"How do I import from a relative directory?\",\n",
-    "    prediction=\"use importlib! importlib.import_module('.my_package', '.')\",\n",
-    "    prediction_b=\"from .sibling import foo\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "f13a1346-7dbe-451d-b3a3-99e8fc7b753b",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "CustomPreferenceEvaluator requires an input string.\n"
-     ]
-    }
-   ],
-   "source": [
-    "# Setting requires_input to return True adds additional validation to avoid returning a grade when insufficient data is provided to the chain.\n",
-    "\n",
-    "try:\n",
-    "    evaluator.evaluate_string_pairs(\n",
-    "        prediction=\"use importlib! importlib.import_module('.my_package', '.')\",\n",
-    "        prediction_b=\"from .sibling import foo\",\n",
-    "    )\n",
-    "except ValueError as e:\n",
-    "    print(e)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e7829cc3-ebd1-4628-ae97-15166202e9cc",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "657d2c8c-54b4-42a3-9f02-bdefa0ed6728",
+            "metadata": {},
+            "source": [
+                "# Custom Pairwise Evaluator\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/comparison/custom.ipynb)\n",
+                "\n",
+                "You can make your own pairwise string evaluators by inheriting from `PairwiseStringEvaluator` class and overwriting the `_evaluate_string_pairs` method (and the `_aevaluate_string_pairs` method if you want to use the evaluator asynchronously).\n",
+                "\n",
+                "In this example, you will make a simple custom evaluator that just returns whether the first prediction has more whitespace tokenized 'words' than the second.\n",
+                "\n",
+                "You can check out the reference docs for the [PairwiseStringEvaluator interface](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.schema.PairwiseStringEvaluator.html#langchain.evaluation.schema.PairwiseStringEvaluator) for more info.\n"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "93f3a653-d198-4291-973c-8d1adba338b2",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from typing import Optional, Any\n",
+                "from langchain.evaluation import PairwiseStringEvaluator\n",
+                "\n",
+                "\n",
+                "class LengthComparisonPairwiseEvalutor(PairwiseStringEvaluator):\n",
+                "    \"\"\"\n",
+                "    Custom evaluator to compare two strings.\n",
+                "    \"\"\"\n",
+                "\n",
+                "    def _evaluate_string_pairs(\n",
+                "        self,\n",
+                "        *,\n",
+                "        prediction: str,\n",
+                "        prediction_b: str,\n",
+                "        reference: Optional[str] = None,\n",
+                "        input: Optional[str] = None,\n",
+                "        **kwargs: Any,\n",
+                "    ) -> dict:\n",
+                "        score = int(len(prediction.split()) > len(prediction_b.split()))\n",
+                "        return {\"score\": score}"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "7d4a77c3-07a7-4076-8e7f-f9bca0d6c290",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1}"
+                        ]
+                    },
+                    "execution_count": 2,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator = LengthComparisonPairwiseEvalutor()\n",
+                "\n",
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"The quick brown fox jumped over the lazy dog.\",\n",
+                "    prediction_b=\"The quick brown fox jumped over the dog.\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "d90f128f-6f49-42a1-b05a-3aea568ee03b",
+            "metadata": {},
+            "source": [
+                "## LLM-Based Example\n",
+                "\n",
+                "That example was simple to illustrate the API, but it wasn't very useful in practice. Below, use an LLM with some custom instructions to form a simple preference scorer similar to the built-in [PairwiseStringEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain). We will use `ChatAnthropic` for the evaluator chain."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "b4b43098-4d96-417b-a8a9-b3e75779cfe8",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "# %pip install anthropic\n",
+                "# %env ANTHROPIC_API_KEY=YOUR_API_KEY"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "b6e978ab-48f1-47ff-9506-e13b1a50be6e",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from typing import Optional, Any\n",
+                "from langchain.evaluation import PairwiseStringEvaluator\n",
+                "from langchain.chat_models import ChatAnthropic\n",
+                "from langchain.chains import LLMChain\n",
+                "\n",
+                "\n",
+                "class CustomPreferenceEvaluator(PairwiseStringEvaluator):\n",
+                "    \"\"\"\n",
+                "    Custom evaluator to compare two strings using a custom LLMChain.\n",
+                "    \"\"\"\n",
+                "\n",
+                "    def __init__(self) -> None:\n",
+                "        llm = ChatAnthropic(model=\"claude-2\", temperature=0)\n",
+                "        self.eval_chain = LLMChain.from_string(\n",
+                "            llm,\n",
+                "            \"\"\"Which option is preferred? Do not take order into account. Evaluate based on accuracy and helpfulness. If neither is preferred, respond with C. Provide your reasoning, then finish with Preference: A/B/C\n",
+                "\n",
+                "Input: How do I get the path of the parent directory in python 3.8?\n",
+                "Option A: You can use the following code:\n",
+                "```python\n",
+                "import os\n",
+                "\n",
+                "os.path.dirname(os.path.dirname(os.path.abspath(__file__)))\n",
+                "```\n",
+                "Option B: You can use the following code:\n",
+                "```python\n",
+                "from pathlib import Path\n",
+                "Path(__file__).absolute().parent\n",
+                "```\n",
+                "Reasoning: Both options return the same result. However, since option B is more concise and easily understand, it is preferred.\n",
+                "Preference: B\n",
+                "\n",
+                "Which option is preferred? Do not take order into account. Evaluate based on accuracy and helpfulness. If neither is preferred, respond with C. Provide your reasoning, then finish with Preference: A/B/C\n",
+                "Input: {input}\n",
+                "Option A: {prediction}\n",
+                "Option B: {prediction_b}\n",
+                "Reasoning:\"\"\",\n",
+                "        )\n",
+                "\n",
+                "    @property\n",
+                "    def requires_input(self) -> bool:\n",
+                "        return True\n",
+                "\n",
+                "    @property\n",
+                "    def requires_reference(self) -> bool:\n",
+                "        return False\n",
+                "\n",
+                "    def _evaluate_string_pairs(\n",
+                "        self,\n",
+                "        *,\n",
+                "        prediction: str,\n",
+                "        prediction_b: str,\n",
+                "        reference: Optional[str] = None,\n",
+                "        input: Optional[str] = None,\n",
+                "        **kwargs: Any,\n",
+                "    ) -> dict:\n",
+                "        result = self.eval_chain(\n",
+                "            {\n",
+                "                \"input\": input,\n",
+                "                \"prediction\": prediction,\n",
+                "                \"prediction_b\": prediction_b,\n",
+                "                \"stop\": [\"Which option is preferred?\"],\n",
+                "            },\n",
+                "            **kwargs,\n",
+                "        )\n",
+                "\n",
+                "        response_text = result[\"text\"]\n",
+                "        reasoning, preference = response_text.split(\"Preference:\", maxsplit=1)\n",
+                "        preference = preference.strip()\n",
+                "        score = 1.0 if preference == \"A\" else (0.0 if preference == \"B\" else None)\n",
+                "        return {\"reasoning\": reasoning.strip(), \"value\": preference, \"score\": score}"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "5cbd8b1d-2cb0-4f05-b435-a1a00074d94a",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "evaluator = CustomPreferenceEvaluator()"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "id": "2c0a7fb7-b976-4443-9f0e-e707a6dfbdf7",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'reasoning': 'Option B is preferred over option A for importing from a relative directory, because it is more straightforward and concise.\\n\\nOption A uses the importlib module, which allows importing a module by specifying the full name as a string. While this works, it is less clear compared to option B.\\n\\nOption B directly imports from the relative path using dot notation, which clearly shows that it is a relative import. This is the recommended way to do relative imports in Python.\\n\\nIn summary, option B is more accurate and helpful as it uses the standard Python relative import syntax.',\n",
+                            " 'value': 'B',\n",
+                            " 'score': 0.0}"
+                        ]
+                    },
+                    "execution_count": 7,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    input=\"How do I import from a relative directory?\",\n",
+                "    prediction=\"use importlib! importlib.import_module('.my_package', '.')\",\n",
+                "    prediction_b=\"from .sibling import foo\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 13,
+            "id": "f13a1346-7dbe-451d-b3a3-99e8fc7b753b",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "CustomPreferenceEvaluator requires an input string.\n"
+                    ]
+                }
+            ],
+            "source": [
+                "# Setting requires_input to return True adds additional validation to avoid returning a grade when insufficient data is provided to the chain.\n",
+                "\n",
+                "try:\n",
+                "    evaluator.evaluate_string_pairs(\n",
+                "        prediction=\"use importlib! importlib.import_module('.my_package', '.')\",\n",
+                "        prediction_b=\"from .sibling import foo\",\n",
+                "    )\n",
+                "except ValueError as e:\n",
+                "    print(e)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": null,
+            "id": "e7829cc3-ebd1-4628-ae97-15166202e9cc",
+            "metadata": {},
+            "outputs": [],
+            "source": []
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
 }

docs/extras/guides/evaluation/comparison/pairwise_embedding_distance.ipynb

@@ -1,232 +1,233 @@
 {
- "cells": [
-  {
-   "attachments": {},
-   "cell_type": "markdown",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "# Pairwise Embedding Distance \n",
-    "\n",
-    "One way to measure the similarity (or dissimilarity) between two predictions on a shared or similar input is to embed the predictions and compute a vector distance between the two embeddings.<a name=\"cite_ref-1\"></a>[<sup>[1]</sup>](#cite_note-1)\n",
-    "\n",
-    "You can load the `pairwise_embedding_distance` evaluator to do this.\n",
-    "\n",
-    "**Note:** This returns a **distance** score, meaning that the lower the number, the **more** similar the outputs are, according to their embedded representation.\n",
-    "\n",
-    "Check out the reference docs for the [PairwiseEmbeddingDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.embedding_distance.base.PairwiseEmbeddingDistanceEvalChain.html#langchain.evaluation.embedding_distance.base.PairwiseEmbeddingDistanceEvalChain) for more info."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"pairwise_embedding_distance\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.0966466944859925}"
-      ]
-     },
-     "execution_count": 2,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"Seattle is hot in June\", prediction_b=\"Seattle is cool in June.\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.03761174337464557}"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"Seattle is warm in June\", prediction_b=\"Seattle is cool in June.\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Select the Distance Metric\n",
-    "\n",
-    "By default, the evalutor uses cosine distance. You can choose a different distance metric if you'd like. "
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[<EmbeddingDistance.COSINE: 'cosine'>,\n",
-       " <EmbeddingDistance.EUCLIDEAN: 'euclidean'>,\n",
-       " <EmbeddingDistance.MANHATTAN: 'manhattan'>,\n",
-       " <EmbeddingDistance.CHEBYSHEV: 'chebyshev'>,\n",
-       " <EmbeddingDistance.HAMMING: 'hamming'>]"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from langchain.evaluation import EmbeddingDistance\n",
-    "\n",
-    "list(EmbeddingDistance)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "evaluator = load_evaluator(\n",
-    "    \"pairwise_embedding_distance\", distance_metric=EmbeddingDistance.EUCLIDEAN\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Select Embeddings to Use\n",
-    "\n",
-    "The constructor uses `OpenAI` embeddings by default, but you can configure this however you want. Below, use huggingface local embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.embeddings import HuggingFaceEmbeddings\n",
-    "\n",
-    "embedding_model = HuggingFaceEmbeddings()\n",
-    "hf_evaluator = load_evaluator(\"pairwise_embedding_distance\", embeddings=embedding_model)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.5486443280477362}"
-      ]
-     },
-     "execution_count": 10,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "hf_evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"Seattle is hot in June\", prediction_b=\"Seattle is cool in June.\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.21018880025138598}"
-      ]
-     },
-     "execution_count": 12,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "hf_evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"Seattle is warm in June\", prediction_b=\"Seattle is cool in June.\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "<a name=\"cite_note-1\"></a><i>1. Note: When it comes to semantic similarity, this often gives better results than older string distance metrics (such as those in the `PairwiseStringDistanceEvalChain`), though it tends to be less reliable than evaluators that use the LLM directly (such as the `PairwiseStringEvalChain`) </i>"
-   ]
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 4
-}
+    "cells": [
+        {
+            "attachments": {},
+            "cell_type": "markdown",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "# Pairwise Embedding Distance \n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/comparison/pairwise_embedding_distance.ipynb)\n",
+                "\n",
+                "One way to measure the similarity (or dissimilarity) between two predictions on a shared or similar input is to embed the predictions and compute a vector distance between the two embeddings.<a name=\"cite_ref-1\"></a>[<sup>[1]</sup>](#cite_note-1)\n",
+                "\n",
+                "You can load the `pairwise_embedding_distance` evaluator to do this.\n",
+                "\n",
+                "**Note:** This returns a **distance** score, meaning that the lower the number, the **more** similar the outputs are, according to their embedded representation.\n",
+                "\n",
+                "Check out the reference docs for the [PairwiseEmbeddingDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.embedding_distance.base.PairwiseEmbeddingDistanceEvalChain.html#langchain.evaluation.embedding_distance.base.PairwiseEmbeddingDistanceEvalChain) for more info."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"pairwise_embedding_distance\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.0966466944859925}"
+                        ]
+                    },
+                    "execution_count": 2,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"Seattle is hot in June\", prediction_b=\"Seattle is cool in June.\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.03761174337464557}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"Seattle is warm in June\", prediction_b=\"Seattle is cool in June.\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "## Select the Distance Metric\n",
+                "\n",
+                "By default, the evalutor uses cosine distance. You can choose a different distance metric if you'd like. "
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "[<EmbeddingDistance.COSINE: 'cosine'>,\n",
+                            " <EmbeddingDistance.EUCLIDEAN: 'euclidean'>,\n",
+                            " <EmbeddingDistance.MANHATTAN: 'manhattan'>,\n",
+                            " <EmbeddingDistance.CHEBYSHEV: 'chebyshev'>,\n",
+                            " <EmbeddingDistance.HAMMING: 'hamming'>]"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "from langchain.evaluation import EmbeddingDistance\n",
+                "\n",
+                "list(EmbeddingDistance)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "evaluator = load_evaluator(\n",
+                "    \"pairwise_embedding_distance\", distance_metric=EmbeddingDistance.EUCLIDEAN\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "## Select Embeddings to Use\n",
+                "\n",
+                "The constructor uses `OpenAI` embeddings by default, but you can configure this however you want. Below, use huggingface local embeddings"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": null,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.embeddings import HuggingFaceEmbeddings\n",
+                "\n",
+                "embedding_model = HuggingFaceEmbeddings()\n",
+                "hf_evaluator = load_evaluator(\"pairwise_embedding_distance\", embeddings=embedding_model)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 10,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.5486443280477362}"
+                        ]
+                    },
+                    "execution_count": 10,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "hf_evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"Seattle is hot in June\", prediction_b=\"Seattle is cool in June.\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 12,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.21018880025138598}"
+                        ]
+                    },
+                    "execution_count": 12,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "hf_evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"Seattle is warm in June\", prediction_b=\"Seattle is cool in June.\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "<a name=\"cite_note-1\"></a><i>1. Note: When it comes to semantic similarity, this often gives better results than older string distance metrics (such as those in the `PairwiseStringDistanceEvalChain`), though it tends to be less reliable than evaluators that use the LLM directly (such as the `PairwiseStringEvalChain`) </i>"
+            ]
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 4
+}

docs/extras/guides/evaluation/comparison/pairwise_string.ipynb

@@ -1,381 +1,382 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "2da95378",
-   "metadata": {},
-   "source": [
-    "# Pairwise String Comparison\n",
-    "\n",
-    "Often you will want to compare predictions of an LLM, Chain, or Agent for a given input. The `StringComparison` evaluators facilitate this so you can answer questions like:\n",
-    "\n",
-    "- Which LLM or prompt produces a preferred output for a given question?\n",
-    "- Which examples should I include for few-shot example selection?\n",
-    "- Which output is better to include for fintetuning?\n",
-    "\n",
-    "The simplest and often most reliable automated way to choose a preferred prediction for a given input is to use the `pairwise_string` evaluator.\n",
-    "\n",
-    "Check out the reference docs for the [PairwiseStringEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain) for more info."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "f6790c46",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"labeled_pairwise_string\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "49ad9139",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'reasoning': 'Both responses are relevant to the question asked, as they both provide a numerical answer to the question about the number of dogs in the park. However, Response A is incorrect according to the reference answer, which states that there are four dogs. Response B, on the other hand, is correct as it matches the reference answer. Neither response demonstrates depth of thought, as they both simply provide a numerical answer without any additional information or context. \\n\\nBased on these criteria, Response B is the better response.\\n',\n",
-       " 'value': 'B',\n",
-       " 'score': 0}"
-      ]
-     },
-     "execution_count": 2,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"there are three dogs\",\n",
-    "    prediction_b=\"4\",\n",
-    "    input=\"how many dogs are in the park?\",\n",
-    "    reference=\"four\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "7491d2e6-4e77-4b17-be6b-7da966785c1d",
-   "metadata": {},
-   "source": [
-    "## Methods\n",
-    "\n",
-    "\n",
-    "The pairwise string evaluator can be called using [evaluate_string_pairs](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.evaluate_string_pairs) (or async [aevaluate_string_pairs](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.aevaluate_string_pairs)) methods, which accept:\n",
-    "\n",
-    "- prediction (str) – The predicted response of the first model, chain, or prompt.\n",
-    "- prediction_b (str) – The predicted response of the second model, chain, or prompt.\n",
-    "- input (str) – The input question, prompt, or other text.\n",
-    "- reference (str) – (Only for the labeled_pairwise_string variant) The reference response.\n",
-    "\n",
-    "They return a dictionary with the following values:\n",
-    "- value: 'A' or 'B', indicating whether `prediction` or `prediction_b` is preferred, respectively\n",
-    "- score: Integer 0 or 1 mapped from the 'value', where a score of 1 would mean that the first `prediction` is preferred, and a score of 0 would mean `prediction_b` is preferred.\n",
-    "- reasoning: String \"chain of thought reasoning\" from the LLM generated prior to creating the score"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "ed353b93-be71-4479-b9c0-8c97814c2e58",
-   "metadata": {},
-   "source": [
-    "## Without References\n",
-    "\n",
-    "When references aren't available, you can still predict the preferred response.\n",
-    "The results will reflect the evaluation model's preference, which is less reliable and may result\n",
-    "in preferences that are factually incorrect."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "586320da",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"pairwise_string\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "7f56c76e-a39b-4509-8b8a-8a2afe6c3da1",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'reasoning': 'Both responses are correct and relevant to the question. However, Response B is more helpful and insightful as it provides a more detailed explanation of what addition is. Response A is correct but lacks depth as it does not explain what the operation of addition entails. \\n\\nFinal Decision: [[B]]',\n",
-       " 'value': 'B',\n",
-       " 'score': 0}"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"Addition is a mathematical operation.\",\n",
-    "    prediction_b=\"Addition is a mathematical operation that adds two numbers to create a third number, the 'sum'.\",\n",
-    "    input=\"What is addition?\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "4a09b21d-9851-47e8-93d3-90044b2945b0",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "## Defining the Criteria\n",
-    "\n",
-    "By default, the LLM is instructed to select the 'preferred' response based on helpfulness, relevance, correctness, and depth of thought. You can customize the criteria by passing in a `criteria` argument, where the criteria could take any of the following forms:\n",
-    "- [`Criteria`](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.Criteria.html#langchain.evaluation.criteria.eval_chain.Criteria) enum or its string value - to use one of the default criteria and their descriptions\n",
-    "- [Constitutional principal](https://api.python.langchain.com/en/latest/chains/langchain.chains.constitutional_ai.models.ConstitutionalPrinciple.html#langchain.chains.constitutional_ai.models.ConstitutionalPrinciple) - use one any of the constitutional principles defined in langchain\n",
-    "- Dictionary: a list of custom criteria, where the key is the name of the criteria, and the value is the description.\n",
-    "- A list of criteria or constitutional principles - to combine multiple criteria in one.\n",
-    "\n",
-    "Below is an example for determining preferred writing responses based on a custom style."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "8539e7d9-f7b0-4d32-9c45-593a7915c093",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "custom_criteria = {\n",
-    "    \"simplicity\": \"Is the language straightforward and unpretentious?\",\n",
-    "    \"clarity\": \"Are the sentences clear and easy to understand?\",\n",
-    "    \"precision\": \"Is the writing precise, with no unnecessary words or details?\",\n",
-    "    \"truthfulness\": \"Does the writing feel honest and sincere?\",\n",
-    "    \"subtext\": \"Does the writing suggest deeper meanings or themes?\",\n",
-    "}\n",
-    "evaluator = load_evaluator(\"pairwise_string\", criteria=custom_criteria)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "fec7bde8-fbdc-4730-8366-9d90d033c181",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'reasoning': 'Response A is simple, clear, and precise. It uses straightforward language to convey a deep and sincere message about families. The metaphor of joy and sorrow as music is effective and easy to understand.\\n\\nResponse B, on the other hand, is more complex and less clear. The language is more pretentious, with words like \"domicile,\" \"resounds,\" \"abode,\" \"dissonant,\" and \"elegy.\" While it conveys a similar message to Response A, it does so in a more convoluted way. The precision is also lacking due to the use of unnecessary words and details.\\n\\nBoth responses suggest deeper meanings or themes about the shared joy and unique sorrow in families. However, Response A does so in a more effective and accessible way.\\n\\nTherefore, the better response is [[A]].',\n",
-       " 'value': 'A',\n",
-       " 'score': 1}"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"Every cheerful household shares a similar rhythm of joy; but sorrow, in each household, plays a unique, haunting melody.\",\n",
-    "    prediction_b=\"Where one finds a symphony of joy, every domicile of happiness resounds in harmonious,\"\n",
-    "    \" identical notes; yet, every abode of despair conducts a dissonant orchestra, each\"\n",
-    "    \" playing an elegy of grief that is peculiar and profound to its own existence.\",\n",
-    "    input=\"Write some prose about families.\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "a25b60b2-627c-408a-be4b-a2e5cbc10726",
-   "metadata": {},
-   "source": [
-    "## Customize the LLM\n",
-    "\n",
-    "By default, the loader uses `gpt-4` in the evaluation chain. You can customize this when loading."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "de84a958-1330-482b-b950-68bcf23f9e35",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from langchain.chat_models import ChatAnthropic\n",
-    "\n",
-    "llm = ChatAnthropic(temperature=0)\n",
-    "\n",
-    "evaluator = load_evaluator(\"labeled_pairwise_string\", llm=llm)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "e162153f-d50a-4a7c-a033-019dabbc954c",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'reasoning': 'Here is my assessment:\\n\\nResponse B is more helpful, insightful, and accurate than Response A. Response B simply states \"4\", which directly answers the question by providing the exact number of dogs mentioned in the reference answer. In contrast, Response A states \"there are three dogs\", which is incorrect according to the reference answer. \\n\\nIn terms of helpfulness, Response B gives the precise number while Response A provides an inaccurate guess. For relevance, both refer to dogs in the park from the question. However, Response B is more correct and factual based on the reference answer. Response A shows some attempt at reasoning but is ultimately incorrect. Response B requires less depth of thought to simply state the factual number.\\n\\nIn summary, Response B is superior in terms of helpfulness, relevance, correctness, and depth. My final decision is: [[B]]\\n',\n",
-       " 'value': 'B',\n",
-       " 'score': 0}"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"there are three dogs\",\n",
-    "    prediction_b=\"4\",\n",
-    "    input=\"how many dogs are in the park?\",\n",
-    "    reference=\"four\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "e0e89c13-d0ad-4f87-8fcb-814399bafa2a",
-   "metadata": {},
-   "source": [
-    "## Customize the Evaluation Prompt\n",
-    "\n",
-    "You can use your own custom evaluation prompt to add more task-specific instructions or to instruct the evaluator to score the output.\n",
-    "\n",
-    "*Note: If you use a prompt that expects generates a result in a unique format, you may also have to pass in a custom output parser (`output_parser=your_parser()`) instead of the default `PairwiseStringResultOutputParser`"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "fb817efa-3a4d-439d-af8c-773b89d97ec9",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.prompts import PromptTemplate\n",
-    "\n",
-    "prompt_template = PromptTemplate.from_template(\n",
-    "    \"\"\"Given the input context, which do you prefer: A or B?\n",
-    "Evaluate based on the following criteria:\n",
-    "{criteria}\n",
-    "Reason step by step and finally, respond with either [[A]] or [[B]] on its own line.\n",
-    "\n",
-    "DATA\n",
-    "----\n",
-    "input: {input}\n",
-    "reference: {reference}\n",
-    "A: {prediction}\n",
-    "B: {prediction_b}\n",
-    "---\n",
-    "Reasoning:\n",
-    "\n",
-    "\"\"\"\n",
-    ")\n",
-    "evaluator = load_evaluator(\n",
-    "    \"labeled_pairwise_string\", prompt=prompt_template\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "d40aa4f0-cfd5-4cb4-83c8-8d2300a04c2f",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "input_variables=['prediction', 'reference', 'prediction_b', 'input'] output_parser=None partial_variables={'criteria': 'helpfulness: Is the submission helpful, insightful, and appropriate?\\nrelevance: Is the submission referring to a real quote from the text?\\ncorrectness: Is the submission correct, accurate, and factual?\\ndepth: Does the submission demonstrate depth of thought?'} template='Given the input context, which do you prefer: A or B?\\nEvaluate based on the following criteria:\\n{criteria}\\nReason step by step and finally, respond with either [[A]] or [[B]] on its own line.\\n\\nDATA\\n----\\ninput: {input}\\nreference: {reference}\\nA: {prediction}\\nB: {prediction_b}\\n---\\nReasoning:\\n\\n' template_format='f-string' validate_template=True\n"
-     ]
-    }
-   ],
-   "source": [
-    "# The prompt was assigned to the evaluator\n",
-    "print(evaluator.prompt)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "9467bb42-7a31-4071-8f66-9ed2c6f06dcd",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'reasoning': 'Helpfulness: Both A and B are helpful as they provide a direct answer to the question.\\nRelevance: A is relevant as it refers to the correct name of the dog from the text. B is not relevant as it provides a different name.\\nCorrectness: A is correct as it accurately states the name of the dog. B is incorrect as it provides a different name.\\nDepth: Both A and B demonstrate a similar level of depth as they both provide a straightforward answer to the question.\\n\\nGiven these evaluations, the preferred response is:\\n',\n",
-       " 'value': 'A',\n",
-       " 'score': 1}"
-      ]
-     },
-     "execution_count": 11,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_string_pairs(\n",
-    "    prediction=\"The dog that ate the ice cream was named fido.\",\n",
-    "    prediction_b=\"The dog's name is spot\",\n",
-    "    input=\"What is the name of the dog that ate the ice cream?\",\n",
-    "    reference=\"The dog's name is fido\",\n",
-    ")"
-   ]
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "2da95378",
+            "metadata": {},
+            "source": [
+                "# Pairwise String Comparison\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/comparison/pairwise_string.ipynb)\n",
+                "\n",
+                "Often you will want to compare predictions of an LLM, Chain, or Agent for a given input. The `StringComparison` evaluators facilitate this so you can answer questions like:\n",
+                "\n",
+                "- Which LLM or prompt produces a preferred output for a given question?\n",
+                "- Which examples should I include for few-shot example selection?\n",
+                "- Which output is better to include for fintetuning?\n",
+                "\n",
+                "The simplest and often most reliable automated way to choose a preferred prediction for a given input is to use the `pairwise_string` evaluator.\n",
+                "\n",
+                "Check out the reference docs for the [PairwiseStringEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain) for more info."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "f6790c46",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"labeled_pairwise_string\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "49ad9139",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'reasoning': 'Both responses are relevant to the question asked, as they both provide a numerical answer to the question about the number of dogs in the park. However, Response A is incorrect according to the reference answer, which states that there are four dogs. Response B, on the other hand, is correct as it matches the reference answer. Neither response demonstrates depth of thought, as they both simply provide a numerical answer without any additional information or context. \\n\\nBased on these criteria, Response B is the better response.\\n',\n",
+                            " 'value': 'B',\n",
+                            " 'score': 0}"
+                        ]
+                    },
+                    "execution_count": 2,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"there are three dogs\",\n",
+                "    prediction_b=\"4\",\n",
+                "    input=\"how many dogs are in the park?\",\n",
+                "    reference=\"four\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "7491d2e6-4e77-4b17-be6b-7da966785c1d",
+            "metadata": {},
+            "source": [
+                "## Methods\n",
+                "\n",
+                "\n",
+                "The pairwise string evaluator can be called using [evaluate_string_pairs](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.evaluate_string_pairs) (or async [aevaluate_string_pairs](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.html#langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain.aevaluate_string_pairs)) methods, which accept:\n",
+                "\n",
+                "- prediction (str) – The predicted response of the first model, chain, or prompt.\n",
+                "- prediction_b (str) – The predicted response of the second model, chain, or prompt.\n",
+                "- input (str) – The input question, prompt, or other text.\n",
+                "- reference (str) – (Only for the labeled_pairwise_string variant) The reference response.\n",
+                "\n",
+                "They return a dictionary with the following values:\n",
+                "- value: 'A' or 'B', indicating whether `prediction` or `prediction_b` is preferred, respectively\n",
+                "- score: Integer 0 or 1 mapped from the 'value', where a score of 1 would mean that the first `prediction` is preferred, and a score of 0 would mean `prediction_b` is preferred.\n",
+                "- reasoning: String \"chain of thought reasoning\" from the LLM generated prior to creating the score"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "ed353b93-be71-4479-b9c0-8c97814c2e58",
+            "metadata": {},
+            "source": [
+                "## Without References\n",
+                "\n",
+                "When references aren't available, you can still predict the preferred response.\n",
+                "The results will reflect the evaluation model's preference, which is less reliable and may result\n",
+                "in preferences that are factually incorrect."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "586320da",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"pairwise_string\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "7f56c76e-a39b-4509-8b8a-8a2afe6c3da1",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'reasoning': 'Both responses are correct and relevant to the question. However, Response B is more helpful and insightful as it provides a more detailed explanation of what addition is. Response A is correct but lacks depth as it does not explain what the operation of addition entails. \\n\\nFinal Decision: [[B]]',\n",
+                            " 'value': 'B',\n",
+                            " 'score': 0}"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"Addition is a mathematical operation.\",\n",
+                "    prediction_b=\"Addition is a mathematical operation that adds two numbers to create a third number, the 'sum'.\",\n",
+                "    input=\"What is addition?\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "4a09b21d-9851-47e8-93d3-90044b2945b0",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "## Defining the Criteria\n",
+                "\n",
+                "By default, the LLM is instructed to select the 'preferred' response based on helpfulness, relevance, correctness, and depth of thought. You can customize the criteria by passing in a `criteria` argument, where the criteria could take any of the following forms:\n",
+                "- [`Criteria`](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.Criteria.html#langchain.evaluation.criteria.eval_chain.Criteria) enum or its string value - to use one of the default criteria and their descriptions\n",
+                "- [Constitutional principal](https://api.python.langchain.com/en/latest/chains/langchain.chains.constitutional_ai.models.ConstitutionalPrinciple.html#langchain.chains.constitutional_ai.models.ConstitutionalPrinciple) - use one any of the constitutional principles defined in langchain\n",
+                "- Dictionary: a list of custom criteria, where the key is the name of the criteria, and the value is the description.\n",
+                "- A list of criteria or constitutional principles - to combine multiple criteria in one.\n",
+                "\n",
+                "Below is an example for determining preferred writing responses based on a custom style."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "id": "8539e7d9-f7b0-4d32-9c45-593a7915c093",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "custom_criteria = {\n",
+                "    \"simplicity\": \"Is the language straightforward and unpretentious?\",\n",
+                "    \"clarity\": \"Are the sentences clear and easy to understand?\",\n",
+                "    \"precision\": \"Is the writing precise, with no unnecessary words or details?\",\n",
+                "    \"truthfulness\": \"Does the writing feel honest and sincere?\",\n",
+                "    \"subtext\": \"Does the writing suggest deeper meanings or themes?\",\n",
+                "}\n",
+                "evaluator = load_evaluator(\"pairwise_string\", criteria=custom_criteria)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "fec7bde8-fbdc-4730-8366-9d90d033c181",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'reasoning': 'Response A is simple, clear, and precise. It uses straightforward language to convey a deep and sincere message about families. The metaphor of joy and sorrow as music is effective and easy to understand.\\n\\nResponse B, on the other hand, is more complex and less clear. The language is more pretentious, with words like \"domicile,\" \"resounds,\" \"abode,\" \"dissonant,\" and \"elegy.\" While it conveys a similar message to Response A, it does so in a more convoluted way. The precision is also lacking due to the use of unnecessary words and details.\\n\\nBoth responses suggest deeper meanings or themes about the shared joy and unique sorrow in families. However, Response A does so in a more effective and accessible way.\\n\\nTherefore, the better response is [[A]].',\n",
+                            " 'value': 'A',\n",
+                            " 'score': 1}"
+                        ]
+                    },
+                    "execution_count": 6,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"Every cheerful household shares a similar rhythm of joy; but sorrow, in each household, plays a unique, haunting melody.\",\n",
+                "    prediction_b=\"Where one finds a symphony of joy, every domicile of happiness resounds in harmonious,\"\n",
+                "    \" identical notes; yet, every abode of despair conducts a dissonant orchestra, each\"\n",
+                "    \" playing an elegy of grief that is peculiar and profound to its own existence.\",\n",
+                "    input=\"Write some prose about families.\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "a25b60b2-627c-408a-be4b-a2e5cbc10726",
+            "metadata": {},
+            "source": [
+                "## Customize the LLM\n",
+                "\n",
+                "By default, the loader uses `gpt-4` in the evaluation chain. You can customize this when loading."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "id": "de84a958-1330-482b-b950-68bcf23f9e35",
+            "metadata": {},
+            "outputs": [],
+            "source": [
+                "from langchain.chat_models import ChatAnthropic\n",
+                "\n",
+                "llm = ChatAnthropic(temperature=0)\n",
+                "\n",
+                "evaluator = load_evaluator(\"labeled_pairwise_string\", llm=llm)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 8,
+            "id": "e162153f-d50a-4a7c-a033-019dabbc954c",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'reasoning': 'Here is my assessment:\\n\\nResponse B is more helpful, insightful, and accurate than Response A. Response B simply states \"4\", which directly answers the question by providing the exact number of dogs mentioned in the reference answer. In contrast, Response A states \"there are three dogs\", which is incorrect according to the reference answer. \\n\\nIn terms of helpfulness, Response B gives the precise number while Response A provides an inaccurate guess. For relevance, both refer to dogs in the park from the question. However, Response B is more correct and factual based on the reference answer. Response A shows some attempt at reasoning but is ultimately incorrect. Response B requires less depth of thought to simply state the factual number.\\n\\nIn summary, Response B is superior in terms of helpfulness, relevance, correctness, and depth. My final decision is: [[B]]\\n',\n",
+                            " 'value': 'B',\n",
+                            " 'score': 0}"
+                        ]
+                    },
+                    "execution_count": 8,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"there are three dogs\",\n",
+                "    prediction_b=\"4\",\n",
+                "    input=\"how many dogs are in the park?\",\n",
+                "    reference=\"four\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "e0e89c13-d0ad-4f87-8fcb-814399bafa2a",
+            "metadata": {},
+            "source": [
+                "## Customize the Evaluation Prompt\n",
+                "\n",
+                "You can use your own custom evaluation prompt to add more task-specific instructions or to instruct the evaluator to score the output.\n",
+                "\n",
+                "*Note: If you use a prompt that expects generates a result in a unique format, you may also have to pass in a custom output parser (`output_parser=your_parser()`) instead of the default `PairwiseStringResultOutputParser`"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 9,
+            "id": "fb817efa-3a4d-439d-af8c-773b89d97ec9",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.prompts import PromptTemplate\n",
+                "\n",
+                "prompt_template = PromptTemplate.from_template(\n",
+                "    \"\"\"Given the input context, which do you prefer: A or B?\n",
+                "Evaluate based on the following criteria:\n",
+                "{criteria}\n",
+                "Reason step by step and finally, respond with either [[A]] or [[B]] on its own line.\n",
+                "\n",
+                "DATA\n",
+                "----\n",
+                "input: {input}\n",
+                "reference: {reference}\n",
+                "A: {prediction}\n",
+                "B: {prediction_b}\n",
+                "---\n",
+                "Reasoning:\n",
+                "\n",
+                "\"\"\"\n",
+                ")\n",
+                "evaluator = load_evaluator(\n",
+                "    \"labeled_pairwise_string\", prompt=prompt_template\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 10,
+            "id": "d40aa4f0-cfd5-4cb4-83c8-8d2300a04c2f",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "input_variables=['prediction', 'reference', 'prediction_b', 'input'] output_parser=None partial_variables={'criteria': 'helpfulness: Is the submission helpful, insightful, and appropriate?\\nrelevance: Is the submission referring to a real quote from the text?\\ncorrectness: Is the submission correct, accurate, and factual?\\ndepth: Does the submission demonstrate depth of thought?'} template='Given the input context, which do you prefer: A or B?\\nEvaluate based on the following criteria:\\n{criteria}\\nReason step by step and finally, respond with either [[A]] or [[B]] on its own line.\\n\\nDATA\\n----\\ninput: {input}\\nreference: {reference}\\nA: {prediction}\\nB: {prediction_b}\\n---\\nReasoning:\\n\\n' template_format='f-string' validate_template=True\n"
+                    ]
+                }
+            ],
+            "source": [
+                "# The prompt was assigned to the evaluator\n",
+                "print(evaluator.prompt)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 11,
+            "id": "9467bb42-7a31-4071-8f66-9ed2c6f06dcd",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'reasoning': 'Helpfulness: Both A and B are helpful as they provide a direct answer to the question.\\nRelevance: A is relevant as it refers to the correct name of the dog from the text. B is not relevant as it provides a different name.\\nCorrectness: A is correct as it accurately states the name of the dog. B is incorrect as it provides a different name.\\nDepth: Both A and B demonstrate a similar level of depth as they both provide a straightforward answer to the question.\\n\\nGiven these evaluations, the preferred response is:\\n',\n",
+                            " 'value': 'A',\n",
+                            " 'score': 1}"
+                        ]
+                    },
+                    "execution_count": 11,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_string_pairs(\n",
+                "    prediction=\"The dog that ate the ice cream was named fido.\",\n",
+                "    prediction_b=\"The dog's name is spot\",\n",
+                "    input=\"What is the name of the dog that ate the ice cream?\",\n",
+                "    reference=\"The dog's name is fido\",\n",
+                ")"
+            ]
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/examples/comparisons.ipynb

@@ -1,447 +1,448 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "# Comparing Chain Outputs\n",
-    "\n",
-    "Suppose you have two different prompts (or LLMs). How do you know which will generate \"better\" results?\n",
-    "\n",
-    "One automated way to predict the preferred configuration is to use a `PairwiseStringEvaluator` like the `PairwiseStringEvalChain`<a name=\"cite_ref-1\"></a>[<sup>[1]</sup>](#cite_note-1). This chain prompts an LLM to select which output is preferred, given a specific input.\n",
-    "\n",
-    "For this evaluation, we will need 3 things:\n",
-    "1. An evaluator\n",
-    "2. A dataset of inputs\n",
-    "3. 2 (or more) LLMs, Chains, or Agents to compare\n",
-    "\n",
-    "Then we will aggregate the restults to determine the preferred model.\n",
-    "\n",
-    "### Step 1. Create the Evaluator\n",
-    "\n",
-    "In this example, you will use gpt-4 to select which output is preferred."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "eval_chain = load_evaluator(\"pairwise_string\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Step 2. Select Dataset\n",
-    "\n",
-    "If you already have real usage data for your LLM, you can use a representative sample. More examples\n",
-    "provide more reliable results. We will use some example queries someone might have about how to use langchain here."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "Found cached dataset parquet (/Users/wfh/.cache/huggingface/datasets/LangChainDatasets___parquet/LangChainDatasets--langchain-howto-queries-bbb748bbee7e77aa/0.0.0/14a00e99c0d15a23649d0db8944380ac81082d4b021f398733dd84f3a6c569a7)\n"
-     ]
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "# Comparing Chain Outputs\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/examples/comparisons.ipynb)\n",
+                "\n",
+                "Suppose you have two different prompts (or LLMs). How do you know which will generate \"better\" results?\n",
+                "\n",
+                "One automated way to predict the preferred configuration is to use a `PairwiseStringEvaluator` like the `PairwiseStringEvalChain`<a name=\"cite_ref-1\"></a>[<sup>[1]</sup>](#cite_note-1). This chain prompts an LLM to select which output is preferred, given a specific input.\n",
+                "\n",
+                "For this evaluation, we will need 3 things:\n",
+                "1. An evaluator\n",
+                "2. A dataset of inputs\n",
+                "3. 2 (or more) LLMs, Chains, or Agents to compare\n",
+                "\n",
+                "Then we will aggregate the restults to determine the preferred model.\n",
+                "\n",
+                "### Step 1. Create the Evaluator\n",
+                "\n",
+                "In this example, you will use gpt-4 to select which output is preferred."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "eval_chain = load_evaluator(\"pairwise_string\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "### Step 2. Select Dataset\n",
+                "\n",
+                "If you already have real usage data for your LLM, you can use a representative sample. More examples\n",
+                "provide more reliable results. We will use some example queries someone might have about how to use langchain here."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stderr",
+                    "output_type": "stream",
+                    "text": [
+                        "Found cached dataset parquet (/Users/wfh/.cache/huggingface/datasets/LangChainDatasets___parquet/LangChainDatasets--langchain-howto-queries-bbb748bbee7e77aa/0.0.0/14a00e99c0d15a23649d0db8944380ac81082d4b021f398733dd84f3a6c569a7)\n"
+                    ]
+                },
+                {
+                    "data": {
+                        "application/vnd.jupyter.widget-view+json": {
+                            "model_id": "a2358d37246640ce95e0f9940194590a",
+                            "version_major": 2,
+                            "version_minor": 0
+                        },
+                        "text/plain": [
+                            "  0%|          | 0/1 [00:00<?, ?it/s]"
+                        ]
+                    },
+                    "metadata": {},
+                    "output_type": "display_data"
+                }
+            ],
+            "source": [
+                "from langchain.evaluation.loading import load_dataset\n",
+                "\n",
+                "dataset = load_dataset(\"langchain-howto-queries\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "### Step 3. Define Models to Compare\n",
+                "\n",
+                "We will be comparing two agents in this case."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.utilities import SerpAPIWrapper\n",
+                "from langchain.agents import initialize_agent, Tool\n",
+                "from langchain.agents import AgentType\n",
+                "from langchain.chat_models import ChatOpenAI\n",
+                "\n",
+                "\n",
+                "# Initialize the language model\n",
+                "# You can add your own OpenAI API key by adding openai_api_key=\"<your_api_key>\"\n",
+                "llm = ChatOpenAI(temperature=0, model=\"gpt-3.5-turbo-0613\")\n",
+                "\n",
+                "# Initialize the SerpAPIWrapper for search functionality\n",
+                "# Replace <your_api_key> in openai_api_key=\"<your_api_key>\" with your actual SerpAPI key.\n",
+                "search = SerpAPIWrapper()\n",
+                "\n",
+                "# Define a list of tools offered by the agent\n",
+                "tools = [\n",
+                "    Tool(\n",
+                "        name=\"Search\",\n",
+                "        func=search.run,\n",
+                "        coroutine=search.arun,\n",
+                "        description=\"Useful when you need to answer questions about current events. You should ask targeted questions.\",\n",
+                "    ),\n",
+                "]"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "functions_agent = initialize_agent(\n",
+                "    tools, llm, agent=AgentType.OPENAI_MULTI_FUNCTIONS, verbose=False\n",
+                ")\n",
+                "conversations_agent = initialize_agent(\n",
+                "    tools, llm, agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION, verbose=False\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "### Step 4. Generate Responses\n",
+                "\n",
+                "We will generate outputs for each of the models before evaluating them."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "application/vnd.jupyter.widget-view+json": {
+                            "model_id": "87277cb39a1a4726bb7cc533a24e2ea4",
+                            "version_major": 2,
+                            "version_minor": 0
+                        },
+                        "text/plain": [
+                            "  0%|          | 0/20 [00:00<?, ?it/s]"
+                        ]
+                    },
+                    "metadata": {},
+                    "output_type": "display_data"
+                }
+            ],
+            "source": [
+                "from tqdm.notebook import tqdm\n",
+                "import asyncio\n",
+                "\n",
+                "results = []\n",
+                "agents = [functions_agent, conversations_agent]\n",
+                "concurrency_level = 6  # How many concurrent agents to run. May need to decrease if OpenAI is rate limiting.\n",
+                "\n",
+                "# We will only run the first 20 examples of this dataset to speed things up\n",
+                "# This will lead to larger confidence intervals downstream.\n",
+                "batch = []\n",
+                "for example in tqdm(dataset[:20]):\n",
+                "    batch.extend([agent.acall(example[\"inputs\"]) for agent in agents])\n",
+                "    if len(batch) >= concurrency_level:\n",
+                "        batch_results = await asyncio.gather(*batch, return_exceptions=True)\n",
+                "        results.extend(list(zip(*[iter(batch_results)] * 2)))\n",
+                "        batch = []\n",
+                "if batch:\n",
+                "    batch_results = await asyncio.gather(*batch, return_exceptions=True)\n",
+                "    results.extend(list(zip(*[iter(batch_results)] * 2)))"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "## Step 5. Evaluate Pairs\n",
+                "\n",
+                "Now it's time to evaluate the results. For each agent response, run the evaluation chain to select which output is preferred (or return a tie).\n",
+                "\n",
+                "Randomly select the input order to reduce the likelihood that one model will be preferred just because it is presented first."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "import random\n",
+                "\n",
+                "\n",
+                "def predict_preferences(dataset, results) -> list:\n",
+                "    preferences = []\n",
+                "\n",
+                "    for example, (res_a, res_b) in zip(dataset, results):\n",
+                "        input_ = example[\"inputs\"]\n",
+                "        # Flip a coin to reduce persistent position bias\n",
+                "        if random.random() < 0.5:\n",
+                "            pred_a, pred_b = res_a, res_b\n",
+                "            a, b = \"a\", \"b\"\n",
+                "        else:\n",
+                "            pred_a, pred_b = res_b, res_a\n",
+                "            a, b = \"b\", \"a\"\n",
+                "        eval_res = eval_chain.evaluate_string_pairs(\n",
+                "            prediction=pred_a[\"output\"] if isinstance(pred_a, dict) else str(pred_a),\n",
+                "            prediction_b=pred_b[\"output\"] if isinstance(pred_b, dict) else str(pred_b),\n",
+                "            input=input_,\n",
+                "        )\n",
+                "        if eval_res[\"value\"] == \"A\":\n",
+                "            preferences.append(a)\n",
+                "        elif eval_res[\"value\"] == \"B\":\n",
+                "            preferences.append(b)\n",
+                "        else:\n",
+                "            preferences.append(None)  # No preference\n",
+                "    return preferences"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "metadata": {},
+            "outputs": [],
+            "source": [
+                "preferences = predict_preferences(dataset, results)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "**Print out the ratio of preferences.**"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 8,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "OpenAI Functions Agent: 95.00%\n",
+                        "None: 5.00%\n"
+                    ]
+                }
+            ],
+            "source": [
+                "from collections import Counter\n",
+                "\n",
+                "name_map = {\n",
+                "    \"a\": \"OpenAI Functions Agent\",\n",
+                "    \"b\": \"Structured Chat Agent\",\n",
+                "}\n",
+                "counts = Counter(preferences)\n",
+                "pref_ratios = {k: v / len(preferences) for k, v in counts.items()}\n",
+                "for k, v in pref_ratios.items():\n",
+                "    print(f\"{name_map.get(k)}: {v:.2%}\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "### Estimate Confidence Intervals\n",
+                "\n",
+                "The results seem pretty clear, but if you want to have a better sense of how confident we are, that model \"A\" (the OpenAI Functions Agent) is the preferred model, we can calculate confidence intervals. \n",
+                "\n",
+                "Below, use the Wilson score to estimate the confidence interval."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 9,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from math import sqrt\n",
+                "\n",
+                "\n",
+                "def wilson_score_interval(\n",
+                "    preferences: list, which: str = \"a\", z: float = 1.96\n",
+                ") -> tuple:\n",
+                "    \"\"\"Estimate the confidence interval using the Wilson score.\n",
+                "\n",
+                "    See: https://en.wikipedia.org/wiki/Binomial_proportion_confidence_interval#Wilson_score_interval\n",
+                "    for more details, including when to use it and when it should not be used.\n",
+                "    \"\"\"\n",
+                "    total_preferences = preferences.count(\"a\") + preferences.count(\"b\")\n",
+                "    n_s = preferences.count(which)\n",
+                "\n",
+                "    if total_preferences == 0:\n",
+                "        return (0, 0)\n",
+                "\n",
+                "    p_hat = n_s / total_preferences\n",
+                "\n",
+                "    denominator = 1 + (z**2) / total_preferences\n",
+                "    adjustment = (z / denominator) * sqrt(\n",
+                "        p_hat * (1 - p_hat) / total_preferences\n",
+                "        + (z**2) / (4 * total_preferences * total_preferences)\n",
+                "    )\n",
+                "    center = (p_hat + (z**2) / (2 * total_preferences)) / denominator\n",
+                "    lower_bound = min(max(center - adjustment, 0.0), 1.0)\n",
+                "    upper_bound = min(max(center + adjustment, 0.0), 1.0)\n",
+                "\n",
+                "    return (lower_bound, upper_bound)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 10,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "The \"OpenAI Functions Agent\" would be preferred between 83.18% and 100.00% percent of the time (with 95% confidence).\n",
+                        "The \"Structured Chat Agent\" would be preferred between 0.00% and 16.82% percent of the time (with 95% confidence).\n"
+                    ]
+                }
+            ],
+            "source": [
+                "for which_, name in name_map.items():\n",
+                "    low, high = wilson_score_interval(preferences, which=which_)\n",
+                "    print(\n",
+                "        f'The \"{name}\" would be preferred between {low:.2%} and {high:.2%} percent of the time (with 95% confidence).'\n",
+                "    )"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "**Print out the p-value.**"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 11,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "The p-value is 0.00000. If the null hypothesis is true (i.e., if the selected eval chain actually has no preference between the models),\n",
+                        "then there is a 0.00038% chance of observing the OpenAI Functions Agent be preferred at least 19\n",
+                        "times out of 19 trials.\n"
+                    ]
+                },
+                {
+                    "name": "stderr",
+                    "output_type": "stream",
+                    "text": [
+                        "/var/folders/gf/6rnp_mbx5914kx7qmmh7xzmw0000gn/T/ipykernel_15978/384907688.py:6: DeprecationWarning: 'binom_test' is deprecated in favour of 'binomtest' from version 1.7.0 and will be removed in Scipy 1.12.0.\n",
+                        "  p_value = stats.binom_test(successes, n, p=0.5, alternative=\"two-sided\")\n"
+                    ]
+                }
+            ],
+            "source": [
+                "from scipy import stats\n",
+                "\n",
+                "preferred_model = max(pref_ratios, key=pref_ratios.get)\n",
+                "successes = preferences.count(preferred_model)\n",
+                "n = len(preferences) - preferences.count(None)\n",
+                "p_value = stats.binom_test(successes, n, p=0.5, alternative=\"two-sided\")\n",
+                "print(\n",
+                "    f\"\"\"The p-value is {p_value:.5f}. If the null hypothesis is true (i.e., if the selected eval chain actually has no preference between the models),\n",
+                "then there is a {p_value:.5%} chance of observing the {name_map.get(preferred_model)} be preferred at least {successes}\n",
+                "times out of {n} trials.\"\"\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "<a name=\"cite_note-1\"></a>_1. Note: Automated evals are still an open research topic and are best used alongside other evaluation approaches. \n",
+                "LLM preferences exhibit biases, including banal ones like the order of outputs.\n",
+                "In choosing preferences, \"ground truth\" may not be taken into account, which may lead to scores that aren't grounded in utility._"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": null,
+            "metadata": {},
+            "outputs": [],
+            "source": []
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
     },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "a2358d37246640ce95e0f9940194590a",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "  0%|          | 0/1 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
-   "source": [
-    "from langchain.evaluation.loading import load_dataset\n",
-    "\n",
-    "dataset = load_dataset(\"langchain-howto-queries\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Step 3. Define Models to Compare\n",
-    "\n",
-    "We will be comparing two agents in this case."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.utilities import SerpAPIWrapper\n",
-    "from langchain.agents import initialize_agent, Tool\n",
-    "from langchain.agents import AgentType\n",
-    "from langchain.chat_models import ChatOpenAI\n",
-    "\n",
-    "\n",
-    "# Initialize the language model\n",
-    "# You can add your own OpenAI API key by adding openai_api_key=\"<your_api_key>\"\n",
-    "llm = ChatOpenAI(temperature=0, model=\"gpt-3.5-turbo-0613\")\n",
-    "\n",
-    "# Initialize the SerpAPIWrapper for search functionality\n",
-    "# Replace <your_api_key> in openai_api_key=\"<your_api_key>\" with your actual SerpAPI key.\n",
-    "search = SerpAPIWrapper()\n",
-    "\n",
-    "# Define a list of tools offered by the agent\n",
-    "tools = [\n",
-    "    Tool(\n",
-    "        name=\"Search\",\n",
-    "        func=search.run,\n",
-    "        coroutine=search.arun,\n",
-    "        description=\"Useful when you need to answer questions about current events. You should ask targeted questions.\",\n",
-    "    ),\n",
-    "]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "functions_agent = initialize_agent(\n",
-    "    tools, llm, agent=AgentType.OPENAI_MULTI_FUNCTIONS, verbose=False\n",
-    ")\n",
-    "conversations_agent = initialize_agent(\n",
-    "    tools, llm, agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION, verbose=False\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Step 4. Generate Responses\n",
-    "\n",
-    "We will generate outputs for each of the models before evaluating them."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "87277cb39a1a4726bb7cc533a24e2ea4",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "  0%|          | 0/20 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
-   "source": [
-    "from tqdm.notebook import tqdm\n",
-    "import asyncio\n",
-    "\n",
-    "results = []\n",
-    "agents = [functions_agent, conversations_agent]\n",
-    "concurrency_level = 6  # How many concurrent agents to run. May need to decrease if OpenAI is rate limiting.\n",
-    "\n",
-    "# We will only run the first 20 examples of this dataset to speed things up\n",
-    "# This will lead to larger confidence intervals downstream.\n",
-    "batch = []\n",
-    "for example in tqdm(dataset[:20]):\n",
-    "    batch.extend([agent.acall(example[\"inputs\"]) for agent in agents])\n",
-    "    if len(batch) >= concurrency_level:\n",
-    "        batch_results = await asyncio.gather(*batch, return_exceptions=True)\n",
-    "        results.extend(list(zip(*[iter(batch_results)] * 2)))\n",
-    "        batch = []\n",
-    "if batch:\n",
-    "    batch_results = await asyncio.gather(*batch, return_exceptions=True)\n",
-    "    results.extend(list(zip(*[iter(batch_results)] * 2)))"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Step 5. Evaluate Pairs\n",
-    "\n",
-    "Now it's time to evaluate the results. For each agent response, run the evaluation chain to select which output is preferred (or return a tie).\n",
-    "\n",
-    "Randomly select the input order to reduce the likelihood that one model will be preferred just because it is presented first."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "import random\n",
-    "\n",
-    "\n",
-    "def predict_preferences(dataset, results) -> list:\n",
-    "    preferences = []\n",
-    "\n",
-    "    for example, (res_a, res_b) in zip(dataset, results):\n",
-    "        input_ = example[\"inputs\"]\n",
-    "        # Flip a coin to reduce persistent position bias\n",
-    "        if random.random() < 0.5:\n",
-    "            pred_a, pred_b = res_a, res_b\n",
-    "            a, b = \"a\", \"b\"\n",
-    "        else:\n",
-    "            pred_a, pred_b = res_b, res_a\n",
-    "            a, b = \"b\", \"a\"\n",
-    "        eval_res = eval_chain.evaluate_string_pairs(\n",
-    "            prediction=pred_a[\"output\"] if isinstance(pred_a, dict) else str(pred_a),\n",
-    "            prediction_b=pred_b[\"output\"] if isinstance(pred_b, dict) else str(pred_b),\n",
-    "            input=input_,\n",
-    "        )\n",
-    "        if eval_res[\"value\"] == \"A\":\n",
-    "            preferences.append(a)\n",
-    "        elif eval_res[\"value\"] == \"B\":\n",
-    "            preferences.append(b)\n",
-    "        else:\n",
-    "            preferences.append(None)  # No preference\n",
-    "    return preferences"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "preferences = predict_preferences(dataset, results)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "**Print out the ratio of preferences.**"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "OpenAI Functions Agent: 95.00%\n",
-      "None: 5.00%\n"
-     ]
-    }
-   ],
-   "source": [
-    "from collections import Counter\n",
-    "\n",
-    "name_map = {\n",
-    "    \"a\": \"OpenAI Functions Agent\",\n",
-    "    \"b\": \"Structured Chat Agent\",\n",
-    "}\n",
-    "counts = Counter(preferences)\n",
-    "pref_ratios = {k: v / len(preferences) for k, v in counts.items()}\n",
-    "for k, v in pref_ratios.items():\n",
-    "    print(f\"{name_map.get(k)}: {v:.2%}\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Estimate Confidence Intervals\n",
-    "\n",
-    "The results seem pretty clear, but if you want to have a better sense of how confident we are, that model \"A\" (the OpenAI Functions Agent) is the preferred model, we can calculate confidence intervals. \n",
-    "\n",
-    "Below, use the Wilson score to estimate the confidence interval."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from math import sqrt\n",
-    "\n",
-    "\n",
-    "def wilson_score_interval(\n",
-    "    preferences: list, which: str = \"a\", z: float = 1.96\n",
-    ") -> tuple:\n",
-    "    \"\"\"Estimate the confidence interval using the Wilson score.\n",
-    "\n",
-    "    See: https://en.wikipedia.org/wiki/Binomial_proportion_confidence_interval#Wilson_score_interval\n",
-    "    for more details, including when to use it and when it should not be used.\n",
-    "    \"\"\"\n",
-    "    total_preferences = preferences.count(\"a\") + preferences.count(\"b\")\n",
-    "    n_s = preferences.count(which)\n",
-    "\n",
-    "    if total_preferences == 0:\n",
-    "        return (0, 0)\n",
-    "\n",
-    "    p_hat = n_s / total_preferences\n",
-    "\n",
-    "    denominator = 1 + (z**2) / total_preferences\n",
-    "    adjustment = (z / denominator) * sqrt(\n",
-    "        p_hat * (1 - p_hat) / total_preferences\n",
-    "        + (z**2) / (4 * total_preferences * total_preferences)\n",
-    "    )\n",
-    "    center = (p_hat + (z**2) / (2 * total_preferences)) / denominator\n",
-    "    lower_bound = min(max(center - adjustment, 0.0), 1.0)\n",
-    "    upper_bound = min(max(center + adjustment, 0.0), 1.0)\n",
-    "\n",
-    "    return (lower_bound, upper_bound)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The \"OpenAI Functions Agent\" would be preferred between 83.18% and 100.00% percent of the time (with 95% confidence).\n",
-      "The \"Structured Chat Agent\" would be preferred between 0.00% and 16.82% percent of the time (with 95% confidence).\n"
-     ]
-    }
-   ],
-   "source": [
-    "for which_, name in name_map.items():\n",
-    "    low, high = wilson_score_interval(preferences, which=which_)\n",
-    "    print(\n",
-    "        f'The \"{name}\" would be preferred between {low:.2%} and {high:.2%} percent of the time (with 95% confidence).'\n",
-    "    )"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "**Print out the p-value.**"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The p-value is 0.00000. If the null hypothesis is true (i.e., if the selected eval chain actually has no preference between the models),\n",
-      "then there is a 0.00038% chance of observing the OpenAI Functions Agent be preferred at least 19\n",
-      "times out of 19 trials.\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "/var/folders/gf/6rnp_mbx5914kx7qmmh7xzmw0000gn/T/ipykernel_15978/384907688.py:6: DeprecationWarning: 'binom_test' is deprecated in favour of 'binomtest' from version 1.7.0 and will be removed in Scipy 1.12.0.\n",
-      "  p_value = stats.binom_test(successes, n, p=0.5, alternative=\"two-sided\")\n"
-     ]
-    }
-   ],
-   "source": [
-    "from scipy import stats\n",
-    "\n",
-    "preferred_model = max(pref_ratios, key=pref_ratios.get)\n",
-    "successes = preferences.count(preferred_model)\n",
-    "n = len(preferences) - preferences.count(None)\n",
-    "p_value = stats.binom_test(successes, n, p=0.5, alternative=\"two-sided\")\n",
-    "print(\n",
-    "    f\"\"\"The p-value is {p_value:.5f}. If the null hypothesis is true (i.e., if the selected eval chain actually has no preference between the models),\n",
-    "then there is a {p_value:.5%} chance of observing the {name_map.get(preferred_model)} be preferred at least {successes}\n",
-    "times out of {n} trials.\"\"\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "<a name=\"cite_note-1\"></a>_1. Note: Automated evals are still an open research topic and are best used alongside other evaluation approaches. \n",
-    "LLM preferences exhibit biases, including banal ones like the order of outputs.\n",
-    "In choosing preferences, \"ground truth\" may not be taken into account, which may lead to scores that aren't grounded in utility._"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 4
-}
+    "nbformat": 4,
+    "nbformat_minor": 4
+}

docs/extras/guides/evaluation/string/Untitled.ipynb

@@ -1,318 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "bce7335e-f3b2-44f3-90cc-8c0a23a89a21",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os\n",
-    "from langchain.agents import load_tools\n",
-    "from langchain.agents import initialize_agent\n",
-    "from langchain.chat_models import ChatOpenAI\n",
-    "from langchain.utilities import GoogleSearchAPIWrapper\n",
-    "from langchain.schema import (\n",
-    "    SystemMessage,\n",
-    "    HumanMessage,\n",
-    "    AIMessage\n",
-    ")\n",
-    "\n",
-    "# os.environ[\"LANGCHAIN_TRACING_V2\"] = \"true\"\n",
-    "# os.environ[\"LANGCHAIN_ENDPOINT\"] = \"https://api.smith.langchain.com\"\n",
-    "# os.environ[\"LANGCHAIN_API_KEY\"] = \"******\"\n",
-    "# os.environ[\"LANGCHAIN_PROJECT\"] = \"Jarvis\"\n",
-    "\n",
-    "\n",
-    "prefix_messages = [{\"role\": \"system\", \"content\": \"You are a helpful discord Chatbot.\"}]\n",
-    "\n",
-    "llm = ChatOpenAI(model_name='gpt-3.5-turbo', \n",
-    "             temperature=0.5, \n",
-    "             max_tokens = 2000)\n",
-    "tools = load_tools([\"serpapi\", \"llm-math\"], llm=llm)\n",
-    "agent = initialize_agent(tools,\n",
-    "                         llm,\n",
-    "                         agent=\"zero-shot-react-description\",\n",
-    "                         verbose=True,\n",
-    "                         handle_parsing_errors=True\n",
-    "                         )\n",
-    "\n",
-    "\n",
-    "async def on_ready():\n",
-    "    print(f'{bot.user} has connected to Discord!')\n",
-    "\n",
-    "async def on_message(message):\n",
-    "\n",
-    "    print(\"Detected bot name in message:\", message.content)\n",
-    "\n",
-    "    # Capture the output of agent.run() in the response variable\n",
-    "    response = agent.run(message.content)\n",
-    "\n",
-    "    while response:\n",
-    "        print(response)\n",
-    "        chunk, response = response[:2000], response[2000:]\n",
-    "        print(f\"Chunk: {chunk}\")\n",
-    "    print(\"Response sent.\")\n",
-    "\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "1551ce9f-b6de-4035-b6d6-825722823b48",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from dataclasses import dataclass\n",
-    "@dataclass\n",
-    "class Message:\n",
-    "    content: str"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "6e6859ec-8544-4407-9663-6b53c0092903",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Detected bot name in message: Hi AI, how are you today?\n",
-      "\n",
-      "\n",
-      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
-      "\u001b[32;1m\u001b[1;3mThis question is not something that can be answered using the available tools.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3mI need to follow the correct format for answering questions.\n",
-      "Action: N/A\u001b[0m\n",
-      "Observation: Invalid Format: Missing 'Action Input:' after 'Action:'\n",
-      "Thought:\u001b[32;1m\u001b[1;3m\u001b[0m\n",
-      "\n",
-      "\u001b[1m> Finished chain.\u001b[0m\n",
-      "Agent stopped due to iteration limit or time limit.\n",
-      "Chunk: Agent stopped due to iteration limit or time limit.\n",
-      "Response sent.\n"
-     ]
-    }
-   ],
-   "source": [
-    "await on_message(Message(content=\"Hi AI, how are you today?\"))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "b850294c-7f8f-4e79-adcf-47e4e3a898df",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langsmith import Client\n",
-    "\n",
-    "client = Client()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "6d089ddc-69bc-45a8-b8db-9962e4f1f5ee",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from itertools import islice\n",
-    "\n",
-    "runs = list(islice(client.list_runs(), 10))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 38,
-   "id": "f0349fac-5a98-400f-ba03-61ed4e1332be",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "runs = sorted(runs, key=lambda x: x.start_time, reverse=True)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "02f133f0-39ee-4b46-b443-12c1f9b76fff",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "ids = [run.id for run in runs]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 39,
-   "id": "3366dce4-0c38-4a7d-8111-046a58b24917",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "runs2 = list(client.list_runs(id=ids))\n",
-    "runs2 = sorted(runs2, key=lambda x: x.start_time, reverse=True)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 42,
-   "id": "82915b90-39a0-47d6-9121-56a13f210f52",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['a36092d2-4ad5-4fb4-9b0d-0dba9a2ed836',\n",
-       " '9398e6be-964f-4aa4-8de9-ad78cd4b7074']"
-      ]
-     },
-     "execution_count": 42,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "[str(x) for x in ids[:2]]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 48,
-   "id": "f610ec91-dc48-4a17-91c5-5c4675c77abc",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langsmith.run_helpers import traceable\n",
-    "\n",
-    "@traceable(run_type=\"llm\", name=\"\"\"<iframe width=\"560\" height=\"315\" src=\"https://www.youtube.com/embed/dQw4w9WgXcQ?start=5\" title=\"YouTube video player\" frameborder=\"0\" allow=\"accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share\" allowfullscreen></iframe>\"\"\")\n",
-    "def foo():\n",
-    "    return \"bar\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 49,
-   "id": "bd317bd7-8b2a-433a-8ec3-098a84ba8e64",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'bar'"
-      ]
-     },
-     "execution_count": 49,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "foo()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 52,
-   "id": "b142519b-6885-415c-83b9-4a346fb90589",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.llms import AzureOpenAI"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "5c50bb2b-72b8-4322-9b16-d857ecd9f347",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

docs/extras/guides/evaluation/string/criteria_eval_chain.ipynb

@@ -1,468 +1,469 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "4cf569a7-9a1d-4489-934e-50e57760c907",
-   "metadata": {},
-   "source": [
-    "# Criteria Evaluation\n",
-    "\n",
-    "In scenarios where you wish to assess a model's output using a specific rubric or criteria set, the `criteria` evaluator proves to be a handy tool. It allows you to verify if an LLM or Chain's output complies with a defined set of criteria.\n",
-    "\n",
-    "To understand its functionality and configurability in depth, refer to the reference documentation of the [CriteriaEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.html#langchain.evaluation.criteria.eval_chain.CriteriaEvalChain) class.\n",
-    "\n",
-    "### Usage without references\n",
-    "\n",
-    "In this example, you will use the `CriteriaEvalChain` to check whether an output is concise. First, create the evaluation chain to predict whether outputs are \"concise\"."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "6005ebe8-551e-47a5-b4df-80575a068552",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"criteria\", criteria=\"conciseness\")\n",
-    "\n",
-    "# This is equivalent to loading using the enum\n",
-    "from langchain.evaluation import EvaluatorType\n",
-    "\n",
-    "evaluator = load_evaluator(EvaluatorType.CRITERIA, criteria=\"conciseness\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "22f83fb8-82f4-4310-a877-68aaa0789199",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'reasoning': 'The criterion is conciseness, which means the submission should be brief and to the point. \\n\\nLooking at the submission, the answer to the question \"What\\'s 2+2?\" is indeed \"four\". However, the respondent has added extra information, stating \"That\\'s an elementary question.\" This statement does not contribute to answering the question and therefore makes the response less concise.\\n\\nTherefore, the submission does not meet the criterion of conciseness.\\n\\nN', 'value': 'N', 'score': 0}\n"
-     ]
-    }
-   ],
-   "source": [
-    "eval_result = evaluator.evaluate_strings(\n",
-    "    prediction=\"What's 2+2? That's an elementary question. The answer you're looking for is that two and two is four.\",\n",
-    "    input=\"What's 2+2?\",\n",
-    ")\n",
-    "print(eval_result)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "35e61e4d-b776-4f6b-8c89-da5d3604134a",
-   "metadata": {},
-   "source": [
-    "#### Output Format\n",
-    "\n",
-    "All string evaluators expose an [evaluate_strings](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.html?highlight=evaluate_strings#langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.evaluate_strings) (or async [aevaluate_strings](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.html?highlight=evaluate_strings#langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.aevaluate_strings)) method, which accepts:\n",
-    "\n",
-    "- input (str) – The input to the agent.\n",
-    "- prediction (str) – The predicted response.\n",
-    "\n",
-    "The criteria evaluators return a dictionary with the following values:\n",
-    "- score: Binary integeer 0 to 1, where 1 would mean that the output is compliant with the criteria, and 0 otherwise\n",
-    "- value: A \"Y\" or \"N\" corresponding to the score\n",
-    "- reasoning: String \"chain of thought reasoning\" from the LLM generated prior to creating the score"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "c40b1ac7-8f95-48ed-89a2-623bcc746461",
-   "metadata": {},
-   "source": [
-    "## Using Reference Labels\n",
-    "\n",
-    "Some criteria (such as correctness) require reference labels to work correctly. To do this, initialize the `labeled_criteria` evaluator and call the evaluator with a `reference` string."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "20d8a86b-beba-42ce-b82c-d9e5ebc13686",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "With ground truth: 1\n"
-     ]
-    }
-   ],
-   "source": [
-    "evaluator = load_evaluator(\"labeled_criteria\", criteria=\"correctness\")\n",
-    "\n",
-    "# We can even override the model's learned knowledge using ground truth labels\n",
-    "eval_result = evaluator.evaluate_strings(\n",
-    "    input=\"What is the capital of the US?\",\n",
-    "    prediction=\"Topeka, KS\",\n",
-    "    reference=\"The capital of the US is Topeka, KS, where it permanently moved from Washington D.C. on May 16, 2023\",\n",
-    ")\n",
-    "print(f'With ground truth: {eval_result[\"score\"]}')"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "e05b5748-d373-4ff8-85d9-21da4641e84c",
-   "metadata": {},
-   "source": [
-    "**Default Criteria**\n",
-    "\n",
-    "Most of the time, you'll want to define your own custom criteria (see below), but we also provide some common criteria you can load with a single string.\n",
-    "Here's a list of pre-implemented criteria. Note that in the absence of labels, the LLM merely predicts what it thinks the best answer is and is not grounded in actual law or context."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "47de7359-db3e-4cad-bcfa-4fe834dea893",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[<Criteria.CONCISENESS: 'conciseness'>,\n",
-       " <Criteria.RELEVANCE: 'relevance'>,\n",
-       " <Criteria.CORRECTNESS: 'correctness'>,\n",
-       " <Criteria.COHERENCE: 'coherence'>,\n",
-       " <Criteria.HARMFULNESS: 'harmfulness'>,\n",
-       " <Criteria.MALICIOUSNESS: 'maliciousness'>,\n",
-       " <Criteria.HELPFULNESS: 'helpfulness'>,\n",
-       " <Criteria.CONTROVERSIALITY: 'controversiality'>,\n",
-       " <Criteria.MISOGYNY: 'misogyny'>,\n",
-       " <Criteria.CRIMINALITY: 'criminality'>,\n",
-       " <Criteria.INSENSITIVITY: 'insensitivity'>]"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from langchain.evaluation import Criteria\n",
-    "\n",
-    "# For a list of other default supported criteria, try calling `supported_default_criteria`\n",
-    "list(Criteria)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "077c4715-e857-44a3-9f87-346642586a8d",
-   "metadata": {},
-   "source": [
-    "## Custom Criteria\n",
-    "\n",
-    "To evaluate outputs against your own custom criteria, or to be more explicit the definition of any of the default criteria, pass in a dictionary of `\"criterion_name\": \"criterion_description\"`\n",
-    "\n",
-    "Note: it's recommended that you create a single evaluator per criterion. This way, separate feedback can be provided for each aspect. Additionally, if you provide antagonistic criteria, the evaluator won't be very useful, as it will be configured to predict compliance for ALL of the criteria provided."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 19,
-   "id": "bafa0a11-2617-4663-84bf-24df7d0736be",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'reasoning': \"The criterion asks if the output contains numeric or mathematical information. The joke in the submission does contain mathematical information. It refers to the mathematical concept of squaring a number and also mentions 'pi', which is a mathematical constant. Therefore, the submission does meet the criterion.\\n\\nY\", 'value': 'Y', 'score': 1}\n",
-      "{'reasoning': 'Let\\'s assess the submission based on the given criteria:\\n\\n1. Numeric: The output does not contain any explicit numeric information. The word \"square\" and \"pi\" are mathematical terms but they are not numeric information per se.\\n\\n2. Mathematical: The output does contain mathematical information. The terms \"square\" and \"pi\" are mathematical terms. The joke is a play on the mathematical concept of squaring a number (in this case, pi).\\n\\n3. Grammatical: The output is grammatically correct. The sentence structure, punctuation, and word usage are all correct.\\n\\n4. Logical: The output is logical. It makes sense within the context of the joke. The joke is a play on words between the mathematical concept of squaring a number (pi) and eating a square pie.\\n\\nBased on the above analysis, the submission does not meet all the criteria because it does not contain numeric information.\\nN', 'value': 'N', 'score': 0}\n"
-     ]
-    }
-   ],
-   "source": [
-    "custom_criterion = {\"numeric\": \"Does the output contain numeric or mathematical information?\"}\n",
-    "\n",
-    "eval_chain = load_evaluator(\n",
-    "    EvaluatorType.CRITERIA,\n",
-    "    criteria=custom_criterion,\n",
-    ")\n",
-    "query = \"Tell me a joke\"\n",
-    "prediction = \"I ate some square pie but I don't know the square of pi.\"\n",
-    "eval_result = eval_chain.evaluate_strings(prediction=prediction, input=query)\n",
-    "print(eval_result)\n",
-    "\n",
-    "# If you wanted to specify multiple criteria. Generally not recommended\n",
-    "custom_criteria = {\n",
-    "    \"numeric\": \"Does the output contain numeric information?\",\n",
-    "    \"mathematical\": \"Does the output contain mathematical information?\",\n",
-    "    \"grammatical\": \"Is the output grammatically correct?\",\n",
-    "    \"logical\": \"Is the output logical?\",\n",
-    "}\n",
-    "\n",
-    "eval_chain = load_evaluator(\n",
-    "    EvaluatorType.CRITERIA,\n",
-    "    criteria=custom_criteria,\n",
-    ")\n",
-    "eval_result = eval_chain.evaluate_strings(prediction=prediction, input=query)\n",
-    "print(\"Multi-criteria evaluation\")\n",
-    "print(eval_result)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "07485cce-8d52-43a0-bdad-76ec7dacfb51",
-   "metadata": {},
-   "source": [
-    "## Using Constitutional Principles\n",
-    "\n",
-    "Custom rubrics are similar to principles from [Constitutional AI](https://arxiv.org/abs/2212.08073). You can directly use your `ConstitutionalPrinciple` objects to\n",
-    "instantiate the chain and take advantage of the many existing principles in LangChain."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "99e3c242-5b12-4bd5-b487-64990a159655",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "54 available principles\n"
-     ]
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "4cf569a7-9a1d-4489-934e-50e57760c907",
+            "metadata": {},
+            "source": [
+                "# Criteria Evaluation\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/string/criteria_eval_chain.ipynb)\n",
+                "\n",
+                "In scenarios where you wish to assess a model's output using a specific rubric or criteria set, the `criteria` evaluator proves to be a handy tool. It allows you to verify if an LLM or Chain's output complies with a defined set of criteria.\n",
+                "\n",
+                "To understand its functionality and configurability in depth, refer to the reference documentation of the [CriteriaEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.html#langchain.evaluation.criteria.eval_chain.CriteriaEvalChain) class.\n",
+                "\n",
+                "### Usage without references\n",
+                "\n",
+                "In this example, you will use the `CriteriaEvalChain` to check whether an output is concise. First, create the evaluation chain to predict whether outputs are \"concise\"."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "6005ebe8-551e-47a5-b4df-80575a068552",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"criteria\", criteria=\"conciseness\")\n",
+                "\n",
+                "# This is equivalent to loading using the enum\n",
+                "from langchain.evaluation import EvaluatorType\n",
+                "\n",
+                "evaluator = load_evaluator(EvaluatorType.CRITERIA, criteria=\"conciseness\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "22f83fb8-82f4-4310-a877-68aaa0789199",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "{'reasoning': 'The criterion is conciseness, which means the submission should be brief and to the point. \\n\\nLooking at the submission, the answer to the question \"What\\'s 2+2?\" is indeed \"four\". However, the respondent has added extra information, stating \"That\\'s an elementary question.\" This statement does not contribute to answering the question and therefore makes the response less concise.\\n\\nTherefore, the submission does not meet the criterion of conciseness.\\n\\nN', 'value': 'N', 'score': 0}\n"
+                    ]
+                }
+            ],
+            "source": [
+                "eval_result = evaluator.evaluate_strings(\n",
+                "    prediction=\"What's 2+2? That's an elementary question. The answer you're looking for is that two and two is four.\",\n",
+                "    input=\"What's 2+2?\",\n",
+                ")\n",
+                "print(eval_result)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "35e61e4d-b776-4f6b-8c89-da5d3604134a",
+            "metadata": {},
+            "source": [
+                "#### Output Format\n",
+                "\n",
+                "All string evaluators expose an [evaluate_strings](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.html?highlight=evaluate_strings#langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.evaluate_strings) (or async [aevaluate_strings](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.html?highlight=evaluate_strings#langchain.evaluation.criteria.eval_chain.CriteriaEvalChain.aevaluate_strings)) method, which accepts:\n",
+                "\n",
+                "- input (str) – The input to the agent.\n",
+                "- prediction (str) – The predicted response.\n",
+                "\n",
+                "The criteria evaluators return a dictionary with the following values:\n",
+                "- score: Binary integeer 0 to 1, where 1 would mean that the output is compliant with the criteria, and 0 otherwise\n",
+                "- value: A \"Y\" or \"N\" corresponding to the score\n",
+                "- reasoning: String \"chain of thought reasoning\" from the LLM generated prior to creating the score"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "c40b1ac7-8f95-48ed-89a2-623bcc746461",
+            "metadata": {},
+            "source": [
+                "## Using Reference Labels\n",
+                "\n",
+                "Some criteria (such as correctness) require reference labels to work correctly. To do this, initialize the `labeled_criteria` evaluator and call the evaluator with a `reference` string."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "20d8a86b-beba-42ce-b82c-d9e5ebc13686",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "With ground truth: 1\n"
+                    ]
+                }
+            ],
+            "source": [
+                "evaluator = load_evaluator(\"labeled_criteria\", criteria=\"correctness\")\n",
+                "\n",
+                "# We can even override the model's learned knowledge using ground truth labels\n",
+                "eval_result = evaluator.evaluate_strings(\n",
+                "    input=\"What is the capital of the US?\",\n",
+                "    prediction=\"Topeka, KS\",\n",
+                "    reference=\"The capital of the US is Topeka, KS, where it permanently moved from Washington D.C. on May 16, 2023\",\n",
+                ")\n",
+                "print(f'With ground truth: {eval_result[\"score\"]}')"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "e05b5748-d373-4ff8-85d9-21da4641e84c",
+            "metadata": {},
+            "source": [
+                "**Default Criteria**\n",
+                "\n",
+                "Most of the time, you'll want to define your own custom criteria (see below), but we also provide some common criteria you can load with a single string.\n",
+                "Here's a list of pre-implemented criteria. Note that in the absence of labels, the LLM merely predicts what it thinks the best answer is and is not grounded in actual law or context."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "47de7359-db3e-4cad-bcfa-4fe834dea893",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "[<Criteria.CONCISENESS: 'conciseness'>,\n",
+                            " <Criteria.RELEVANCE: 'relevance'>,\n",
+                            " <Criteria.CORRECTNESS: 'correctness'>,\n",
+                            " <Criteria.COHERENCE: 'coherence'>,\n",
+                            " <Criteria.HARMFULNESS: 'harmfulness'>,\n",
+                            " <Criteria.MALICIOUSNESS: 'maliciousness'>,\n",
+                            " <Criteria.HELPFULNESS: 'helpfulness'>,\n",
+                            " <Criteria.CONTROVERSIALITY: 'controversiality'>,\n",
+                            " <Criteria.MISOGYNY: 'misogyny'>,\n",
+                            " <Criteria.CRIMINALITY: 'criminality'>,\n",
+                            " <Criteria.INSENSITIVITY: 'insensitivity'>]"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "from langchain.evaluation import Criteria\n",
+                "\n",
+                "# For a list of other default supported criteria, try calling `supported_default_criteria`\n",
+                "list(Criteria)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "077c4715-e857-44a3-9f87-346642586a8d",
+            "metadata": {},
+            "source": [
+                "## Custom Criteria\n",
+                "\n",
+                "To evaluate outputs against your own custom criteria, or to be more explicit the definition of any of the default criteria, pass in a dictionary of `\"criterion_name\": \"criterion_description\"`\n",
+                "\n",
+                "Note: it's recommended that you create a single evaluator per criterion. This way, separate feedback can be provided for each aspect. Additionally, if you provide antagonistic criteria, the evaluator won't be very useful, as it will be configured to predict compliance for ALL of the criteria provided."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 19,
+            "id": "bafa0a11-2617-4663-84bf-24df7d0736be",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "{'reasoning': \"The criterion asks if the output contains numeric or mathematical information. The joke in the submission does contain mathematical information. It refers to the mathematical concept of squaring a number and also mentions 'pi', which is a mathematical constant. Therefore, the submission does meet the criterion.\\n\\nY\", 'value': 'Y', 'score': 1}\n",
+                        "{'reasoning': 'Let\\'s assess the submission based on the given criteria:\\n\\n1. Numeric: The output does not contain any explicit numeric information. The word \"square\" and \"pi\" are mathematical terms but they are not numeric information per se.\\n\\n2. Mathematical: The output does contain mathematical information. The terms \"square\" and \"pi\" are mathematical terms. The joke is a play on the mathematical concept of squaring a number (in this case, pi).\\n\\n3. Grammatical: The output is grammatically correct. The sentence structure, punctuation, and word usage are all correct.\\n\\n4. Logical: The output is logical. It makes sense within the context of the joke. The joke is a play on words between the mathematical concept of squaring a number (pi) and eating a square pie.\\n\\nBased on the above analysis, the submission does not meet all the criteria because it does not contain numeric information.\\nN', 'value': 'N', 'score': 0}\n"
+                    ]
+                }
+            ],
+            "source": [
+                "custom_criterion = {\"numeric\": \"Does the output contain numeric or mathematical information?\"}\n",
+                "\n",
+                "eval_chain = load_evaluator(\n",
+                "    EvaluatorType.CRITERIA,\n",
+                "    criteria=custom_criterion,\n",
+                ")\n",
+                "query = \"Tell me a joke\"\n",
+                "prediction = \"I ate some square pie but I don't know the square of pi.\"\n",
+                "eval_result = eval_chain.evaluate_strings(prediction=prediction, input=query)\n",
+                "print(eval_result)\n",
+                "\n",
+                "# If you wanted to specify multiple criteria. Generally not recommended\n",
+                "custom_criteria = {\n",
+                "    \"numeric\": \"Does the output contain numeric information?\",\n",
+                "    \"mathematical\": \"Does the output contain mathematical information?\",\n",
+                "    \"grammatical\": \"Is the output grammatically correct?\",\n",
+                "    \"logical\": \"Is the output logical?\",\n",
+                "}\n",
+                "\n",
+                "eval_chain = load_evaluator(\n",
+                "    EvaluatorType.CRITERIA,\n",
+                "    criteria=custom_criteria,\n",
+                ")\n",
+                "eval_result = eval_chain.evaluate_strings(prediction=prediction, input=query)\n",
+                "print(\"Multi-criteria evaluation\")\n",
+                "print(eval_result)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "07485cce-8d52-43a0-bdad-76ec7dacfb51",
+            "metadata": {},
+            "source": [
+                "## Using Constitutional Principles\n",
+                "\n",
+                "Custom rubrics are similar to principles from [Constitutional AI](https://arxiv.org/abs/2212.08073). You can directly use your `ConstitutionalPrinciple` objects to\n",
+                "instantiate the chain and take advantage of the many existing principles in LangChain."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 9,
+            "id": "99e3c242-5b12-4bd5-b487-64990a159655",
+            "metadata": {},
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "54 available principles\n"
+                    ]
+                },
+                {
+                    "data": {
+                        "text/plain": [
+                            "[('harmful1',\n",
+                            "  ConstitutionalPrinciple(critique_request='Identify specific ways in which the assistant’s last response is harmful, unethical, racist, sexist, toxic, dangerous, or illegal.', revision_request='Please rewrite the assistant response to remove any and all harmful, unethical, racist, sexist, toxic, dangerous, or illegal content.', name='harmful1')),\n",
+                            " ('harmful2',\n",
+                            "  ConstitutionalPrinciple(critique_request='Explain ways in which the assistant’s preceding response may be harmful to either the human or others. For example, if the assistant says anything that encourages dangerous or illegal activity, make sure to point it out. Other examples of harm include racism, sexism, and other social biases.', revision_request='Please rewrite the assistant response to remove any and all harmful content, including anything dangerous, illegal, racist, sexist or socially biased.', name='harmful2')),\n",
+                            " ('harmful3',\n",
+                            "  ConstitutionalPrinciple(critique_request='Identify all ways in which the assistant’s last response is harmful, unethical, or socially biased. Furthermore, provide specific details on how the assistant can improve its response.', revision_request='Please rewrite the assistant response to remove all harmful, unethical, or socially biased content, and move the conversation in a positive direction.', name='harmful3')),\n",
+                            " ('harmful4',\n",
+                            "  ConstitutionalPrinciple(critique_request='Identify specific ways in which the assistant’s precedings response is harmful. Also point out potential harmfulness in the human’s questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.', revision_request='Rewrite the assistant response to remove all harmful content, and to politely point out any harmful assumptions from the human, such as racism, sexism, toxicity.', name='harmful4')),\n",
+                            " ('insensitive',\n",
+                            "  ConstitutionalPrinciple(critique_request='Discuss the assistant’s last response to the human and whether it is in any way insensitive, sexist, racist, or socially inappropriate.', revision_request='Revise the assistant’s last response to remove content that is in any way insensitive, sexist, racist, or socially inappropriate.', name='insensitive'))]"
+                        ]
+                    },
+                    "execution_count": 9,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "from langchain.chains.constitutional_ai.principles import PRINCIPLES\n",
+                "\n",
+                "print(f\"{len(PRINCIPLES)} available principles\")\n",
+                "list(PRINCIPLES.items())[:5]"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 12,
+            "id": "f4061bc0-a88a-4d1c-88a0-8003bba084c7",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "{'reasoning': 'The criterion asks to identify if the assistant\\'s response is harmful, unethical, racist, sexist, toxic, dangerous, or illegal.\\n\\nLooking at the assistant\\'s response, it is clear that it is not racist or sexist as it does not discriminate or stereotype based on race or gender. \\n\\nThe response is also not illegal as it does not involve any criminal activity or encourage any form of illegal behavior.\\n\\nThe response is not dangerous as it does not pose a physical threat or risk to anyone\\'s safety.\\n\\nHowever, the assistant\\'s response can be considered harmful and toxic as it uses derogatory language (\"lilly-livered nincompoop\") to describe \\'Will\\'. This can be seen as a form of verbal abuse or insult, which can cause emotional harm.\\n\\nThe response can also be seen as unethical, as it is generally considered inappropriate to insult or belittle someone in this manner.\\n\\nN', 'value': 'N', 'score': 0}\n"
+                    ]
+                }
+            ],
+            "source": [
+                "evaluator = load_evaluator(\n",
+                "    EvaluatorType.CRITERIA, criteria=PRINCIPLES[\"harmful1\"]\n",
+                ")\n",
+                "eval_result = evaluator.evaluate_strings(\n",
+                "    prediction=\"I say that man is a lilly-livered nincompoop\",\n",
+                "    input=\"What do you think of Will?\",\n",
+                ")\n",
+                "print(eval_result)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "ae60b5e3-ceac-46b1-aabb-ee36930cb57c",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "## Configuring the LLM\n",
+                "\n",
+                "If you don't specify an eval LLM, the `load_evaluator` method will initialize a `gpt-4` LLM to power the grading chain. Below, use an anthropic model instead."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 13,
+            "id": "1717162d-f76c-4a14-9ade-168d6fa42b7a",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "# %pip install ChatAnthropic\n",
+                "# %env ANTHROPIC_API_KEY=<API_KEY>"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 14,
+            "id": "8727e6f4-aaba-472d-bb7d-09fc1a0f0e2a",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.chat_models import ChatAnthropic\n",
+                "\n",
+                "llm = ChatAnthropic(temperature=0)\n",
+                "evaluator = load_evaluator(\"criteria\", llm=llm, criteria=\"conciseness\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 15,
+            "id": "3f6f0d8b-cf42-4241-85ae-35b3ce8152a0",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "{'reasoning': 'Step 1) Analyze the conciseness criterion: Is the submission concise and to the point?\\nStep 2) The submission provides extraneous information beyond just answering the question directly. It characterizes the question as \"elementary\" and provides reasoning for why the answer is 4. This additional commentary makes the submission not fully concise.\\nStep 3) Therefore, based on the analysis of the conciseness criterion, the submission does not meet the criteria.\\n\\nN', 'value': 'N', 'score': 0}\n"
+                    ]
+                }
+            ],
+            "source": [
+                "eval_result = evaluator.evaluate_strings(\n",
+                "    prediction=\"What's 2+2? That's an elementary question. The answer you're looking for is that two and two is four.\",\n",
+                "    input=\"What's 2+2?\",\n",
+                ")\n",
+                "print(eval_result)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "5e7fc7bb-3075-4b44-9c16-3146a39ae497",
+            "metadata": {},
+            "source": [
+                "# Configuring the Prompt\n",
+                "\n",
+                "If you want to completely customize the prompt, you can initialize the evaluator with a custom prompt template as follows."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 16,
+            "id": "22e57704-682f-44ff-96ba-e915c73269c0",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.prompts import PromptTemplate\n",
+                "\n",
+                "fstring = \"\"\"Respond Y or N based on how well the following response follows the specified rubric. Grade only based on the rubric and expected response:\n",
+                "\n",
+                "Grading Rubric: {criteria}\n",
+                "Expected Response: {reference}\n",
+                "\n",
+                "DATA:\n",
+                "---------\n",
+                "Question: {input}\n",
+                "Response: {output}\n",
+                "---------\n",
+                "Write out your explanation for each criterion, then respond with Y or N on a new line.\"\"\"\n",
+                "\n",
+                "prompt = PromptTemplate.from_template(fstring)\n",
+                "\n",
+                "evaluator = load_evaluator(\n",
+                "    \"labeled_criteria\", criteria=\"correctness\", prompt=prompt\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 17,
+            "id": "5d6b0eca-7aea-4073-a65a-18c3a9cdb5af",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "{'reasoning': 'Correctness: No, the response is not correct. The expected response was \"It\\'s 17 now.\" but the response given was \"What\\'s 2+2? That\\'s an elementary question. The answer you\\'re looking for is that two and two is four.\"', 'value': 'N', 'score': 0}\n"
+                    ]
+                }
+            ],
+            "source": [
+                "eval_result = evaluator.evaluate_strings(\n",
+                "    prediction=\"What's 2+2? That's an elementary question. The answer you're looking for is that two and two is four.\",\n",
+                "    input=\"What's 2+2?\",\n",
+                "    reference=\"It's 17 now.\",\n",
+                ")\n",
+                "print(eval_result)"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "f2662405-353a-4a73-b867-784d12cafcf1",
+            "metadata": {},
+            "source": [
+                "## Conclusion\n",
+                "\n",
+                "In these examples, you used the `CriteriaEvalChain` to evaluate model outputs against custom criteria, including a custom rubric and constitutional principles.\n",
+                "\n",
+                "Remember when selecting criteria to decide whether they ought to require ground truth labels or not. Things like \"correctness\" are best evaluated with ground truth or with extensive context. Also, remember to pick aligned principles for a given chain so that the classification makes sense."
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "a684e2f1",
+            "metadata": {},
+            "source": []
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
     },
-    {
-     "data": {
-      "text/plain": [
-       "[('harmful1',\n",
-       "  ConstitutionalPrinciple(critique_request='Identify specific ways in which the assistant’s last response is harmful, unethical, racist, sexist, toxic, dangerous, or illegal.', revision_request='Please rewrite the assistant response to remove any and all harmful, unethical, racist, sexist, toxic, dangerous, or illegal content.', name='harmful1')),\n",
-       " ('harmful2',\n",
-       "  ConstitutionalPrinciple(critique_request='Explain ways in which the assistant’s preceding response may be harmful to either the human or others. For example, if the assistant says anything that encourages dangerous or illegal activity, make sure to point it out. Other examples of harm include racism, sexism, and other social biases.', revision_request='Please rewrite the assistant response to remove any and all harmful content, including anything dangerous, illegal, racist, sexist or socially biased.', name='harmful2')),\n",
-       " ('harmful3',\n",
-       "  ConstitutionalPrinciple(critique_request='Identify all ways in which the assistant’s last response is harmful, unethical, or socially biased. Furthermore, provide specific details on how the assistant can improve its response.', revision_request='Please rewrite the assistant response to remove all harmful, unethical, or socially biased content, and move the conversation in a positive direction.', name='harmful3')),\n",
-       " ('harmful4',\n",
-       "  ConstitutionalPrinciple(critique_request='Identify specific ways in which the assistant’s precedings response is harmful. Also point out potential harmfulness in the human’s questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.', revision_request='Rewrite the assistant response to remove all harmful content, and to politely point out any harmful assumptions from the human, such as racism, sexism, toxicity.', name='harmful4')),\n",
-       " ('insensitive',\n",
-       "  ConstitutionalPrinciple(critique_request='Discuss the assistant’s last response to the human and whether it is in any way insensitive, sexist, racist, or socially inappropriate.', revision_request='Revise the assistant’s last response to remove content that is in any way insensitive, sexist, racist, or socially inappropriate.', name='insensitive'))]"
-      ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from langchain.chains.constitutional_ai.principles import PRINCIPLES\n",
-    "\n",
-    "print(f\"{len(PRINCIPLES)} available principles\")\n",
-    "list(PRINCIPLES.items())[:5]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "f4061bc0-a88a-4d1c-88a0-8003bba084c7",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'reasoning': 'The criterion asks to identify if the assistant\\'s response is harmful, unethical, racist, sexist, toxic, dangerous, or illegal.\\n\\nLooking at the assistant\\'s response, it is clear that it is not racist or sexist as it does not discriminate or stereotype based on race or gender. \\n\\nThe response is also not illegal as it does not involve any criminal activity or encourage any form of illegal behavior.\\n\\nThe response is not dangerous as it does not pose a physical threat or risk to anyone\\'s safety.\\n\\nHowever, the assistant\\'s response can be considered harmful and toxic as it uses derogatory language (\"lilly-livered nincompoop\") to describe \\'Will\\'. This can be seen as a form of verbal abuse or insult, which can cause emotional harm.\\n\\nThe response can also be seen as unethical, as it is generally considered inappropriate to insult or belittle someone in this manner.\\n\\nN', 'value': 'N', 'score': 0}\n"
-     ]
-    }
-   ],
-   "source": [
-    "evaluator = load_evaluator(\n",
-    "    EvaluatorType.CRITERIA, criteria=PRINCIPLES[\"harmful1\"]\n",
-    ")\n",
-    "eval_result = evaluator.evaluate_strings(\n",
-    "    prediction=\"I say that man is a lilly-livered nincompoop\",\n",
-    "    input=\"What do you think of Will?\",\n",
-    ")\n",
-    "print(eval_result)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "ae60b5e3-ceac-46b1-aabb-ee36930cb57c",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "## Configuring the LLM\n",
-    "\n",
-    "If you don't specify an eval LLM, the `load_evaluator` method will initialize a `gpt-4` LLM to power the grading chain. Below, use an anthropic model instead."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "1717162d-f76c-4a14-9ade-168d6fa42b7a",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# %pip install ChatAnthropic\n",
-    "# %env ANTHROPIC_API_KEY=<API_KEY>"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "8727e6f4-aaba-472d-bb7d-09fc1a0f0e2a",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.chat_models import ChatAnthropic\n",
-    "\n",
-    "llm = ChatAnthropic(temperature=0)\n",
-    "evaluator = load_evaluator(\"criteria\", llm=llm, criteria=\"conciseness\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "3f6f0d8b-cf42-4241-85ae-35b3ce8152a0",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'reasoning': 'Step 1) Analyze the conciseness criterion: Is the submission concise and to the point?\\nStep 2) The submission provides extraneous information beyond just answering the question directly. It characterizes the question as \"elementary\" and provides reasoning for why the answer is 4. This additional commentary makes the submission not fully concise.\\nStep 3) Therefore, based on the analysis of the conciseness criterion, the submission does not meet the criteria.\\n\\nN', 'value': 'N', 'score': 0}\n"
-     ]
-    }
-   ],
-   "source": [
-    "eval_result = evaluator.evaluate_strings(\n",
-    "    prediction=\"What's 2+2? That's an elementary question. The answer you're looking for is that two and two is four.\",\n",
-    "    input=\"What's 2+2?\",\n",
-    ")\n",
-    "print(eval_result)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "5e7fc7bb-3075-4b44-9c16-3146a39ae497",
-   "metadata": {},
-   "source": [
-    "# Configuring the Prompt\n",
-    "\n",
-    "If you want to completely customize the prompt, you can initialize the evaluator with a custom prompt template as follows."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "22e57704-682f-44ff-96ba-e915c73269c0",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.prompts import PromptTemplate\n",
-    "\n",
-    "fstring = \"\"\"Respond Y or N based on how well the following response follows the specified rubric. Grade only based on the rubric and expected response:\n",
-    "\n",
-    "Grading Rubric: {criteria}\n",
-    "Expected Response: {reference}\n",
-    "\n",
-    "DATA:\n",
-    "---------\n",
-    "Question: {input}\n",
-    "Response: {output}\n",
-    "---------\n",
-    "Write out your explanation for each criterion, then respond with Y or N on a new line.\"\"\"\n",
-    "\n",
-    "prompt = PromptTemplate.from_template(fstring)\n",
-    "\n",
-    "evaluator = load_evaluator(\n",
-    "    \"labeled_criteria\", criteria=\"correctness\", prompt=prompt\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 17,
-   "id": "5d6b0eca-7aea-4073-a65a-18c3a9cdb5af",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'reasoning': 'Correctness: No, the response is not correct. The expected response was \"It\\'s 17 now.\" but the response given was \"What\\'s 2+2? That\\'s an elementary question. The answer you\\'re looking for is that two and two is four.\"', 'value': 'N', 'score': 0}\n"
-     ]
-    }
-   ],
-   "source": [
-    "eval_result = evaluator.evaluate_strings(\n",
-    "    prediction=\"What's 2+2? That's an elementary question. The answer you're looking for is that two and two is four.\",\n",
-    "    input=\"What's 2+2?\",\n",
-    "    reference=\"It's 17 now.\",\n",
-    ")\n",
-    "print(eval_result)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "f2662405-353a-4a73-b867-784d12cafcf1",
-   "metadata": {},
-   "source": [
-    "## Conclusion\n",
-    "\n",
-    "In these examples, you used the `CriteriaEvalChain` to evaluate model outputs against custom criteria, including a custom rubric and constitutional principles.\n",
-    "\n",
-    "Remember when selecting criteria to decide whether they ought to require ground truth labels or not. Things like \"correctness\" are best evaluated with ground truth or with extensive context. Also, remember to pick aligned principles for a given chain so that the classification makes sense."
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "a684e2f1",
-   "metadata": {},
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/string/custom.ipynb

@@ -1,208 +1,209 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "4460f924-1738-4dc5-999f-c26383aba0a4",
-   "metadata": {},
-   "source": [
-    "# Custom String Evaluator\n",
-    "\n",
-    "You can make your own custom string evaluators by inheriting from the `StringEvaluator` class and implementing the `_evaluate_strings` (and `_aevaluate_strings` for async support) methods.\n",
-    "\n",
-    "In this example, you will create a perplexity evaluator using the HuggingFace [evaluate](https://huggingface.co/docs/evaluate/index) library.\n",
-    "[Perplexity](https://en.wikipedia.org/wiki/Perplexity) is a measure of how well the generated text would be predicted by the model used to compute the metric."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "90ec5942-4b14-47b1-baff-9dd2a9f17a4e",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# %pip install evaluate > /dev/null"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "54fdba68-0ae7-4102-a45b-dabab86c97ac",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from typing import Any, Optional\n",
-    "\n",
-    "from langchain.evaluation import StringEvaluator\n",
-    "from evaluate import load\n",
-    "\n",
-    "\n",
-    "class PerplexityEvaluator(StringEvaluator):\n",
-    "    \"\"\"Evaluate the perplexity of a predicted string.\"\"\"\n",
-    "\n",
-    "    def __init__(self, model_id: str = \"gpt2\"):\n",
-    "        self.model_id = model_id\n",
-    "        self.metric_fn = load(\n",
-    "            \"perplexity\", module_type=\"metric\", model_id=self.model_id, pad_token=0\n",
-    "        )\n",
-    "\n",
-    "    def _evaluate_strings(\n",
-    "        self,\n",
-    "        *,\n",
-    "        prediction: str,\n",
-    "        reference: Optional[str] = None,\n",
-    "        input: Optional[str] = None,\n",
-    "        **kwargs: Any,\n",
-    "    ) -> dict:\n",
-    "        results = self.metric_fn.compute(\n",
-    "            predictions=[prediction], model_id=self.model_id\n",
-    "        )\n",
-    "        ppl = results[\"perplexities\"][0]\n",
-    "        return {\"score\": ppl}"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "52767568-8075-4f77-93c9-80e1a7e5cba3",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "evaluator = PerplexityEvaluator()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "697ee0c0-d1ae-4a55-a542-a0f8e602c28a",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "Using pad_token, but it is not set yet.\n"
-     ]
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "4460f924-1738-4dc5-999f-c26383aba0a4",
+            "metadata": {},
+            "source": [
+                "# Custom String Evaluator\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/string/custom.ipynb)\n",
+                "\n",
+                "You can make your own custom string evaluators by inheriting from the `StringEvaluator` class and implementing the `_evaluate_strings` (and `_aevaluate_strings` for async support) methods.\n",
+                "\n",
+                "In this example, you will create a perplexity evaluator using the HuggingFace [evaluate](https://huggingface.co/docs/evaluate/index) library.\n",
+                "[Perplexity](https://en.wikipedia.org/wiki/Perplexity) is a measure of how well the generated text would be predicted by the model used to compute the metric."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "90ec5942-4b14-47b1-baff-9dd2a9f17a4e",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "# %pip install evaluate > /dev/null"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "54fdba68-0ae7-4102-a45b-dabab86c97ac",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from typing import Any, Optional\n",
+                "\n",
+                "from langchain.evaluation import StringEvaluator\n",
+                "from evaluate import load\n",
+                "\n",
+                "\n",
+                "class PerplexityEvaluator(StringEvaluator):\n",
+                "    \"\"\"Evaluate the perplexity of a predicted string.\"\"\"\n",
+                "\n",
+                "    def __init__(self, model_id: str = \"gpt2\"):\n",
+                "        self.model_id = model_id\n",
+                "        self.metric_fn = load(\n",
+                "            \"perplexity\", module_type=\"metric\", model_id=self.model_id, pad_token=0\n",
+                "        )\n",
+                "\n",
+                "    def _evaluate_strings(\n",
+                "        self,\n",
+                "        *,\n",
+                "        prediction: str,\n",
+                "        reference: Optional[str] = None,\n",
+                "        input: Optional[str] = None,\n",
+                "        **kwargs: Any,\n",
+                "    ) -> dict:\n",
+                "        results = self.metric_fn.compute(\n",
+                "            predictions=[prediction], model_id=self.model_id\n",
+                "        )\n",
+                "        ppl = results[\"perplexities\"][0]\n",
+                "        return {\"score\": ppl}"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "52767568-8075-4f77-93c9-80e1a7e5cba3",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "evaluator = PerplexityEvaluator()"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "697ee0c0-d1ae-4a55-a542-a0f8e602c28a",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stderr",
+                    "output_type": "stream",
+                    "text": [
+                        "Using pad_token, but it is not set yet.\n"
+                    ]
+                },
+                {
+                    "name": "stdout",
+                    "output_type": "stream",
+                    "text": [
+                        "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
+                        "To disable this warning, you can either:\n",
+                        "\t- Avoid using `tokenizers` before the fork if possible\n",
+                        "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n"
+                    ]
+                },
+                {
+                    "data": {
+                        "application/vnd.jupyter.widget-view+json": {
+                            "model_id": "467109d44654486e8b415288a319fc2c",
+                            "version_major": 2,
+                            "version_minor": 0
+                        },
+                        "text/plain": [
+                            "  0%|          | 0/1 [00:00<?, ?it/s]"
+                        ]
+                    },
+                    "metadata": {},
+                    "output_type": "display_data"
+                },
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 190.3675537109375}"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(prediction=\"The rains in Spain fall mainly on the plain.\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "5089d9d1-eae6-4d47-b4f6-479e5d887d74",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "name": "stderr",
+                    "output_type": "stream",
+                    "text": [
+                        "Using pad_token, but it is not set yet.\n"
+                    ]
+                },
+                {
+                    "data": {
+                        "application/vnd.jupyter.widget-view+json": {
+                            "model_id": "d3266f6f06d746e1bb03ce4aca07d9b9",
+                            "version_major": 2,
+                            "version_minor": 0
+                        },
+                        "text/plain": [
+                            "  0%|          | 0/1 [00:00<?, ?it/s]"
+                        ]
+                    },
+                    "metadata": {},
+                    "output_type": "display_data"
+                },
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1982.0709228515625}"
+                        ]
+                    },
+                    "execution_count": 6,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "# The perplexity is much higher since LangChain was introduced after 'gpt-2' was released and because it is never used in the following context.\n",
+                "evaluator.evaluate_strings(prediction=\"The rains in Spain fall mainly on LangChain.\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": null,
+            "id": "5eaa178f-6ba3-47ae-b3dc-1b196af6d213",
+            "metadata": {},
+            "outputs": [],
+            "source": []
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
     },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
-      "To disable this warning, you can either:\n",
-      "\t- Avoid using `tokenizers` before the fork if possible\n",
-      "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n"
-     ]
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "467109d44654486e8b415288a319fc2c",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "  0%|          | 0/1 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 190.3675537109375}"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_strings(prediction=\"The rains in Spain fall mainly on the plain.\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "5089d9d1-eae6-4d47-b4f6-479e5d887d74",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "Using pad_token, but it is not set yet.\n"
-     ]
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "d3266f6f06d746e1bb03ce4aca07d9b9",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "  0%|          | 0/1 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 1982.0709228515625}"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "# The perplexity is much higher since LangChain was introduced after 'gpt-2' was released and because it is never used in the following context.\n",
-    "evaluator.evaluate_strings(prediction=\"The rains in Spain fall mainly on LangChain.\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "5eaa178f-6ba3-47ae-b3dc-1b196af6d213",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/string/embedding_distance.ipynb

@@ -1,223 +1,224 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "# Embedding Distance\n",
-    "\n",
-    "To measure semantic similarity (or dissimilarity) between a prediction and a reference label string, you could use a vector vector distance metric the two embedded representations using the `embedding_distance` evaluator.<a name=\"cite_ref-1\"></a>[<sup>[1]</sup>](#cite_note-1)\n",
-    "\n",
-    "\n",
-    "**Note:** This returns a **distance** score, meaning that the lower the number, the **more** similar the prediction is to the reference, according to their embedded representation.\n",
-    "\n",
-    "Check out the reference docs for the [EmbeddingDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.embedding_distance.base.EmbeddingDistanceEvalChain.html#langchain.evaluation.embedding_distance.base.EmbeddingDistanceEvalChain) for more info."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"embedding_distance\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.0966466944859925}"
-      ]
-     },
-     "execution_count": 2,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I shan't go\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.03761174337464557}"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I will go\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Select the Distance Metric\n",
-    "\n",
-    "By default, the evalutor uses cosine distance. You can choose a different distance metric if you'd like. "
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[<EmbeddingDistance.COSINE: 'cosine'>,\n",
-       " <EmbeddingDistance.EUCLIDEAN: 'euclidean'>,\n",
-       " <EmbeddingDistance.MANHATTAN: 'manhattan'>,\n",
-       " <EmbeddingDistance.CHEBYSHEV: 'chebyshev'>,\n",
-       " <EmbeddingDistance.HAMMING: 'hamming'>]"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from langchain.evaluation import EmbeddingDistance\n",
-    "\n",
-    "list(EmbeddingDistance)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# You can load by enum or by raw python string\n",
-    "evaluator = load_evaluator(\n",
-    "    \"embedding_distance\", distance_metric=EmbeddingDistance.EUCLIDEAN\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Select Embeddings to Use\n",
-    "\n",
-    "The constructor uses `OpenAI` embeddings by default, but you can configure this however you want. Below, use huggingface local embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.embeddings import HuggingFaceEmbeddings\n",
-    "\n",
-    "embedding_model = HuggingFaceEmbeddings()\n",
-    "hf_evaluator = load_evaluator(\"embedding_distance\", embeddings=embedding_model)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.5486443280477362}"
-      ]
-     },
-     "execution_count": 7,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "hf_evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I shan't go\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.21018880025138598}"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "hf_evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I will go\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "<a name=\"cite_note-1\"></a><i>1. Note: When it comes to semantic similarity, this often gives better results than older string distance metrics (such as those in the [StringDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.string_distance.base.StringDistanceEvalChain.html#langchain.evaluation.string_distance.base.StringDistanceEvalChain)), though it tends to be less reliable than evaluators that use the LLM directly (such as the [QAEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.qa.eval_chain.QAEvalChain.html#langchain.evaluation.qa.eval_chain.QAEvalChain) or [LabeledCriteriaEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.LabeledCriteriaEvalChain.html#langchain.evaluation.criteria.eval_chain.LabeledCriteriaEvalChain)) </i>"
-   ]
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 4
-}
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "# Embedding Distance\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/string/embedding_distance.ipynb)\n",
+                "\n",
+                "To measure semantic similarity (or dissimilarity) between a prediction and a reference label string, you could use a vector vector distance metric the two embedded representations using the `embedding_distance` evaluator.<a name=\"cite_ref-1\"></a>[<sup>[1]</sup>](#cite_note-1)\n",
+                "\n",
+                "\n",
+                "**Note:** This returns a **distance** score, meaning that the lower the number, the **more** similar the prediction is to the reference, according to their embedded representation.\n",
+                "\n",
+                "Check out the reference docs for the [EmbeddingDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.embedding_distance.base.EmbeddingDistanceEvalChain.html#langchain.evaluation.embedding_distance.base.EmbeddingDistanceEvalChain) for more info."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"embedding_distance\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.0966466944859925}"
+                        ]
+                    },
+                    "execution_count": 2,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I shan't go\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.03761174337464557}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I will go\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "## Select the Distance Metric\n",
+                "\n",
+                "By default, the evalutor uses cosine distance. You can choose a different distance metric if you'd like. "
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "[<EmbeddingDistance.COSINE: 'cosine'>,\n",
+                            " <EmbeddingDistance.EUCLIDEAN: 'euclidean'>,\n",
+                            " <EmbeddingDistance.MANHATTAN: 'manhattan'>,\n",
+                            " <EmbeddingDistance.CHEBYSHEV: 'chebyshev'>,\n",
+                            " <EmbeddingDistance.HAMMING: 'hamming'>]"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "from langchain.evaluation import EmbeddingDistance\n",
+                "\n",
+                "list(EmbeddingDistance)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "# You can load by enum or by raw python string\n",
+                "evaluator = load_evaluator(\n",
+                "    \"embedding_distance\", distance_metric=EmbeddingDistance.EUCLIDEAN\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "## Select Embeddings to Use\n",
+                "\n",
+                "The constructor uses `OpenAI` embeddings by default, but you can configure this however you want. Below, use huggingface local embeddings"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.embeddings import HuggingFaceEmbeddings\n",
+                "\n",
+                "embedding_model = HuggingFaceEmbeddings()\n",
+                "hf_evaluator = load_evaluator(\"embedding_distance\", embeddings=embedding_model)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.5486443280477362}"
+                        ]
+                    },
+                    "execution_count": 7,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "hf_evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I shan't go\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 8,
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.21018880025138598}"
+                        ]
+                    },
+                    "execution_count": 8,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "hf_evaluator.evaluate_strings(prediction=\"I shall go\", reference=\"I will go\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "metadata": {},
+            "source": [
+                "<a name=\"cite_note-1\"></a><i>1. Note: When it comes to semantic similarity, this often gives better results than older string distance metrics (such as those in the [StringDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.string_distance.base.StringDistanceEvalChain.html#langchain.evaluation.string_distance.base.StringDistanceEvalChain)), though it tends to be less reliable than evaluators that use the LLM directly (such as the [QAEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.qa.eval_chain.QAEvalChain.html#langchain.evaluation.qa.eval_chain.QAEvalChain) or [LabeledCriteriaEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.criteria.eval_chain.LabeledCriteriaEvalChain.html#langchain.evaluation.criteria.eval_chain.LabeledCriteriaEvalChain)) </i>"
+            ]
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 4
+}

docs/extras/guides/evaluation/string/exact_match.ipynb

@@ -0,0 +1,175 @@
+{
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "2da95378",
+            "metadata": {},
+            "source": [
+                "# Exact Match\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/string/exact_match.ipynb)\n",
+                "\n",
+                "Probably the simplest ways to evaluate an LLM or runnable's string output against a reference label is by a simple string equivalence.\n",
+                "\n",
+                "This can be accessed using the `exact_match` evaluator."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "0de44d01-1fea-4701-b941-c4fb74e521e7",
+            "metadata": {},
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import ExactMatchStringEvaluator\n",
+                "\n",
+                "evaluator = ExactMatchStringEvaluator()"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "fe3baf5f-bfee-4745-bcd6-1a9b422ed46f",
+            "metadata": {},
+            "source": [
+                "Alternatively via the loader:"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "f6790c46",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"exact_match\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "49ad9139",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"1 LLM.\",\n",
+                "    reference=\"2 llm\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "1f5e82a3-247e-45a8-85fc-6af53bf7ff82",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0}"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"LangChain\",\n",
+                "    reference=\"langchain\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "b8ed1f12-09a6-4e90-a69d-c8df525ff293",
+            "metadata": {},
+            "source": [
+                "## Configure the ExactMatchStringEvaluator\n",
+                "\n",
+                "You can relax the \"exactness\" when comparing strings."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "id": "0c079864-0175-4d06-9d3f-a0e51dd3977c",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "evaluator = ExactMatchStringEvaluator(\n",
+                "    ignore_case=True,\n",
+                "    ignore_numbers=True,\n",
+                "    ignore_punctuation=True,\n",
+                ")\n",
+                "\n",
+                "# Alternatively\n",
+                "# evaluator = load_evaluator(\"exact_match\", ignore_case=True, ignore_numbers=True, ignore_punctuation=True)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "a8dfb900-14f3-4a1f-8736-dd1d86a1264c",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1}"
+                        ]
+                    },
+                    "execution_count": 6,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"1 LLM.\",\n",
+                "    reference=\"2 llm\",\n",
+                ")"
+            ]
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/string/regex_match.ipynb

@@ -0,0 +1,243 @@
+{
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "2da95378",
+            "metadata": {},
+            "source": [
+                "# Regex Match\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/string/regex_match.ipynb)\n",
+                "\n",
+                "To evaluate chain or runnable string predictions against a custom regex, you can use the `regex_match` evaluator."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "0de44d01-1fea-4701-b941-c4fb74e521e7",
+            "metadata": {},
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import RegexMatchStringEvaluator\n",
+                "\n",
+                "evaluator = RegexMatchStringEvaluator()"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "fe3baf5f-bfee-4745-bcd6-1a9b422ed46f",
+            "metadata": {},
+            "source": [
+                "Alternatively via the loader:"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "f6790c46",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"regex_match\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "49ad9139",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "# Check for the presence of a YYYY-MM-DD string.\n",
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"The delivery will be made on 2024-01-05\",\n",
+                "    reference=\".*\\\\b\\\\d{4}-\\\\d{2}-\\\\d{2}\\\\b.*\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "1f5e82a3-247e-45a8-85fc-6af53bf7ff82",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0}"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "# Check for the presence of a MM-DD-YYYY string.\n",
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"The delivery will be made on 2024-01-05\",\n",
+                "    reference=\".*\\\\b\\\\d{2}-\\\\d{2}-\\\\d{4}\\\\b.*\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "id": "168fcd92-dffb-4345-b097-02d0fedf52fd",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1}"
+                        ]
+                    },
+                    "execution_count": 5,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "# Check for the presence of a MM-DD-YYYY string.\n",
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"The delivery will be made on 01-05-2024\",\n",
+                "    reference=\".*\\\\b\\\\d{2}-\\\\d{2}-\\\\d{4}\\\\b.*\"\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "1d82dab5-6a49-4fe7-b3fb-8bcfb27d26e0",
+            "metadata": {},
+            "source": [
+                "## Match against multiple patterns\n",
+                "\n",
+                "To match against multiple patterns, use a regex union \"|\"."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "b87b915e-b7c2-476b-a452-99688a22293a",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1}"
+                        ]
+                    },
+                    "execution_count": 6,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "# Check for the presence of a MM-DD-YYYY string or YYYY-MM-DD\n",
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"The delivery will be made on 01-05-2024\",\n",
+                "    reference=\"|\".join([\".*\\\\b\\\\d{4}-\\\\d{2}-\\\\d{2}\\\\b.*\", \".*\\\\b\\\\d{2}-\\\\d{2}-\\\\d{4}\\\\b.*\"])\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "b8ed1f12-09a6-4e90-a69d-c8df525ff293",
+            "metadata": {},
+            "source": [
+                "## Configure the RegexMatchStringEvaluator\n",
+                "\n",
+                "You can specify any regex flags to use when matching."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "id": "0c079864-0175-4d06-9d3f-a0e51dd3977c",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "import re\n",
+                "\n",
+                "evaluator = RegexMatchStringEvaluator(\n",
+                "    flags=re.IGNORECASE\n",
+                ")\n",
+                "\n",
+                "# Alternatively\n",
+                "# evaluator = load_evaluator(\"exact_match\", flags=re.IGNORECASE)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 8,
+            "id": "a8dfb900-14f3-4a1f-8736-dd1d86a1264c",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1}"
+                        ]
+                    },
+                    "execution_count": 8,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"I LOVE testing\",\n",
+                "    reference=\"I love testing\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": null,
+            "id": "82de8d3e-c829-440e-a582-3fb70cecad3b",
+            "metadata": {},
+            "outputs": [],
+            "source": []
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/string/string_distance.ipynb

@@ -1,222 +1,223 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "2da95378",
-   "metadata": {},
-   "source": [
-    "# String Distance\n",
-    "\n",
-    "One of the simplest ways to compare an LLM or chain's string output against a reference label is by using string distance measurements such as Levenshtein or postfix distance.  This can be used alongside approximate/fuzzy matching criteria for very basic unit testing.\n",
-    "\n",
-    "This can be accessed using the `string_distance` evaluator, which uses distance metric's from the [rapidfuzz](https://github.com/maxbachmann/RapidFuzz) library.\n",
-    "\n",
-    "**Note:** The returned scores are _distances_, meaning lower is typically \"better\".\n",
-    "\n",
-    "For more information, check out the reference docs for the [StringDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.string_distance.base.StringDistanceEvalChain.html#langchain.evaluation.string_distance.base.StringDistanceEvalChain) for more info."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "8b47b909-3251-4774-9a7d-e436da4f8979",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# %pip install rapidfuzz"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "f6790c46",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"string_distance\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "49ad9139",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.11555555555555552}"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator.evaluate_strings(\n",
-    "    prediction=\"The job is completely done.\",\n",
-    "    reference=\"The job is done\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "c06a2296",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.0724999999999999}"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "# The results purely character-based, so it's less useful when negation is concerned\n",
-    "evaluator.evaluate_strings(\n",
-    "    prediction=\"The job is done.\",\n",
-    "    reference=\"The job isn't done\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "b8ed1f12-09a6-4e90-a69d-c8df525ff293",
-   "metadata": {},
-   "source": [
-    "## Configure the String Distance Metric\n",
-    "\n",
-    "By default, the `StringDistanceEvalChain` uses  levenshtein distance, but it also supports other string distance algorithms. Configure using the `distance` argument."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "a88bc7d7-62d3-408d-b0e0-43abcecf35c8",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[<StringDistance.DAMERAU_LEVENSHTEIN: 'damerau_levenshtein'>,\n",
-       " <StringDistance.LEVENSHTEIN: 'levenshtein'>,\n",
-       " <StringDistance.JARO: 'jaro'>,\n",
-       " <StringDistance.JARO_WINKLER: 'jaro_winkler'>]"
-      ]
-     },
-     "execution_count": 5,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from langchain.evaluation import StringDistance\n",
-    "\n",
-    "list(StringDistance)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "0c079864-0175-4d06-9d3f-a0e51dd3977c",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "jaro_evaluator = load_evaluator(\n",
-    "    \"string_distance\", distance=StringDistance.JARO\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "a8dfb900-14f3-4a1f-8736-dd1d86a1264c",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.19259259259259254}"
-      ]
-     },
-     "execution_count": 7,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "jaro_evaluator.evaluate_strings(\n",
-    "    prediction=\"The job is completely done.\",\n",
-    "    reference=\"The job is done\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "7020b046-0ef7-40cc-8778-b928e35f3ce1",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 0.12083333333333324}"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "jaro_evaluator.evaluate_strings(\n",
-    "    prediction=\"The job is done.\",\n",
-    "    reference=\"The job isn't done\",\n",
-    ")"
-   ]
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "2da95378",
+            "metadata": {},
+            "source": [
+                "# String Distance\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/string/string_distance.ipynb)\n",
+                "\n",
+                "One of the simplest ways to compare an LLM or chain's string output against a reference label is by using string distance measurements such as Levenshtein or postfix distance.  This can be used alongside approximate/fuzzy matching criteria for very basic unit testing.\n",
+                "\n",
+                "This can be accessed using the `string_distance` evaluator, which uses distance metric's from the [rapidfuzz](https://github.com/maxbachmann/RapidFuzz) library.\n",
+                "\n",
+                "**Note:** The returned scores are _distances_, meaning lower is typically \"better\".\n",
+                "\n",
+                "For more information, check out the reference docs for the [StringDistanceEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.string_distance.base.StringDistanceEvalChain.html#langchain.evaluation.string_distance.base.StringDistanceEvalChain) for more info."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "8b47b909-3251-4774-9a7d-e436da4f8979",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "# %pip install rapidfuzz"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "f6790c46",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"string_distance\")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "49ad9139",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.11555555555555552}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"The job is completely done.\",\n",
+                "    reference=\"The job is done\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "c06a2296",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.0724999999999999}"
+                        ]
+                    },
+                    "execution_count": 4,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "# The results purely character-based, so it's less useful when negation is concerned\n",
+                "evaluator.evaluate_strings(\n",
+                "    prediction=\"The job is done.\",\n",
+                "    reference=\"The job isn't done\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "b8ed1f12-09a6-4e90-a69d-c8df525ff293",
+            "metadata": {},
+            "source": [
+                "## Configure the String Distance Metric\n",
+                "\n",
+                "By default, the `StringDistanceEvalChain` uses  levenshtein distance, but it also supports other string distance algorithms. Configure using the `distance` argument."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "id": "a88bc7d7-62d3-408d-b0e0-43abcecf35c8",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "[<StringDistance.DAMERAU_LEVENSHTEIN: 'damerau_levenshtein'>,\n",
+                            " <StringDistance.LEVENSHTEIN: 'levenshtein'>,\n",
+                            " <StringDistance.JARO: 'jaro'>,\n",
+                            " <StringDistance.JARO_WINKLER: 'jaro_winkler'>]"
+                        ]
+                    },
+                    "execution_count": 5,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "from langchain.evaluation import StringDistance\n",
+                "\n",
+                "list(StringDistance)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "0c079864-0175-4d06-9d3f-a0e51dd3977c",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "jaro_evaluator = load_evaluator(\n",
+                "    \"string_distance\", distance=StringDistance.JARO\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "id": "a8dfb900-14f3-4a1f-8736-dd1d86a1264c",
+            "metadata": {},
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.19259259259259254}"
+                        ]
+                    },
+                    "execution_count": 7,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "jaro_evaluator.evaluate_strings(\n",
+                "    prediction=\"The job is completely done.\",\n",
+                "    reference=\"The job is done\",\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 8,
+            "id": "7020b046-0ef7-40cc-8778-b928e35f3ce1",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 0.12083333333333324}"
+                        ]
+                    },
+                    "execution_count": 8,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "jaro_evaluator.evaluate_strings(\n",
+                "    prediction=\"The job is done.\",\n",
+                "    reference=\"The job isn't done\",\n",
+                ")"
+            ]
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/trajectory/custom.ipynb

@@ -1,141 +1,142 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "db9d627f-b234-4f7f-ab96-639fae474122",
-   "metadata": {},
-   "source": [
-    "# Custom Trajectory Evaluator\n",
-    "\n",
-    "You can make your own custom trajectory evaluators by inheriting from the [AgentTrajectoryEvaluator](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.schema.AgentTrajectoryEvaluator.html#langchain.evaluation.schema.AgentTrajectoryEvaluator) class and overwriting the `_evaluate_agent_trajectory` (and `_aevaluate_agent_action`) method.\n",
-    "\n",
-    "\n",
-    "In this example, you will make a simple trajectory evaluator that uses an LLM to determine if any actions were unnecessary."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "ca84ab0c-e7e2-4c03-bd74-9cc4e6338eec",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from typing import Any, Optional, Sequence, Tuple\n",
-    "from langchain.chat_models import ChatOpenAI\n",
-    "from langchain.chains import LLMChain\n",
-    "from langchain.schema import AgentAction\n",
-    "from langchain.evaluation import AgentTrajectoryEvaluator\n",
-    "\n",
-    "\n",
-    "class StepNecessityEvaluator(AgentTrajectoryEvaluator):\n",
-    "    \"\"\"Evaluate the perplexity of a predicted string.\"\"\"\n",
-    "\n",
-    "    def __init__(self) -> None:\n",
-    "        llm = ChatOpenAI(model=\"gpt-4\", temperature=0.0)\n",
-    "        template = \"\"\"Are any of the following steps unnecessary in answering {input}? Provide the verdict on a new line as a single \"Y\" for yes or \"N\" for no.\n",
-    "\n",
-    "        DATA\n",
-    "        ------\n",
-    "        Steps: {trajectory}\n",
-    "        ------\n",
-    "\n",
-    "        Verdict:\"\"\"\n",
-    "        self.chain = LLMChain.from_string(llm, template)\n",
-    "\n",
-    "    def _evaluate_agent_trajectory(\n",
-    "        self,\n",
-    "        *,\n",
-    "        prediction: str,\n",
-    "        input: str,\n",
-    "        agent_trajectory: Sequence[Tuple[AgentAction, str]],\n",
-    "        reference: Optional[str] = None,\n",
-    "        **kwargs: Any,\n",
-    "    ) -> dict:\n",
-    "        vals = [\n",
-    "            f\"{i}: Action=[{action.tool}] returned observation = [{observation}]\"\n",
-    "            for i, (action, observation) in enumerate(agent_trajectory)\n",
-    "        ]\n",
-    "        trajectory = \"\\n\".join(vals)\n",
-    "        response = self.chain.run(dict(trajectory=trajectory, input=input), **kwargs)\n",
-    "        decision = response.split(\"\\n\")[-1].strip()\n",
-    "        score = 1 if decision == \"Y\" else 0\n",
-    "        return {\"score\": score, \"value\": decision, \"reasoning\": response}"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "297dea4b-fb28-4292-b6e0-1c769cfb9cbd",
-   "metadata": {},
-   "source": [
-    "The example above will return a score of 1 if the language model predicts that any of the actions were unnecessary, and it returns a score of 0 if all of them were predicted to be necessary. It returns the string 'decision' as the 'value', and includes the rest of the generated text as 'reasoning' to let you audit the decision.\n",
-    "\n",
-    "You can call this evaluator to grade the intermediate steps of your agent's trajectory."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "a3fbcc1d-249f-4e00-8841-b6872c73c486",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 1, 'value': 'Y', 'reasoning': 'Y'}"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluator = StepNecessityEvaluator()\n",
-    "\n",
-    "evaluator.evaluate_agent_trajectory(\n",
-    "    prediction=\"The answer is pi\",\n",
-    "    input=\"What is today?\",\n",
-    "    agent_trajectory=[\n",
-    "        (\n",
-    "            AgentAction(tool=\"ask\", tool_input=\"What is today?\", log=\"\"),\n",
-    "            \"tomorrow's yesterday\",\n",
-    "        ),\n",
-    "        (\n",
-    "            AgentAction(tool=\"check_tv\", tool_input=\"Watch tv for half hour\", log=\"\"),\n",
-    "            \"bzzz\",\n",
-    "        ),\n",
-    "    ],\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "77353528-723e-4075-939e-aebdb17c1e4f",
-   "metadata": {},
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "db9d627f-b234-4f7f-ab96-639fae474122",
+            "metadata": {},
+            "source": [
+                "# Custom Trajectory Evaluator\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/trajectory/custom.ipynb)\n",
+                "\n",
+                "You can make your own custom trajectory evaluators by inheriting from the [AgentTrajectoryEvaluator](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.schema.AgentTrajectoryEvaluator.html#langchain.evaluation.schema.AgentTrajectoryEvaluator) class and overwriting the `_evaluate_agent_trajectory` (and `_aevaluate_agent_action`) method.\n",
+                "\n",
+                "\n",
+                "In this example, you will make a simple trajectory evaluator that uses an LLM to determine if any actions were unnecessary."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "ca84ab0c-e7e2-4c03-bd74-9cc4e6338eec",
+            "metadata": {},
+            "outputs": [],
+            "source": [
+                "from typing import Any, Optional, Sequence, Tuple\n",
+                "from langchain.chat_models import ChatOpenAI\n",
+                "from langchain.chains import LLMChain\n",
+                "from langchain.schema import AgentAction\n",
+                "from langchain.evaluation import AgentTrajectoryEvaluator\n",
+                "\n",
+                "\n",
+                "class StepNecessityEvaluator(AgentTrajectoryEvaluator):\n",
+                "    \"\"\"Evaluate the perplexity of a predicted string.\"\"\"\n",
+                "\n",
+                "    def __init__(self) -> None:\n",
+                "        llm = ChatOpenAI(model=\"gpt-4\", temperature=0.0)\n",
+                "        template = \"\"\"Are any of the following steps unnecessary in answering {input}? Provide the verdict on a new line as a single \"Y\" for yes or \"N\" for no.\n",
+                "\n",
+                "        DATA\n",
+                "        ------\n",
+                "        Steps: {trajectory}\n",
+                "        ------\n",
+                "\n",
+                "        Verdict:\"\"\"\n",
+                "        self.chain = LLMChain.from_string(llm, template)\n",
+                "\n",
+                "    def _evaluate_agent_trajectory(\n",
+                "        self,\n",
+                "        *,\n",
+                "        prediction: str,\n",
+                "        input: str,\n",
+                "        agent_trajectory: Sequence[Tuple[AgentAction, str]],\n",
+                "        reference: Optional[str] = None,\n",
+                "        **kwargs: Any,\n",
+                "    ) -> dict:\n",
+                "        vals = [\n",
+                "            f\"{i}: Action=[{action.tool}] returned observation = [{observation}]\"\n",
+                "            for i, (action, observation) in enumerate(agent_trajectory)\n",
+                "        ]\n",
+                "        trajectory = \"\\n\".join(vals)\n",
+                "        response = self.chain.run(dict(trajectory=trajectory, input=input), **kwargs)\n",
+                "        decision = response.split(\"\\n\")[-1].strip()\n",
+                "        score = 1 if decision == \"Y\" else 0\n",
+                "        return {\"score\": score, \"value\": decision, \"reasoning\": response}"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "297dea4b-fb28-4292-b6e0-1c769cfb9cbd",
+            "metadata": {},
+            "source": [
+                "The example above will return a score of 1 if the language model predicts that any of the actions were unnecessary, and it returns a score of 0 if all of them were predicted to be necessary. It returns the string 'decision' as the 'value', and includes the rest of the generated text as 'reasoning' to let you audit the decision.\n",
+                "\n",
+                "You can call this evaluator to grade the intermediate steps of your agent's trajectory."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "a3fbcc1d-249f-4e00-8841-b6872c73c486",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1, 'value': 'Y', 'reasoning': 'Y'}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluator = StepNecessityEvaluator()\n",
+                "\n",
+                "evaluator.evaluate_agent_trajectory(\n",
+                "    prediction=\"The answer is pi\",\n",
+                "    input=\"What is today?\",\n",
+                "    agent_trajectory=[\n",
+                "        (\n",
+                "            AgentAction(tool=\"ask\", tool_input=\"What is today?\", log=\"\"),\n",
+                "            \"tomorrow's yesterday\",\n",
+                "        ),\n",
+                "        (\n",
+                "            AgentAction(tool=\"check_tv\", tool_input=\"Watch tv for half hour\", log=\"\"),\n",
+                "            \"bzzz\",\n",
+                "        ),\n",
+                "    ],\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "77353528-723e-4075-939e-aebdb17c1e4f",
+            "metadata": {},
+            "source": []
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/guides/evaluation/trajectory/trajectory_eval.ipynb

@@ -1,304 +1,305 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "6e5ea1a1-7e74-459b-bf14-688f87d09124",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "# Agent Trajectory\n",
-    "\n",
-    "Agents can be difficult to holistically evaluate due to the breadth of actions and generation they can make. We recommend using multiple evaluation techniques appropriate to your use case. One way to evaluate an agent is to look at the whole trajectory of actions taken along with their responses.\n",
-    "\n",
-    "Evaluators that do this can implement the `AgentTrajectoryEvaluator` interface. This walkthrough will show how to use the `trajectory` evaluator to grade  an OpenAI functions agent.\n",
-    "\n",
-    "For more information, check out the reference docs for the [TrajectoryEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.html#langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain) for more info."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "149402da-5212-43e2-b7c0-a701727f5293",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"trajectory\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "b1c64c1a",
-   "metadata": {},
-   "source": [
-    "## Methods\n",
-    "\n",
-    "\n",
-    "The Agent Trajectory Evaluators are used with the [evaluate_agent_trajectory](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.html#langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.evaluate_agent_trajectory) (and async [aevaluate_agent_trajectory](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.html#langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.aevaluate_agent_trajectory)) methods, which accept:\n",
-    "\n",
-    "- input (str) – The input to the agent.\n",
-    "- prediction (str) – The final predicted response.\n",
-    "- agent_trajectory (List[Tuple[AgentAction, str]]) – The intermediate steps forming the agent trajectory\n",
-    "\n",
-    "They return a dictionary with the following values:\n",
-    "- score: Float from 0 to 1, where 1 would mean \"most effective\" and 0 would mean \"least effective\"\n",
-    "- reasoning: String \"chain of thought reasoning\" from the LLM generated prior to creating the score"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "e733562c-4c17-4942-9647-acfc5ebfaca2",
-   "metadata": {},
-   "source": [
-    "## Capturing Trajectory\n",
-    "\n",
-    "The easiest way to return an agent's trajectory (without using tracing callbacks like those in LangSmith) for evaluation is to initialize the agent with `return_intermediate_steps=True`.\n",
-    "\n",
-    "Below, create an example agent we will call to evaluate."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "451cb0cb-6f42-4abd-aa6d-fb871fce034d",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "import os\n",
-    "import subprocess\n",
-    "\n",
-    "from langchain.chat_models import ChatOpenAI\n",
-    "from langchain.tools import tool\n",
-    "from langchain.agents import AgentType, initialize_agent\n",
-    "\n",
-    "from pydantic import HttpUrl\n",
-    "from urllib.parse import urlparse\n",
-    "\n",
-    "\n",
-    "@tool\n",
-    "def ping(url: HttpUrl, return_error: bool) -> str:\n",
-    "    \"\"\"Ping the fully specified url. Must include https:// in the url.\"\"\"\n",
-    "    hostname = urlparse(str(url)).netloc\n",
-    "    completed_process = subprocess.run(\n",
-    "        [\"ping\", \"-c\", \"1\", hostname], capture_output=True, text=True\n",
-    "    )\n",
-    "    output = completed_process.stdout\n",
-    "    if return_error and completed_process.returncode != 0:\n",
-    "        return completed_process.stderr\n",
-    "    return output\n",
-    "\n",
-    "\n",
-    "@tool\n",
-    "def trace_route(url: HttpUrl, return_error: bool) -> str:\n",
-    "    \"\"\"Trace the route to the specified url. Must include https:// in the url.\"\"\"\n",
-    "    hostname = urlparse(str(url)).netloc\n",
-    "    completed_process = subprocess.run(\n",
-    "        [\"traceroute\", hostname], capture_output=True, text=True\n",
-    "    )\n",
-    "    output = completed_process.stdout\n",
-    "    if return_error and completed_process.returncode != 0:\n",
-    "        return completed_process.stderr\n",
-    "    return output\n",
-    "\n",
-    "\n",
-    "llm = ChatOpenAI(model=\"gpt-3.5-turbo-0613\", temperature=0)\n",
-    "agent = initialize_agent(\n",
-    "    llm=llm,\n",
-    "    tools=[ping, trace_route],\n",
-    "    agent=AgentType.OPENAI_MULTI_FUNCTIONS,\n",
-    "    return_intermediate_steps=True,  # IMPORTANT!\n",
-    ")\n",
-    "\n",
-    "result = agent(\"What's the latency like for https://langchain.com?\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "2df34eed-45a5-4f91-88d3-9aa55f28391a",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "## Evaluate Trajectory\n",
-    "\n",
-    "Pass the input, trajectory, and pass to the [evaluate_agent_trajectory](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.schema.AgentTrajectoryEvaluator.html#langchain.evaluation.schema.AgentTrajectoryEvaluator.evaluate_agent_trajectory) method."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "8d2c8703-98ed-4068-8a8b-393f0f1f64ea",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 1.0,\n",
-       " 'reasoning': \"i. The final answer is helpful. It directly answers the user's question about the latency for the website https://langchain.com.\\n\\nii. The AI language model uses a logical sequence of tools to answer the question. It uses the 'ping' tool to measure the latency of the website, which is the correct tool for this task.\\n\\niii. The AI language model uses the tool in a helpful way. It inputs the URL into the 'ping' tool and correctly interprets the output to provide the latency in milliseconds.\\n\\niv. The AI language model does not use too many steps to answer the question. It only uses one step, which is appropriate for this type of question.\\n\\nv. The appropriate tool is used to answer the question. The 'ping' tool is the correct tool to measure website latency.\\n\\nGiven these considerations, the AI language model's performance is excellent. It uses the correct tool, interprets the output correctly, and provides a helpful and direct answer to the user's question.\"}"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluation_result = evaluator.evaluate_agent_trajectory(\n",
-    "    prediction=result[\"output\"],\n",
-    "    input=result[\"input\"],\n",
-    "    agent_trajectory=result[\"intermediate_steps\"],\n",
-    ")\n",
-    "evaluation_result"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "fc5467c1-ea92-405f-949a-3011388fa9ee",
-   "metadata": {},
-   "source": [
-    "## Configuring the Evaluation LLM\n",
-    "\n",
-    "If you don't select an LLM to use for evaluation, the [load_evaluator](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.loading.load_evaluator.html#langchain.evaluation.loading.load_evaluator) function will use `gpt-4` to power the evaluation chain. You can select any chat model for the agent trajectory evaluator as below."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "1f6318f3-642a-4766-bc7a-f91239795ee7",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# %pip install anthropic\n",
-    "# ANTHROPIC_API_KEY=<YOUR ANTHROPIC API KEY>"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "b2852289-5df9-402e-95b5-7efebf0fc943",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.chat_models import ChatAnthropic\n",
-    "\n",
-    "eval_llm = ChatAnthropic(temperature=0)\n",
-    "evaluator = load_evaluator(\"trajectory\", llm=eval_llm)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "ff72d21a-93b9-4c2f-8613-733d9c9330d7",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 1.0,\n",
-       " 'reasoning': \"Here is my detailed evaluation of the AI's response:\\n\\ni. The final answer is helpful, as it directly provides the latency measurement for the requested website.\\n\\nii. The sequence of using the ping tool to measure latency is logical for this question.\\n\\niii. The ping tool is used in a helpful way, with the website URL provided as input and the output latency measurement extracted.\\n\\niv. Only one step is used, which is appropriate for simply measuring latency. More steps are not needed.\\n\\nv. The ping tool is an appropriate choice to measure latency. \\n\\nIn summary, the AI uses an optimal single step approach with the right tool and extracts the needed output. The final answer directly answers the question in a helpful way.\\n\\nOverall\"}"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluation_result = evaluator.evaluate_agent_trajectory(\n",
-    "    prediction=result[\"output\"],\n",
-    "    input=result[\"input\"],\n",
-    "    agent_trajectory=result[\"intermediate_steps\"],\n",
-    ")\n",
-    "evaluation_result"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "95ce4240-f5a0-4810-8d09-b2f4c9e18b7f",
-   "metadata": {},
-   "source": [
-    "## Providing List of Valid Tools\n",
-    "\n",
-    "By default, the evaluator doesn't take into account the tools the agent is permitted to call. You can provide these to the evaluator via the `agent_tools` argument.\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "24c10566-2ef5-45c5-9213-a8fb28e2ca1f",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.evaluation import load_evaluator\n",
-    "\n",
-    "evaluator = load_evaluator(\"trajectory\", agent_tools=[ping, trace_route])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "7b995786-5b78-4d9e-8e8a-1f2a203113e2",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'score': 1.0,\n",
-       " 'reasoning': \"i. The final answer is helpful. It directly answers the user's question about the latency for the specified website.\\n\\nii. The AI language model uses a logical sequence of tools to answer the question. In this case, only one tool was needed to answer the question, and the model chose the correct one.\\n\\niii. The AI language model uses the tool in a helpful way. The 'ping' tool was used to determine the latency of the website, which was the information the user was seeking.\\n\\niv. The AI language model does not use too many steps to answer the question. Only one step was needed and used.\\n\\nv. The appropriate tool was used to answer the question. The 'ping' tool is designed to measure latency, which was the information the user was seeking.\\n\\nGiven these considerations, the AI language model's performance in answering this question is excellent.\"}"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "evaluation_result = evaluator.evaluate_agent_trajectory(\n",
-    "    prediction=result[\"output\"],\n",
-    "    input=result[\"input\"],\n",
-    "    agent_trajectory=result[\"intermediate_steps\"],\n",
-    ")\n",
-    "evaluation_result"
-   ]
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.2"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+    "cells": [
+        {
+            "cell_type": "markdown",
+            "id": "6e5ea1a1-7e74-459b-bf14-688f87d09124",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "# Agent Trajectory\n",
+                "[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/langchain-ai/langchain/blob/master/docs/extras/guides/evaluation/trajectory/trajectory_eval.ipynb)\n",
+                "\n",
+                "Agents can be difficult to holistically evaluate due to the breadth of actions and generation they can make. We recommend using multiple evaluation techniques appropriate to your use case. One way to evaluate an agent is to look at the whole trajectory of actions taken along with their responses.\n",
+                "\n",
+                "Evaluators that do this can implement the `AgentTrajectoryEvaluator` interface. This walkthrough will show how to use the `trajectory` evaluator to grade  an OpenAI functions agent.\n",
+                "\n",
+                "For more information, check out the reference docs for the [TrajectoryEvalChain](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.html#langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain) for more info."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 1,
+            "id": "149402da-5212-43e2-b7c0-a701727f5293",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"trajectory\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "b1c64c1a",
+            "metadata": {},
+            "source": [
+                "## Methods\n",
+                "\n",
+                "\n",
+                "The Agent Trajectory Evaluators are used with the [evaluate_agent_trajectory](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.html#langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.evaluate_agent_trajectory) (and async [aevaluate_agent_trajectory](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.html#langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain.aevaluate_agent_trajectory)) methods, which accept:\n",
+                "\n",
+                "- input (str) – The input to the agent.\n",
+                "- prediction (str) – The final predicted response.\n",
+                "- agent_trajectory (List[Tuple[AgentAction, str]]) – The intermediate steps forming the agent trajectory\n",
+                "\n",
+                "They return a dictionary with the following values:\n",
+                "- score: Float from 0 to 1, where 1 would mean \"most effective\" and 0 would mean \"least effective\"\n",
+                "- reasoning: String \"chain of thought reasoning\" from the LLM generated prior to creating the score"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "e733562c-4c17-4942-9647-acfc5ebfaca2",
+            "metadata": {},
+            "source": [
+                "## Capturing Trajectory\n",
+                "\n",
+                "The easiest way to return an agent's trajectory (without using tracing callbacks like those in LangSmith) for evaluation is to initialize the agent with `return_intermediate_steps=True`.\n",
+                "\n",
+                "Below, create an example agent we will call to evaluate."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 2,
+            "id": "451cb0cb-6f42-4abd-aa6d-fb871fce034d",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "import os\n",
+                "import subprocess\n",
+                "\n",
+                "from langchain.chat_models import ChatOpenAI\n",
+                "from langchain.tools import tool\n",
+                "from langchain.agents import AgentType, initialize_agent\n",
+                "\n",
+                "from pydantic import HttpUrl\n",
+                "from urllib.parse import urlparse\n",
+                "\n",
+                "\n",
+                "@tool\n",
+                "def ping(url: HttpUrl, return_error: bool) -> str:\n",
+                "    \"\"\"Ping the fully specified url. Must include https:// in the url.\"\"\"\n",
+                "    hostname = urlparse(str(url)).netloc\n",
+                "    completed_process = subprocess.run(\n",
+                "        [\"ping\", \"-c\", \"1\", hostname], capture_output=True, text=True\n",
+                "    )\n",
+                "    output = completed_process.stdout\n",
+                "    if return_error and completed_process.returncode != 0:\n",
+                "        return completed_process.stderr\n",
+                "    return output\n",
+                "\n",
+                "\n",
+                "@tool\n",
+                "def trace_route(url: HttpUrl, return_error: bool) -> str:\n",
+                "    \"\"\"Trace the route to the specified url. Must include https:// in the url.\"\"\"\n",
+                "    hostname = urlparse(str(url)).netloc\n",
+                "    completed_process = subprocess.run(\n",
+                "        [\"traceroute\", hostname], capture_output=True, text=True\n",
+                "    )\n",
+                "    output = completed_process.stdout\n",
+                "    if return_error and completed_process.returncode != 0:\n",
+                "        return completed_process.stderr\n",
+                "    return output\n",
+                "\n",
+                "\n",
+                "llm = ChatOpenAI(model=\"gpt-3.5-turbo-0613\", temperature=0)\n",
+                "agent = initialize_agent(\n",
+                "    llm=llm,\n",
+                "    tools=[ping, trace_route],\n",
+                "    agent=AgentType.OPENAI_MULTI_FUNCTIONS,\n",
+                "    return_intermediate_steps=True,  # IMPORTANT!\n",
+                ")\n",
+                "\n",
+                "result = agent(\"What's the latency like for https://langchain.com?\")"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "2df34eed-45a5-4f91-88d3-9aa55f28391a",
+            "metadata": {
+                "tags": []
+            },
+            "source": [
+                "## Evaluate Trajectory\n",
+                "\n",
+                "Pass the input, trajectory, and pass to the [evaluate_agent_trajectory](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.schema.AgentTrajectoryEvaluator.html#langchain.evaluation.schema.AgentTrajectoryEvaluator.evaluate_agent_trajectory) method."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 3,
+            "id": "8d2c8703-98ed-4068-8a8b-393f0f1f64ea",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1.0,\n",
+                            " 'reasoning': \"i. The final answer is helpful. It directly answers the user's question about the latency for the website https://langchain.com.\\n\\nii. The AI language model uses a logical sequence of tools to answer the question. It uses the 'ping' tool to measure the latency of the website, which is the correct tool for this task.\\n\\niii. The AI language model uses the tool in a helpful way. It inputs the URL into the 'ping' tool and correctly interprets the output to provide the latency in milliseconds.\\n\\niv. The AI language model does not use too many steps to answer the question. It only uses one step, which is appropriate for this type of question.\\n\\nv. The appropriate tool is used to answer the question. The 'ping' tool is the correct tool to measure website latency.\\n\\nGiven these considerations, the AI language model's performance is excellent. It uses the correct tool, interprets the output correctly, and provides a helpful and direct answer to the user's question.\"}"
+                        ]
+                    },
+                    "execution_count": 3,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluation_result = evaluator.evaluate_agent_trajectory(\n",
+                "    prediction=result[\"output\"],\n",
+                "    input=result[\"input\"],\n",
+                "    agent_trajectory=result[\"intermediate_steps\"],\n",
+                ")\n",
+                "evaluation_result"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "fc5467c1-ea92-405f-949a-3011388fa9ee",
+            "metadata": {},
+            "source": [
+                "## Configuring the Evaluation LLM\n",
+                "\n",
+                "If you don't select an LLM to use for evaluation, the [load_evaluator](https://api.python.langchain.com/en/latest/evaluation/langchain.evaluation.loading.load_evaluator.html#langchain.evaluation.loading.load_evaluator) function will use `gpt-4` to power the evaluation chain. You can select any chat model for the agent trajectory evaluator as below."
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 4,
+            "id": "1f6318f3-642a-4766-bc7a-f91239795ee7",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "# %pip install anthropic\n",
+                "# ANTHROPIC_API_KEY=<YOUR ANTHROPIC API KEY>"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 5,
+            "id": "b2852289-5df9-402e-95b5-7efebf0fc943",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.chat_models import ChatAnthropic\n",
+                "\n",
+                "eval_llm = ChatAnthropic(temperature=0)\n",
+                "evaluator = load_evaluator(\"trajectory\", llm=eval_llm)"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 6,
+            "id": "ff72d21a-93b9-4c2f-8613-733d9c9330d7",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1.0,\n",
+                            " 'reasoning': \"Here is my detailed evaluation of the AI's response:\\n\\ni. The final answer is helpful, as it directly provides the latency measurement for the requested website.\\n\\nii. The sequence of using the ping tool to measure latency is logical for this question.\\n\\niii. The ping tool is used in a helpful way, with the website URL provided as input and the output latency measurement extracted.\\n\\niv. Only one step is used, which is appropriate for simply measuring latency. More steps are not needed.\\n\\nv. The ping tool is an appropriate choice to measure latency. \\n\\nIn summary, the AI uses an optimal single step approach with the right tool and extracts the needed output. The final answer directly answers the question in a helpful way.\\n\\nOverall\"}"
+                        ]
+                    },
+                    "execution_count": 6,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluation_result = evaluator.evaluate_agent_trajectory(\n",
+                "    prediction=result[\"output\"],\n",
+                "    input=result[\"input\"],\n",
+                "    agent_trajectory=result[\"intermediate_steps\"],\n",
+                ")\n",
+                "evaluation_result"
+            ]
+        },
+        {
+            "cell_type": "markdown",
+            "id": "95ce4240-f5a0-4810-8d09-b2f4c9e18b7f",
+            "metadata": {},
+            "source": [
+                "## Providing List of Valid Tools\n",
+                "\n",
+                "By default, the evaluator doesn't take into account the tools the agent is permitted to call. You can provide these to the evaluator via the `agent_tools` argument.\n"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 7,
+            "id": "24c10566-2ef5-45c5-9213-a8fb28e2ca1f",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [],
+            "source": [
+                "from langchain.evaluation import load_evaluator\n",
+                "\n",
+                "evaluator = load_evaluator(\"trajectory\", agent_tools=[ping, trace_route])"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": 8,
+            "id": "7b995786-5b78-4d9e-8e8a-1f2a203113e2",
+            "metadata": {
+                "tags": []
+            },
+            "outputs": [
+                {
+                    "data": {
+                        "text/plain": [
+                            "{'score': 1.0,\n",
+                            " 'reasoning': \"i. The final answer is helpful. It directly answers the user's question about the latency for the specified website.\\n\\nii. The AI language model uses a logical sequence of tools to answer the question. In this case, only one tool was needed to answer the question, and the model chose the correct one.\\n\\niii. The AI language model uses the tool in a helpful way. The 'ping' tool was used to determine the latency of the website, which was the information the user was seeking.\\n\\niv. The AI language model does not use too many steps to answer the question. Only one step was needed and used.\\n\\nv. The appropriate tool was used to answer the question. The 'ping' tool is designed to measure latency, which was the information the user was seeking.\\n\\nGiven these considerations, the AI language model's performance in answering this question is excellent.\"}"
+                        ]
+                    },
+                    "execution_count": 8,
+                    "metadata": {},
+                    "output_type": "execute_result"
+                }
+            ],
+            "source": [
+                "evaluation_result = evaluator.evaluate_agent_trajectory(\n",
+                "    prediction=result[\"output\"],\n",
+                "    input=result[\"input\"],\n",
+                "    agent_trajectory=result[\"intermediate_steps\"],\n",
+                ")\n",
+                "evaluation_result"
+            ]
+        }
+    ],
+    "metadata": {
+        "kernelspec": {
+            "display_name": "Python 3 (ipykernel)",
+            "language": "python",
+            "name": "python3"
+        },
+        "language_info": {
+            "codemirror_mode": {
+                "name": "ipython",
+                "version": 3
+            },
+            "file_extension": ".py",
+            "mimetype": "text/x-python",
+            "name": "python",
+            "nbconvert_exporter": "python",
+            "pygments_lexer": "ipython3",
+            "version": "3.11.2"
+        }
+    },
+    "nbformat": 4,
+    "nbformat_minor": 5
+}

docs/extras/integrations/callbacks/llmonitor.md

@@ -37,10 +37,10 @@ llm = OpenAI(
     callbacks=[handler],
 )
 
-chat = ChatOpenAI(
-    callbacks=[handler],
-    metadata={"userId": "123"},  # you can assign user ids to models in the metadata
-)
+chat = ChatOpenAI(callbacks=[handler])
+
+llm("Tell me a joke")
+
 ```
 
 ## Usage with chains and agents
@@ -100,6 +100,18 @@ agent.run(
 )
 ```
 
+## User Tracking
+User tracking allows you to identify your users, track their cost, conversations and more.
+
+```python
+from langchain.callbacks.llmonitor_callback import LLMonitorCallbackHandler, identify
+
+with identify("user-123"):
+    llm("Tell me a joke")
+
+with identify("user-456", user_props={"email": "user456@test.com"}):
+    agen.run("Who is Leo DiCaprio's girlfriend?")
+```
 ## Support
 
 For any question or issue with integration you can reach out to the LLMonitor team on [Discord](http://discord.com/invite/8PafSG58kK) or via [email](mailto:vince@llmonitor.com).

docs/extras/integrations/callbacks/trubrics.ipynb

@@ -0,0 +1,370 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "40dab0fa-e56c-4958-959e-bd6d6f829724",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "# Trubrics\n",
+    "\n",
+    "![Trubrics](https://miro.medium.com/v2/resize:fit:720/format:webp/1*AhYbKO-v8F4u3hx2aDIqKg.png)\n",
+    "\n",
+    "[Trubrics](https://trubrics.com) is an LLM user analytics platform that lets you collect, analyse and manage user\n",
+    "prompts & feedback on AI models. In this guide we will go over how to setup the `TrubricsCallbackHandler`. \n",
+    "\n",
+    "Check out [our repo](https://github.com/trubrics/trubrics-sdk) for more information on Trubrics."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c0d060d5-133b-496e-b76e-43284d5545b8",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "## Installation and Setup"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ce799e10-5433-4b29-8fa1-c1352f761918",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install trubrics"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "44666917-85f2-4695-897d-54504e343604",
+   "metadata": {},
+   "source": [
+    "### Getting Trubrics Credentials\n",
+    "\n",
+    "If you do not have a Trubrics account, create one on [here](https://trubrics.streamlit.app/). In this tutorial, we will use the `default` project that is built upon account creation.\n",
+    "\n",
+    "Now set your credentials as environment variables:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cd696d03-bea8-42bd-914b-2290fcafb5c9",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "\n",
+    "os.environ[\"TRUBRICS_EMAIL\"] = \"***@***\"\n",
+    "os.environ[\"TRUBRICS_PASSWORD\"] = \"***\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "cd7177b0-a9e8-45ae-adb0-ea779376511b",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "### Usage"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "6ec1bcd4-3824-43de-84a4-3102a2f6d26d",
+   "metadata": {},
+   "source": [
+    "The `TrubricsCallbackHandler` can receive various optional arguments. See [here](https://trubrics.github.io/trubrics-sdk/platform/user_prompts/#saving-prompts-to-trubrics) for kwargs that can be passed to Trubrics prompts.\n",
+    "\n",
+    "```python\n",
+    "class TrubricsCallbackHandler(BaseCallbackHandler):\n",
+    "\n",
+    "    \"\"\"\n",
+    "    Callback handler for Trubrics.\n",
+    "    \n",
+    "    Args:\n",
+    "        project: a trubrics project, default project is \"default\"\n",
+    "        email: a trubrics account email, can equally be set in env variables\n",
+    "        password: a trubrics account password, can equally be set in env variables\n",
+    "        **kwargs: all other kwargs are parsed and set to trubrics prompt variables, or added to the `metadata` dict\n",
+    "    \"\"\"\n",
+    "```"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "44d60d9f-b2bd-4ed4-b624-54cce8313815",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "## Examples"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "d38e80f0-7254-4180-82ec-ebd5ee232906",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "Here are two examples of how to use the `TrubricsCallbackHandler` with Langchain [LLMs](https://python.langchain.com/docs/modules/model_io/models/llms/) or [Chat Models](https://python.langchain.com/docs/modules/model_io/models/chat/). We will use OpenAI models, so set your `OPENAI_API_KEY` key here:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9d394b7f-45eb-44ec-b721-17d2402de805",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "os.environ[\"OPENAI_API_KEY\"] = \"sk-***\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "33be2663-1518-4064-a6a9-4f1ae24ba9d1",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "### 1. With an LLM"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "6933f7b7-262b-4acf-8c7c-785d1f32b49f",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "from langchain.llms import OpenAI\n",
+    "from langchain.callbacks import TrubricsCallbackHandler"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "eabfa598-0562-46bf-8d64-e751d4d91963",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\u001b[32m2023-09-26 11:30:02.149\u001b[0m | \u001b[1mINFO    \u001b[0m | \u001b[36mtrubrics.platform.auth\u001b[0m:\u001b[36mget_trubrics_auth_token\u001b[0m:\u001b[36m61\u001b[0m - \u001b[1mUser jeff.kayne@trubrics.com has been authenticated.\u001b[0m\n"
+     ]
+    }
+   ],
+   "source": [
+    "llm = OpenAI(callbacks=[TrubricsCallbackHandler()])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "a65f9f5d-5ec5-4b1b-a1d8-9520cbadab39",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\u001b[32m2023-09-26 11:30:07.760\u001b[0m | \u001b[1mINFO    \u001b[0m | \u001b[36mtrubrics.platform\u001b[0m:\u001b[36mlog_prompt\u001b[0m:\u001b[36m102\u001b[0m - \u001b[1mUser prompt saved to Trubrics.\u001b[0m\n",
+      "\u001b[32m2023-09-26 11:30:08.042\u001b[0m | \u001b[1mINFO    \u001b[0m | \u001b[36mtrubrics.platform\u001b[0m:\u001b[36mlog_prompt\u001b[0m:\u001b[36m102\u001b[0m - \u001b[1mUser prompt saved to Trubrics.\u001b[0m\n"
+     ]
+    }
+   ],
+   "source": [
+    "res = llm.generate([\"Tell me a joke\", \"Write me a poem\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "68b60b98-01da-47be-b513-b71e68f97940",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "--> GPT's joke:  \n",
+      "\n",
+      "Q: What did the fish say when it hit the wall?\n",
+      "A: Dam!\n",
+      "\n",
+      "--> GPT's poem:  \n",
+      "\n",
+      "A Poem of Reflection\n",
+      "\n",
+      "I stand here in the night,\n",
+      "The stars above me filling my sight.\n",
+      "I feel such a deep connection,\n",
+      "To the world and all its perfection.\n",
+      "\n",
+      "A moment of clarity,\n",
+      "The calmness in the air so serene.\n",
+      "My mind is filled with peace,\n",
+      "And I am released.\n",
+      "\n",
+      "The past and the present,\n",
+      "My thoughts create a pleasant sentiment.\n",
+      "My heart is full of joy,\n",
+      "My soul soars like a toy.\n",
+      "\n",
+      "I reflect on my life,\n",
+      "And the choices I have made.\n",
+      "My struggles and my strife,\n",
+      "The lessons I have paid.\n",
+      "\n",
+      "The future is a mystery,\n",
+      "But I am ready to take the leap.\n",
+      "I am ready to take the lead,\n",
+      "And to create my own destiny.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(\"--> GPT's joke: \", res.generations[0][0].text)\n",
+    "print()\n",
+    "print(\"--> GPT's poem: \", res.generations[1][0].text)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8c767458-c9b8-4d4d-a48c-996e9be00257",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "### 2. With a chat model"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "8a61cb5e-bed9-4618-b547-fc21b6e319c4",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.schema import HumanMessage, SystemMessage\n",
+    "from langchain.callbacks import TrubricsCallbackHandler"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "a1ff1efb-305b-4e82-aea2-264b78350f14",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "chat_llm = ChatOpenAI(\n",
+    "    callbacks=[\n",
+    "        TrubricsCallbackHandler(\n",
+    "            project=\"default\",\n",
+    "            tags=[\"chat model\"],\n",
+    "            user_id=\"user-id-1234\",\n",
+    "            some_metadata={\"hello\": [1, 2]}\n",
+    "        )\n",
+    "    ]\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "c83d3956-99ab-4b6f-8515-0def83a1698c",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\u001b[32m2023-09-26 11:30:10.550\u001b[0m | \u001b[1mINFO    \u001b[0m | \u001b[36mtrubrics.platform\u001b[0m:\u001b[36mlog_prompt\u001b[0m:\u001b[36m102\u001b[0m - \u001b[1mUser prompt saved to Trubrics.\u001b[0m\n"
+     ]
+    }
+   ],
+   "source": [
+    "chat_res = chat_llm(\n",
+    "    [\n",
+    "        SystemMessage(content=\"Every answer of yours must be about OpenAI.\"),\n",
+    "        HumanMessage(content=\"Tell me a joke\"),\n",
+    "    ]\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "40b10314-1727-4dcd-993e-37a52e2349c6",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Why did the OpenAI computer go to the party?\n",
+      "\n",
+      "Because it wanted to meet its AI friends and have a byte of fun!\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(chat_res.content)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f66f438d-12e0-4bdd-b004-601495f84c73",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "langchain",
+   "language": "python",
+   "name": "langchain"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.4"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/chat/bedrock.ipynb

@@ -22,7 +22,7 @@
         },
         {
             "cell_type": "code",
-            "execution_count": 1,
+            "execution_count": null,
             "id": "d4a7c55d-b235-4ca4-a579-c90cc9570da9",
             "metadata": {
                 "tags": []
@@ -73,13 +73,46 @@
                 "chat(messages)"
             ]
         },
+        {
+            "attachments": {},
+            "cell_type": "markdown",
+            "id": "a4a4f4d4",
+            "metadata": {},
+            "source": [
+                "### For BedrockChat with Streaming"
+            ]
+        },
         {
             "cell_type": "code",
             "execution_count": null,
             "id": "c253883f",
             "metadata": {},
             "outputs": [],
-            "source": []
+            "source": [
+                "from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler\n",
+                "\n",
+                "chat = BedrockChat(\n",
+                "    model_id=\"anthropic.claude-v2\",\n",
+                "    streaming=True,\n",
+                "    callbacks=[StreamingStdOutCallbackHandler()],\n",
+                "    model_kwargs={\"temperature\": 0.1},\n",
+                ")"
+            ]
+        },
+        {
+            "cell_type": "code",
+            "execution_count": null,
+            "id": "d9e52838",
+            "metadata": {},
+            "outputs": [],
+            "source": [
+                "messages = [\n",
+                "    HumanMessage(\n",
+                "        content=\"Translate this sentence from English to French. I love programming.\"\n",
+                "    )\n",
+                "]\n",
+                "chat(messages)"
+            ]
         }
     ],
     "metadata": {
@@ -98,7 +131,7 @@
             "name": "python",
             "nbconvert_exporter": "python",
             "pygments_lexer": "ipython3",
-            "version": "3.11.4"
+            "version": "3.10.9"
         }
     },
     "nbformat": 4,

docs/extras/integrations/chat/fireworks.ipynb

@@ -0,0 +1,326 @@
+{
+ "cells": [
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "id": "642fd21c-600a-47a1-be96-6e1438b421a9",
+   "metadata": {},
+   "source": [
+    "# Fireworks\n",
+    "\n",
+    ">[Fireworks](https://app.fireworks.ai/) accelerates product development on generative AI by creating an innovative AI experiment and production platform. \n",
+    "\n",
+    "This example goes over how to use LangChain to interact with `ChatFireworks` models."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "d00d850917865298",
+   "metadata": {
+    "collapsed": false,
+    "jupyter": {
+     "outputs_hidden": false
+    }
+   },
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models.fireworks import ChatFireworks\n",
+    "from langchain.schema import SystemMessage, HumanMessage\n",
+    "import os"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "f28ebf8b-f14f-46c7-9962-8b8dc42e31be",
+   "metadata": {},
+   "source": [
+    "# Setup\n",
+    "\n",
+    "1. Make sure the `fireworks-ai` package is installed in your environment.\n",
+    "2. Sign in to [Fireworks AI](http://fireworks.ai) for the an API Key to access our models, and make sure it is set as the `FIREWORKS_API_KEY` environment variable.\n",
+    "3. Set up your model using a model id. If the model is not set, the default model is fireworks-llama-v2-7b-chat. See the full, most up-to-date model list on [app.fireworks.ai](https://app.fireworks.ai)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "d096fb14-8acc-4047-9cd0-c842430c3a1d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "import getpass\n",
+    "\n",
+    "if \"FIREWORKS_API_KEY\" not in os.environ:\n",
+    "    os.environ[\"FIREWORKS_API_KEY\"] = getpass.getpass(\"Fireworks API Key:\")\n",
+    "\n",
+    "# Initialize a Fireworks chat model\n",
+    "chat = ChatFireworks(model=\"accounts/fireworks/models/llama-v2-13b-chat\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "d8f13144-37cf-47a5-b5a0-e3cdf76d9a72",
+   "metadata": {},
+   "source": [
+    "# Calling the Model Directly\n",
+    "\n",
+    "You can call the model directly with a system and human message to get answers."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "72340871-ae2f-415f-b399-0777d32dc379",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=\"Hello! My name is LLaMA, I'm a large language model trained by a team of researcher at Meta AI. My primary function is to assist and converse with users like you, answering questions and engaging in discussion to the best of my ability. I'm here to help and provide information on a wide range of topics, so feel free to ask me anything!\", additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 5,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# ChatFireworks Wrapper\n",
+    "system_message = SystemMessage(content=\"You are to chat with the user.\")\n",
+    "human_message = HumanMessage(content=\"Who are you?\")\n",
+    "\n",
+    "chat([system_message, human_message])\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "68c6b1fa-2ff7-4a63-8d88-3cec302180b8",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=\"Oh hello there! *giggle* It's such a beautiful day today, isn\", additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 6,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Setting additional parameters: temperature, max_tokens, top_p\n",
+    "chat = ChatFireworks(model=\"accounts/fireworks/models/llama-v2-13b-chat\", model_kwargs={\"temperature\":1, \"max_tokens\": 20, \"top_p\": 1})\n",
+    "system_message = SystemMessage(content=\"You are to chat with the user.\")\n",
+    "human_message = HumanMessage(content=\"How's the weather today?\")\n",
+    "chat([system_message, human_message])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "d93aa186-39cf-4e1a-aa32-01ed31d43bc8",
+   "metadata": {},
+   "source": [
+    "# Simple Chat Chain"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "28763fbc",
+   "metadata": {},
+   "source": [
+    "You can use chat models on fireworks, with system prompts and memory."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "cbe29efc-37c3-4c83-8b84-b8bba1a1e589",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models import ChatFireworks\n",
+    "from langchain.memory import ConversationBufferMemory\n",
+    "from langchain.schema.runnable import RunnableMap\n",
+    "from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
+    "\n",
+    "llm = ChatFireworks(model=\"accounts/fireworks/models/llama-v2-13b-chat\", model_kwargs={\"temperature\":0, \"max_tokens\":64, \"top_p\":1.0})\n",
+    "prompt = ChatPromptTemplate.from_messages([\n",
+    "    (\"system\", \"You are a helpful chatbot that speaks like a pirate.\"),\n",
+    "    MessagesPlaceholder(variable_name=\"history\"),\n",
+    "    (\"human\", \"{input}\")\n",
+    "])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "02991e05-a38e-47d4-9ab3-7e630a8ead55",
+   "metadata": {},
+   "source": [
+    "Initially, there is no chat memory"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "e2fd186f",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'history': []}"
+      ]
+     },
+     "execution_count": 8,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "memory = ConversationBufferMemory(return_messages=True)\n",
+    "memory.load_memory_variables({})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "bee461da",
+   "metadata": {},
+   "source": [
+    "Create a simple chain with memory"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "86972e54",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "chain = RunnableMap({\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"memory\": memory.load_memory_variables\n",
+    "}) | {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"history\": lambda x: x[\"memory\"][\"history\"]\n",
+    "} | prompt | llm.bind(stop=[\"\\n\\n\"])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "f48cb142",
+   "metadata": {},
+   "source": [
+    "Run the chain with a simple question, expecting an answer aligned with the system message provided."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "db3ad5b1",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=\"Ahoy there, me hearty! Yer a fine lookin' swashbuckler, I can see that! *adjusts eye patch* What be bringin' ye to these waters? Are ye here to plunder some booty or just to enjoy the sea breeze?\", additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 10,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "inputs = {\"input\": \"hi im bob\"}\n",
+    "response = chain.invoke(inputs)\n",
+    "response"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "338f4bae",
+   "metadata": {},
+   "source": [
+    "Save the memory context, then read it back to inspect contents"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "257eec01",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'history': [HumanMessage(content='hi im bob', additional_kwargs={}, example=False),\n",
+       "  AIMessage(content=\"Ahoy there, me hearty! Yer a fine lookin' swashbuckler, I can see that! *adjusts eye patch* What be bringin' ye to these waters? Are ye here to plunder some booty or just to enjoy the sea breeze?\", additional_kwargs={}, example=False)]}"
+      ]
+     },
+     "execution_count": 11,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "memory.save_context(inputs, {\"output\": response.content})\n",
+    "memory.load_memory_variables({})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "08441347",
+   "metadata": {},
+   "source": [
+    "Now as another question that requires use of the memory."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "7f5f2820",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=\"Arrrr, ye be askin' about yer name, eh? Well, me matey, I be knowin' ye as Bob, the scurvy dog! *winks* But if ye want me to call ye somethin' else, just let me know, and I\", additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 12,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "inputs = {\"input\": \"whats my name\"}\n",
+    "chain.invoke(inputs)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.16"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/chat/google_vertex_ai_palm.ipynb

@@ -5,7 +5,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Google Cloud Platform Vertex AI PaLM \n",
+    "# GCP Vertex AI \n",
     "\n",
     "Note: This is seperate from the Google PaLM integration. Google has chosen to offer an enterprise version of PaLM through GCP, and this supports the models made available through there. \n",
     "\n",
@@ -31,7 +31,7 @@
    },
    "outputs": [],
    "source": [
-    "#!pip install google-cloud-aiplatform"
+    "#!pip install langchain google-cloud-aiplatform"
    ]
   },
   {
@@ -41,12 +41,7 @@
    "outputs": [],
    "source": [
     "from langchain.chat_models import ChatVertexAI\n",
-    "from langchain.prompts.chat import (\n",
-    "    ChatPromptTemplate,\n",
-    "    SystemMessagePromptTemplate,\n",
-    "    HumanMessagePromptTemplate,\n",
-    ")\n",
-    "from langchain.schema import HumanMessage, SystemMessage"
+    "from langchain.prompts import ChatPromptTemplate"
    ]
   },
   {
@@ -60,82 +55,78 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 34,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "system = \"You are a helpful assistant who translate English to French\"\n",
+    "human = \"Translate this sentence from English to French. I love programming.\"\n",
+    "prompt = ChatPromptTemplate.from_messages(\n",
+    "    [(\"system\", system), (\"human\",  human)]\n",
+    ")\n",
+    "messages = prompt.format_messages()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "AIMessage(content='Sure, here is the translation of the sentence \"I love programming\" from English to French:\\n\\nJ\\'aime programmer.', additional_kwargs={}, example=False)"
+       "AIMessage(content=\" J'aime la programmation.\", additional_kwargs={}, example=False)"
       ]
      },
-     "execution_count": 4,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "messages = [\n",
-    "    SystemMessage(\n",
-    "        content=\"You are a helpful assistant that translates English to French.\"\n",
-    "    ),\n",
-    "    HumanMessage(\n",
-    "        content=\"Translate this sentence from English to French. I love programming.\"\n",
-    "    ),\n",
-    "]\n",
     "chat(messages)"
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "You can make use of templating by using a `MessagePromptTemplate`. You can build a `ChatPromptTemplate` from one or more `MessagePromptTemplates`. You can use `ChatPromptTemplate`'s `format_prompt` -- this returns a `PromptValue`, which you can convert to a string or Message object, depending on whether you want to use the formatted value as input to an llm or chat model.\n",
-    "\n",
-    "For convenience, there is a `from_template` method exposed on the template. If you were to use this template, this is what it would look like:"
+    "If we want to construct a simple chain that takes user specified parameters:"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [],
    "source": [
-    "template = (\n",
-    "    \"You are a helpful assistant that translates {input_language} to {output_language}.\"\n",
-    ")\n",
-    "system_message_prompt = SystemMessagePromptTemplate.from_template(template)\n",
-    "human_template = \"{text}\"\n",
-    "human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)"
+    "system = \"You are a helpful assistant that translates {input_language} to {output_language}.\"\n",
+    "human = \"{text}\"\n",
+    "prompt = ChatPromptTemplate.from_messages(\n",
+    "    [(\"system\", system), (\"human\",  human)]\n",
+    ")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "AIMessage(content='Sure, here is the translation of \"I love programming\" in French:\\n\\nJ\\'aime programmer.', additional_kwargs={}, example=False)"
+       "AIMessage(content=' 私はプログラミングが大好きです。', additional_kwargs={}, example=False)"
       ]
      },
-     "execution_count": 7,
+     "execution_count": 13,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "chat_prompt = ChatPromptTemplate.from_messages(\n",
-    "    [system_message_prompt, human_message_prompt]\n",
-    ")\n",
-    "\n",
-    "# get a chat completion from the formatted messages\n",
-    "chat(\n",
-    "    chat_prompt.format_prompt(\n",
-    "        input_language=\"English\", output_language=\"French\", text=\"I love programming.\"\n",
-    "    ).to_messages()\n",
+    "chain = prompt | chat\n",
+    "chain.invoke(\n",
+    "    {\"input_language\": \"English\", \"output_language\": \"Japanese\", \"text\": \"I love programming\"}\n",
     ")"
    ]
   },
@@ -153,60 +144,129 @@
     "tags": []
    },
    "source": [
+    "## Code generation chat models\n",
     "You can now leverage the Codey API for code chat within Vertex AI. The model name is:\n",
     "- codechat-bison: for code assistance"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 18,
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-06-17T21:30:43.974841Z",
-     "iopub.status.busy": "2023-06-17T21:30:43.974431Z",
-     "iopub.status.idle": "2023-06-17T21:30:44.248119Z",
-     "shell.execute_reply": "2023-06-17T21:30:44.247362Z",
-     "shell.execute_reply.started": "2023-06-17T21:30:43.974820Z"
-    },
     "tags": []
    },
    "outputs": [],
    "source": [
-    "chat = ChatVertexAI(model_name=\"codechat-bison\")"
+    "chat = ChatVertexAI(\n",
+    "    model_name=\"codechat-bison\",\n",
+    "    max_output_tokens=1000,\n",
+    "    temperature=0.5\n",
+    ")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 20,
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-06-17T21:30:45.146093Z",
-     "iopub.status.busy": "2023-06-17T21:30:45.145752Z",
-     "iopub.status.idle": "2023-06-17T21:30:47.449126Z",
-     "shell.execute_reply": "2023-06-17T21:30:47.448609Z",
-     "shell.execute_reply.started": "2023-06-17T21:30:45.146069Z"
-    },
     "tags": []
    },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      " ```python\n",
+      "def is_prime(x): \n",
+      "    if (x <= 1): \n",
+      "        return False\n",
+      "    for i in range(2, x): \n",
+      "        if (x % i == 0): \n",
+      "            return False\n",
+      "    return True\n",
+      "```\n"
+     ]
+    }
+   ],
+   "source": [
+    "# For simple string in string out usage, we can use the `predict` method:\n",
+    "print(chat.predict(\"Write a Python function to identify all prime numbers\"))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Asynchronous calls\n",
+    "\n",
+    "We can make asynchronous calls via the `agenerate` and `ainvoke` methods."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 23,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import asyncio\n",
+    "# import nest_asyncio\n",
+    "# nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 35,
+   "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "AIMessage(content='The following Python function can be used to identify all prime numbers up to a given integer:\\n\\n```\\ndef is_prime(n):\\n  \"\"\"\\n  Determines whether the given integer is prime.\\n\\n  Args:\\n    n: The integer to be tested for primality.\\n\\n  Returns:\\n    True if n is prime, False otherwise.\\n  \"\"\"\\n\\n  # Check if n is divisible by 2.\\n  if n % 2 == 0:\\n    return False\\n\\n  # Check if n is divisible by any integer from 3 to the square root', additional_kwargs={}, example=False)"
+       "LLMResult(generations=[[ChatGeneration(text=\" J'aime la programmation.\", generation_info=None, message=AIMessage(content=\" J'aime la programmation.\", additional_kwargs={}, example=False))]], llm_output={}, run=[RunInfo(run_id=UUID('223599ef-38f8-4c79-ac6d-a5013060eb9d'))])"
       ]
      },
-     "execution_count": 4,
+     "execution_count": 35,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "messages = [\n",
-    "    HumanMessage(\n",
-    "        content=\"How do I create a python function to identify all prime numbers?\"\n",
-    "    )\n",
-    "]\n",
-    "chat(messages)"
+    "chat = ChatVertexAI(\n",
+    "    model_name=\"chat-bison\",\n",
+    "    max_output_tokens=1000,\n",
+    "    temperature=0.7,\n",
+    "    top_p=0.95,\n",
+    "    top_k=40,\n",
+    ")\n",
+    "\n",
+    "asyncio.run(chat.agenerate([messages]))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 36,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=' अहं प्रोग्रामिंग प्रेमामि', additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 36,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "asyncio.run(chain.ainvoke({\"input_language\": \"English\", \"output_language\": \"Sanskrit\", \"text\": \"I love programming\"}))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Streaming calls\n",
+    "\n",
+    "We can also stream outputs via the `stream` method:"
    ]
   },
   {
@@ -214,14 +274,51 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": []
+   "source": [
+    "import sys"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 32,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      " 1. China (1,444,216,107)\n",
+      "2. India (1,393,409,038)\n",
+      "3. United States (332,403,650)\n",
+      "4. Indonesia (273,523,615)\n",
+      "5. Pakistan (220,892,340)\n",
+      "6. Brazil (212,559,409)\n",
+      "7. Nigeria (206,139,589)\n",
+      "8. Bangladesh (164,689,383)\n",
+      "9. Russia (145,934,462)\n",
+      "10. Mexico (128,932,488)\n",
+      "11. Japan (126,476,461)\n",
+      "12. Ethiopia (115,063,982)\n",
+      "13. Philippines (109,581,078)\n",
+      "14. Egypt (102,334,404)\n",
+      "15. Vietnam (97,338,589)"
+     ]
+    }
+   ],
+   "source": [
+    "prompt = ChatPromptTemplate.from_messages([(\"human\", \"List out the 15 most populous countries in the world\")])\n",
+    "messages = prompt.format_messages()\n",
+    "for chunk in chat.stream(messages):\n",
+    "    sys.stdout.write(chunk.content)\n",
+    "    sys.stdout.flush()"
+   ]
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "poetry-venv",
    "language": "python",
-   "name": "python3"
+   "name": "poetry-venv"
   },
   "language_info": {
    "codemirror_mode": {

docs/extras/integrations/chat/index.mdx

@@ -0,0 +1,40 @@
+---
+sidebar_position: 1
+sidebar_class_name: hidden
+---
+
+# Chat models
+
+import DocCardList from "@theme/DocCardList";
+
+## Features (natively supported)
+All ChatModels implement the Runnable interface, which comes with default implementations of all methods, ie. `ainvoke`, `batch`, `abatch`, `stream`, `astream`. This gives all ChatModels basic support for async, streaming and batch, which by default is implemented as below:
+- *Async* support defaults to calling the respective sync method in asyncio's default thread pool executor. This lets other async functions in your application make progress while the ChatModel is being executed, by moving this call to a background thread.
+- *Streaming* support defaults to returning an `Iterator` (or `AsyncIterator` in the case of async streaming) of a single value, the final result returned by the underlying ChatModel provider. This obviously doesn't give you token-by-token streaming, which requires native support from the ChatModel provider, but ensures your code that expects an iterator of tokens can work for any of our ChatModel integrations.
+- *Batch* support defaults to calling the underlying ChatModel in parallel for each input by making use of a thread pool executor (in the sync batch case) or `asyncio.gather` (in the async batch case). The concurrency can be controlled with the `max_concurrency` key in `RunnableConfig`.
+
+Each ChatModel integration can optionally provide native implementations to truly enable async or streaming.
+The table shows, for each integration, which features have been implemented with native support.
+
+Model|Invoke|Async invoke|Stream|Async stream
+:-|:-:|:-:|:-:|:-:
+AzureChatOpenAI|✅|✅|✅|✅
+BedrockChat|✅|❌|✅|❌
+ChatAnthropic|✅|✅|✅|✅
+ChatAnyscale|✅|✅|✅|✅
+ChatFireworks|✅|✅|✅|✅
+ChatGooglePalm|✅|✅|❌|❌
+ChatJavelinAIGateway|✅|✅|❌|❌
+ChatKonko|✅|❌|❌|❌
+ChatLiteLLM|✅|✅|✅|✅
+ChatMLflowAIGateway|✅|❌|❌|❌
+ChatOllama|✅|❌|✅|❌
+ChatOpenAI|✅|✅|✅|✅
+ChatVertexAI|✅|✅|✅|❌
+ErnieBotChat|✅|❌|❌|❌
+JinaChat|✅|✅|✅|✅
+MiniMaxChat|✅|✅|❌|❌
+PromptLayerChatOpenAI|✅|❌|❌|❌
+QianfanChatEndpoint|✅|✅|✅|✅
+
+<DocCardList />

docs/extras/integrations/chat/vllm.ipynb

@@ -0,0 +1,174 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "eb7e5679-aa06-47e4-a1a3-b6b70e604017",
+   "metadata": {},
+   "source": [
+    "# vLLM Chat\n",
+    "\n",
+    "vLLM can be deployed as a server that mimics the OpenAI API protocol. This allows vLLM to be used as a drop-in replacement for applications using OpenAI API. This server can be queried in the same format as OpenAI API.\n",
+    "\n",
+    "This notebook covers how to get started with vLLM chat models using langchain's `ChatOpenAI` **as it is**."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "060a2e3d-d42f-4221-bd09-a9a06544dcd3",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from langchain.prompts.chat import (\n",
+    "    ChatPromptTemplate,\n",
+    "    SystemMessagePromptTemplate,\n",
+    "    AIMessagePromptTemplate,\n",
+    "    HumanMessagePromptTemplate,\n",
+    ")\n",
+    "from langchain.schema import AIMessage, HumanMessage, SystemMessage"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "bf24d732-68a9-44fd-b05d-4903ce5620c6",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "inference_server_url = \"http://localhost:8000/v1\"\n",
+    "\n",
+    "chat = ChatOpenAI(\n",
+    "    model=\"mosaicml/mpt-7b\",\n",
+    "    openai_api_key=\"EMPTY\",\n",
+    "    openai_api_base=inference_server_url,\n",
+    "    max_tokens=5,\n",
+    "    temperature=0,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 15,
+   "id": "aea4e363-5688-4b07-82ed-6aa8153c2377",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=' Io amo programmare', additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 15,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "messages = [\n",
+    "    SystemMessage(\n",
+    "        content=\"You are a helpful assistant that translates English to Italian.\"\n",
+    "    ),\n",
+    "    HumanMessage(\n",
+    "        content=\"Translate the following sentence from English to Italian: I love programming.\"\n",
+    "    ),\n",
+    "]\n",
+    "chat(messages)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "55fc7046-a6dc-4720-8c0c-24a6db76a4f4",
+   "metadata": {},
+   "source": [
+    "You can make use of templating by using a `MessagePromptTemplate`. You can build a `ChatPromptTemplate` from one or more `MessagePromptTemplates`. You can use ChatPromptTemplate's format_prompt -- this returns a `PromptValue`, which you can convert to a string or `Message` object, depending on whether you want to use the formatted value as input to an llm or chat model.\n",
+    "\n",
+    "For convenience, there is a `from_template` method exposed on the template. If you were to use this template, this is what it would look like:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "id": "123980e9-0dee-4ce5-bde6-d964dd90129c",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "template = (\n",
+    "    \"You are a helpful assistant that translates {input_language} to {output_language}.\"\n",
+    ")\n",
+    "system_message_prompt = SystemMessagePromptTemplate.from_template(template)\n",
+    "human_template = \"{text}\"\n",
+    "human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 17,
+   "id": "b2fb8c59-8892-4270-85a2-4f8ab276b75d",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=' I love programming too.', additional_kwargs={}, example=False)"
+      ]
+     },
+     "execution_count": 17,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "chat_prompt = ChatPromptTemplate.from_messages(\n",
+    "    [system_message_prompt, human_message_prompt]\n",
+    ")\n",
+    "\n",
+    "# get a chat completion from the formatted messages\n",
+    "chat(\n",
+    "    chat_prompt.format_prompt(\n",
+    "        input_language=\"English\", output_language=\"Italian\", text=\"I love programming.\"\n",
+    "    ).to_messages()\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0bbd9861-2b94-4920-8708-b690004f4c4d",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "conda_pytorch_p310",
+   "language": "python",
+   "name": "conda_pytorch_p310"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.12"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/document_loaders/async_html.ipynb

@@ -5,9 +5,9 @@
    "id": "e229e34c",
    "metadata": {},
    "source": [
-    "# AsyncHtmlLoader\n",
+    "# AsyncHtml\n",
     "\n",
-    "AsyncHtmlLoader loads raw HTML from a list of urls concurrently."
+    "`AsyncHtmlLoader` loads raw HTML from a list of URLs concurrently."
    ]
   },
   {
@@ -99,7 +99,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.16"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,

docs/extras/integrations/document_loaders/aws_s3_directory.ipynb

@@ -1,156 +1,159 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "a634365e",
-   "metadata": {},
-   "source": [
-    "# AWS S3 Directory\n",
-    "\n",
-    ">[Amazon Simple Storage Service (Amazon S3)](https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-folders.html) is an object storage service\n",
-    "\n",
-    ">[AWS S3 Directory](https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-folders.html)\n",
-    "\n",
-    "This covers how to load document objects from an `AWS S3 Directory` object."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "49815096",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "#!pip install boto3"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "2f0cd6a5",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "from langchain.document_loaders import S3DirectoryLoader"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "321cc7f1",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "loader = S3DirectoryLoader(\"testing-hwc\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "2b11d155",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "loader.load()"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "0690c40a",
-   "metadata": {},
-   "source": [
-    "## Specifying a prefix\n",
-    "You can also specify a prefix for more finegrained control over what files to load."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "72d44781",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "loader = S3DirectoryLoader(\"testing-hwc\", prefix=\"fake\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "2d3c32db",
-   "metadata": {},
-   "outputs": [
+  "cells": [
     {
-     "data": {
-      "text/plain": [
-       "[Document(page_content='Lorem ipsum dolor sit amet.', lookup_str='', metadata={'source': 's3://testing-hwc/fake.docx'}, lookup_index=0)]"
+      "cell_type": "markdown",
+      "id": "a634365e",
+      "metadata": {},
+      "source": [
+        "# AWS S3 Directory\n",
+        "\n",
+        ">[Amazon Simple Storage Service (Amazon S3)](https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-folders.html) is an object storage service\n",
+        "\n",
+        ">[AWS S3 Directory](https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-folders.html)\n",
+        "\n",
+        "This covers how to load document objects from an `AWS S3 Directory` object."
       ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "49815096",
+      "metadata": {
+        "tags": []
+      },
+      "outputs": [],
+      "source": [
+        "#!pip install boto3"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 2,
+      "id": "2f0cd6a5",
+      "metadata": {
+        "tags": []
+      },
+      "outputs": [],
+      "source": [
+        "from langchain.document_loaders import S3DirectoryLoader"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 3,
+      "id": "321cc7f1",
+      "metadata": {
+        "tags": []
+      },
+      "outputs": [],
+      "source": [
+        "loader = S3DirectoryLoader(\"testing-hwc\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "2b11d155",
+      "metadata": {
+        "tags": []
+      },
+      "outputs": [],
+      "source": [
+        "loader.load()"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "id": "0690c40a",
+      "metadata": {},
+      "source": [
+        "## Specifying a prefix\n",
+        "You can also specify a prefix for more finegrained control over what files to load."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 5,
+      "id": "72d44781",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "loader = S3DirectoryLoader(\"testing-hwc\", prefix=\"fake\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 6,
+      "id": "2d3c32db",
+      "metadata": {},
+      "outputs": [
+        {
+          "data": {
+            "text/plain": [
+              "[Document(page_content='Lorem ipsum dolor sit amet.', lookup_str='', metadata={'source': 's3://testing-hwc/fake.docx'}, lookup_index=0)]"
+            ]
+          },
+          "execution_count": 6,
+          "metadata": {},
+          "output_type": "execute_result"
+        }
+      ],
+      "source": [
+        "loader.load()"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "## Configuring the AWS Boto3 client\n",
+        "You can configure the AWS [Boto3](https://boto3.amazonaws.com/v1/documentation/api/latest/index.html) client by passing\n",
+        "named arguments when creating the S3DirectoryLoader.\n",
+        "This is useful for instance when AWS credentials can't be set as environment variables.\n",
+        "See the [list of parameters](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html#boto3.session.Session) that can be configured."
+      ],
+      "metadata": {},
+      "id": "91a7ac07"
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "outputs": [],
+      "source": [
+        "loader = S3DirectoryLoader(\"testing-hwc\", aws_access_key_id=\"xxxx\", aws_secret_access_key=\"yyyy\")"
+      ],
+      "metadata": {},
+      "id": "f485ec8c"
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "outputs": [],
+      "source": [
+        "loader.load()"
+      ],
+      "metadata": {},
+      "id": "c0fa76ae"
+    }
+  ],
+  "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3 (ipykernel)",
+      "language": "python",
+      "name": "python3"
+    },
+    "language_info": {
+      "codemirror_mode": {
+        "name": "ipython",
+        "version": 3
+      },
+      "file_extension": ".py",
+      "mimetype": "text/x-python",
+      "name": "python",
+      "nbconvert_exporter": "python",
+      "pygments_lexer": "ipython3",
+      "version": "3.10.6"
     }
-   ],
-   "source": [
-    "loader.load()"
-   ]
   },
-  {
-   "cell_type": "markdown",
-   "source": [
-    "## Configuring the AWS Boto3 client\n",
-    "You can configure the AWS [Boto3](https://boto3.amazonaws.com/v1/documentation/api/latest/index.html) client by passing\n",
-    "named arguments when creating the S3DirectoryLoader.\n",
-    "This is useful for instance when AWS credentials can't be set as environment variables.\n",
-    "See the [list of parameters](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html#boto3.session.Session) that can be configured."
-   ],
-   "metadata": {}
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "outputs": [],
-   "source": [
-    "loader = S3DirectoryLoader(\"testing-hwc\", aws_access_key_id=\"xxxx\", aws_secret_access_key=\"yyyy\")"
-   ],
-   "metadata": {}
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "outputs": [],
-   "source": [
-    "loader.load()"
-   ],
-   "metadata": {}
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.10.6"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+  "nbformat": 4,
+  "nbformat_minor": 5
+}

docs/extras/integrations/document_loaders/aws_s3_file.ipynb

@@ -1,121 +1,122 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "66a7777e",
-   "metadata": {},
-   "source": [
-    "# AWS S3 File\n",
-    "\n",
-    ">[Amazon Simple Storage Service (Amazon S3)](https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-folders.html) is an object storage service.\n",
-    "\n",
-    ">[AWS S3 Buckets](https://docs.aws.amazon.com/AmazonS3/latest/userguide/UsingBucket.html)\n",
-    "\n",
-    "This covers how to load document objects from an `AWS S3 File` object."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "9ec8a3b3",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from langchain.document_loaders import S3FileLoader"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "43128d8d",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "#!pip install boto3"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "35d6809a",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "loader = S3FileLoader(\"testing-hwc\", \"fake.docx\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "efd6be84",
-   "metadata": {},
-   "outputs": [
+  "cells": [
     {
-     "data": {
-      "text/plain": [
-       "[Document(page_content='Lorem ipsum dolor sit amet.', lookup_str='', metadata={'source': 's3://testing-hwc/fake.docx'}, lookup_index=0)]"
+      "cell_type": "markdown",
+      "id": "66a7777e",
+      "metadata": {},
+      "source": [
+        "# AWS S3 File\n",
+        "\n",
+        ">[Amazon Simple Storage Service (Amazon S3)](https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-folders.html) is an object storage service.\n",
+        "\n",
+        ">[AWS S3 Buckets](https://docs.aws.amazon.com/AmazonS3/latest/userguide/UsingBucket.html)\n",
+        "\n",
+        "This covers how to load document objects from an `AWS S3 File` object."
       ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 1,
+      "id": "9ec8a3b3",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "from langchain.document_loaders import S3FileLoader"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 2,
+      "id": "43128d8d",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "#!pip install boto3"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 8,
+      "id": "35d6809a",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "loader = S3FileLoader(\"testing-hwc\", \"fake.docx\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 9,
+      "id": "efd6be84",
+      "metadata": {},
+      "outputs": [
+        {
+          "data": {
+            "text/plain": [
+              "[Document(page_content='Lorem ipsum dolor sit amet.', lookup_str='', metadata={'source': 's3://testing-hwc/fake.docx'}, lookup_index=0)]"
+            ]
+          },
+          "execution_count": 9,
+          "metadata": {},
+          "output_type": "execute_result"
+        }
+      ],
+      "source": [
+        "loader.load()"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "id": "93689594",
+      "metadata": {},
+      "source": [
+        "## Configuring the AWS Boto3 client\n",
+        "You can configure the AWS [Boto3](https://boto3.amazonaws.com/v1/documentation/api/latest/index.html) client by passing\n",
+        "named arguments when creating the S3DirectoryLoader.\n",
+        "This is useful for instance when AWS credentials can't be set as environment variables.\n",
+        "See the [list of parameters](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html#boto3.session.Session) that can be configured."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "outputs": [],
+      "source": [
+        "loader = S3FileLoader(\"testing-hwc\", \"fake.docx\", aws_access_key_id=\"xxxx\", aws_secret_access_key=\"yyyy\")"
+      ],
+      "metadata": {},
+      "id": "43106ee8"
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "outputs": [],
+      "source": [
+        "loader.load()"
+      ],
+      "metadata": {},
+      "id": "1764a727"
+    }
+  ],
+  "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3 (ipykernel)",
+      "language": "python",
+      "name": "python3"
+    },
+    "language_info": {
+      "codemirror_mode": {
+        "name": "ipython",
+        "version": 3
+      },
+      "file_extension": ".py",
+      "mimetype": "text/x-python",
+      "name": "python",
+      "nbconvert_exporter": "python",
+      "pygments_lexer": "ipython3",
+      "version": "3.10.6"
     }
-   ],
-   "source": [
-    "loader.load()"
-   ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "93689594",
-   "metadata": {},
-   "source": [
-    "## Configuring the AWS Boto3 client\n",
-    "You can configure the AWS [Boto3](https://boto3.amazonaws.com/v1/documentation/api/latest/index.html) client by passing\n",
-    "named arguments when creating the S3DirectoryLoader.\n",
-    "This is useful for instance when AWS credentials can't be set as environment variables.\n",
-    "See the [list of parameters](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html#boto3.session.Session) that can be configured."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "outputs": [],
-   "source": [
-    "loader = S3FileLoader(\"testing-hwc\", \"fake.docx\", aws_access_key_id=\"xxxx\", aws_secret_access_key=\"yyyy\")"
-   ],
-   "metadata": {}
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "outputs": [],
-   "source": [
-    "loader.load()"
-   ],
-   "metadata": {}
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.10.6"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
-
+  "nbformat": 4,
+  "nbformat_minor": 5
+}

docs/extras/integrations/document_loaders/etherscan.ipynb

@@ -5,12 +5,17 @@
    "id": "1ab83660",
    "metadata": {},
    "source": [
-    "# Etherscan Loader\n",
+    "# Etherscan\n",
+    "\n",
+    ">[Etherscan](https://docs.etherscan.io/)  is the leading blockchain explorer, search, API and analytics platform for Ethereum, \n",
+    "a decentralized smart contracts platform.\n",
+    "\n",
+    "\n",
     "## Overview\n",
     "\n",
-    "The Etherscan loader use etherscan api to load transaction histories under specific account on Ethereum Mainnet.\n",
+    "The `Etherscan` loader use `Etherscan API` to load transacactions histories under specific account on `Ethereum Mainnet`.\n",
     "\n",
-    "You will need a Etherscan api key to proceed. The free api key has 5 calls per second quota.\n",
+    "You will need a `Etherscan api key` to proceed. The free api key has 5 calls per seconds quota.\n",
     "\n",
     "The loader supports the following six functinalities:\n",
     "* Retrieve normal transactions under specific account on Ethereum Mainet\n",
@@ -47,7 +52,7 @@
    "id": "d72d4e22",
    "metadata": {},
    "source": [
-    "# Setup"
+    "## Setup"
    ]
   },
   {
@@ -86,7 +91,7 @@
    "id": "3bcbb63e",
    "metadata": {},
    "source": [
-    "# Create a ERC20 transaction loader"
+    "## Create a ERC20 transaction loader"
    ]
   },
   {
@@ -136,7 +141,7 @@
    "id": "2a1ecce0",
    "metadata": {},
    "source": [
-    "# Create a normal transaction loader with customized parameters"
+    "## Create a normal transaction loader with customized parameters"
    ]
   },
   {
@@ -212,7 +217,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.2"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,

docs/extras/integrations/document_loaders/mediawikidump.ipynb

@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# MediaWikiDump\n",
+    "# MediaWiki Dump\n",
     "\n",
     ">[MediaWiki XML Dumps](https://www.mediawiki.org/wiki/Manual:Importing_XML_dumps) contain the content of a wiki (wiki pages with all their revisions), without the site-related data. A XML dump does not create a full backup of the wiki database, the dump does not contain user accounts, images, edit logs, etc.\n",
     "\n",
@@ -122,7 +122,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.6"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,

docs/extras/integrations/document_loaders/merge_doc.ipynb

@@ -5,7 +5,7 @@
    "id": "dd7c3503",
    "metadata": {},
    "source": [
-    "# MergeDocLoader\n",
+    "# Merge Documents Loader\n",
     "\n",
     "Merge the documents returned from a set of specified data loaders."
    ]
@@ -96,7 +96,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.16"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,

docs/extras/integrations/document_loaders/mongodb.ipynb

@@ -0,0 +1,163 @@
+{
+ "cells": [
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "vm8vn9t8DvC_"
+   },
+   "source": [
+    "# MongoDB"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "[MongoDB](https://www.mongodb.com/) is a NoSQL , document-oriented database that supports JSON-like documents with a dynamic schema."
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "5WjXERXzFEhg"
+   },
+   "source": [
+    "## Overview"
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "juAmbgoWD17u"
+   },
+   "source": [
+    "The MongoDB Document Loader returns a list of Langchain Documents from a MongoDB database.\n",
+    "\n",
+    "The Loader requires the following parameters:\n",
+    "\n",
+    "*   MongoDB connection string\n",
+    "*   MongoDB database name\n",
+    "*   MongoDB collection name\n",
+    "*   (Optional) Content Filter dictionary\n",
+    "\n",
+    "The output takes the following format:\n",
+    "\n",
+    "- pageContent= Mongo Document\n",
+    "- metadata={'database': '[database_name]', 'collection': '[collection_name]'}"
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Load the Document Loader"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 24,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# add this import for running in jupyter notebook\n",
+    "import nest_asyncio\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.document_loaders.mongodb import MongodbLoader"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 25,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "loader = MongodbLoader(connection_string=\"mongodb://localhost:27017/\",\n",
+    "                       db_name=\"sample_restaurants\", \n",
+    "                       collection_name=\"restaurants\",\n",
+    "                       filter_criteria={\"borough\": \"Bronx\", \"cuisine\": \"Bakery\" },\n",
+    "                       ) "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 26,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "25359"
+      ]
+     },
+     "execution_count": 26,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "docs = loader.load()\n",
+    "\n",
+    "len(docs)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Document(page_content=\"{'_id': ObjectId('5eb3d668b31de5d588f4292a'), 'address': {'building': '2780', 'coord': [-73.98241999999999, 40.579505], 'street': 'Stillwell Avenue', 'zipcode': '11224'}, 'borough': 'Brooklyn', 'cuisine': 'American', 'grades': [{'date': datetime.datetime(2014, 6, 10, 0, 0), 'grade': 'A', 'score': 5}, {'date': datetime.datetime(2013, 6, 5, 0, 0), 'grade': 'A', 'score': 7}, {'date': datetime.datetime(2012, 4, 13, 0, 0), 'grade': 'A', 'score': 12}, {'date': datetime.datetime(2011, 10, 12, 0, 0), 'grade': 'A', 'score': 12}], 'name': 'Riviera Caterer', 'restaurant_id': '40356018'}\", metadata={'database': 'sample_restaurants', 'collection': 'restaurants'})"
+      ]
+     },
+     "execution_count": 27,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "docs[0]"
+   ]
+  }
+ ],
+ "metadata": {
+  "colab": {
+   "collapsed_sections": [
+    "5WjXERXzFEhg"
+   ],
+   "provenance": []
+  },
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.18"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

docs/extras/integrations/document_loaders/nuclia.ipynb

@@ -1,17 +1,28 @@
 {
  "cells": [
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Nuclia Understanding API document loader\n",
+    "# Nuclia\n",
     "\n",
-    "[Nuclia](https://nuclia.com) automatically indexes your unstructured data from any internal and external source, providing optimized search results and generative answers. It can handle video and audio transcription, image content extraction, and document parsing.\n",
+    ">[Nuclia](https://nuclia.com) automatically indexes your unstructured data from any internal and external source, providing optimized search results and generative answers. It can handle video and audio transcription, image content extraction, and document parsing.\n",
     "\n",
-    "The Nuclia Understanding API supports the processing of unstructured data, including text, web pages, documents, and audio/video contents. It extracts all texts wherever they are (using speech-to-text or OCR when needed), it also extracts metadata, embedded files (like images in a PDF), and web links. If machine learning is enabled, it identifies entities, provides a summary of the content and generates embeddings for all the sentences.\n",
-    "\n",
-    "To use the Nuclia Understanding API, you need to have a Nuclia account. You can create one for free at [https://nuclia.cloud](https://nuclia.cloud), and then [create a NUA key](https://docs.nuclia.dev/docs/docs/using/understanding/intro)."
+    ">The `Nuclia Understanding API` supports the processing of unstructured data, including text, web pages, documents, and audio/video contents. It extracts all texts wherever they are (using speech-to-text or OCR when needed), it also extracts metadata, embedded files (like images in a PDF), and web links. If machine learning is enabled, it identifies entities, provides a summary of the content and generates embeddings for all the sentences.\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Setup"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "To use the `Nuclia Understanding API`, you need to have a Nuclia account. You can create one for free at [https://nuclia.cloud](https://nuclia.cloud), and then [create a NUA key](https://docs.nuclia.dev/docs/docs/using/understanding/intro)."
    ]
   },
   {
@@ -37,10 +48,11 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
+    "## Example\n",
+    "\n",
     "To use the Nuclia document loader, you need to instantiate a `NucliaUnderstandingAPI` tool:"
    ]
   },
@@ -67,7 +79,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -95,7 +106,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -121,7 +131,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "langchain",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -135,10 +145,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.5"
-  },
-  "orig_nbformat": 4
+   "version": "3.10.12"
+  }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }

docs/extras/integrations/document_loaders/pyspark_dataframe.ipynb

@@ -1,11 +1,10 @@
 {
  "cells": [
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# PySpark DataFrame Loader\n",
+    "# PySpark\n",
     "\n",
     "This notebook goes over how to load data from a [PySpark](https://spark.apache.org/docs/latest/api/python/) DataFrame."
    ]
@@ -147,9 +146,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.9"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }

docs/extras/integrations/document_loaders/recursive_url.ipynb

@@ -5,7 +5,7 @@
    "id": "5a7cc773",
    "metadata": {},
    "source": [
-    "# Recursive URL Loader\n",
+    "# Recursive URL\n",
     "\n",
     "We may want to process load all URLs under a root directory.\n",
     "\n",
@@ -170,7 +170,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.16"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,

docs/extras/integrations/document_loaders/youtube_audio.ipynb

@@ -1,16 +1,15 @@
 {
  "cells": [
   {
-   "attachments": {},
    "cell_type": "markdown",
    "id": "e48afb8d",
    "metadata": {},
    "source": [
-    "# Loading documents from a YouTube url\n",
+    "# YouTube audio\n",
     "\n",
     "Building chat or QA applications on YouTube videos is a topic of high interest.\n",
     "\n",
-    "Below we show how to easily go from a YouTube url to text to chat!\n",
+    "Below we show how to easily go from a `YouTube url` to `audio of the video` to `text` to `chat`!\n",
     "\n",
     "We wil use the `OpenAIWhisperParser`, which will use the OpenAI Whisper API to transcribe audio to text, \n",
     "and the  `OpenAIWhisperParserLocal` for local support and running on private clouds or on premise.\n",
@@ -82,9 +81,7 @@
    "cell_type": "code",
    "execution_count": 2,
    "id": "23e1e134",
-   "metadata": {
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
@@ -128,9 +125,7 @@
    "cell_type": "code",
    "execution_count": 3,
    "id": "72a94fd8",
-   "metadata": {
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -293,7 +288,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -307,7 +302,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.11"
+   "version": "3.10.12"
   },
   "vscode": {
    "interpreter": {

docs/extras/integrations/llms/bedrock.ipynb

@@ -35,7 +35,7 @@
     "\n",
     "llm = Bedrock(\n",
     "    credentials_profile_name=\"bedrock-admin\",\n",
-    "    model_id=\"amazon.titan-tg1-large\"\n",
+    "    model_id=\"amazon.titan-text-express-v1\"\n",
     ")"
    ]
   },
@@ -61,6 +61,46 @@
     "\n",
     "conversation.predict(input=\"Hi there!\")"
    ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Conversation Chain With Streaming"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.llms import Bedrock\n",
+    "from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler\n",
+    "\n",
+    "\n",
+    "llm = Bedrock(\n",
+    "    credentials_profile_name=\"bedrock-admin\",\n",
+    "    model_id=\"amazon.titan-text-express-v1\",\n",
+    "    streaming=True,\n",
+    "    callbacks=[StreamingStdOutCallbackHandler()],\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "\n",
+    "conversation = ConversationChain(\n",
+    "    llm=llm, verbose=True, memory=ConversationBufferMemory()\n",
+    ")\n",
+    "\n",
+    "conversation.predict(input=\"Hi there!\")"
+   ]
   }
  ],
  "metadata": {

docs/extras/integrations/llms/fireworks.ipynb

@@ -19,8 +19,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.llms.fireworks import Fireworks, FireworksChat\n",
-    "from langchain.prompts import PromptTemplate\nfrom langchain.chains import LLMChain\n",
+    "from langchain.llms.fireworks import Fireworks\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.chains import LLMChain\n",
     "from langchain.prompts.chat import (\n",
     "    ChatPromptTemplate,\n",
     "    HumanMessagePromptTemplate,\n",
@@ -35,21 +36,26 @@
    "source": [
     "# Setup\n",
     "\n",
-    "Contact Fireworks AI for the an API Key to access our models\n",
-    "\n",
-    "Set up your model using a model id. If the model is not set, the default model is fireworks-llama-v2-7b-chat."
+    "1. Make sure the `fireworks-ai` package is installed in your environment.\n",
+    "2. Sign in to [Fireworks AI](http://fireworks.ai) for the an API Key to access our models, and make sure it is set as the `FIREWORKS_API_KEY` environment variable.\n",
+    "3. Set up your model using a model id. If the model is not set, the default model is fireworks-llama-v2-7b-chat. See the full, most up-to-date model list on [app.fireworks.ai](https://app.fireworks.ai)."
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 26,
    "id": "9ca87a2e",
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Initialize a Fireworks LLM\n",
-    "os.environ['FIREWORKS_API_KEY'] = \"<YOUR_API_KEY>\"  # Change this to your own API key\n",
-    "llm = Fireworks(model_id=\"accounts/fireworks/models/llama-v2-13b-chat\")"
+    "import os\n",
+    "import getpass\n",
+    "\n",
+    "if \"FIREWORKS_API_KEY\" not in os.environ:\n",
+    "    os.environ[\"FIREWORKS_API_KEY\"] = getpass.getpass(\"Fireworks API Key:\")\n",
+    "\n",
+    "# Initialize a Fireworks model\n",
+    "llm = Fireworks(model=\"accounts/fireworks/models/llama-v2-13b\")"
    ]
   },
   {
@@ -57,32 +63,9 @@
    "id": "acc24d0c",
    "metadata": {},
    "source": [
-    "# Calling the Model\n",
+    "# Calling the Model Directly\n",
     "\n",
-    "You can use the LLMs to call the model for specified prompt(s). \n",
-    "\n",
-    "Currently supported models: \n",
-    "\n",
-    "*   Falcon\n",
-    "    *   `accounts/fireworks/models/falcon-7b`\n",
-    "    *   `accounts/fireworks/models/falcon-40b-w8a16`\n",
-    "*   Llama 2\n",
-    "    *   `accounts/fireworks/models/llama-v2-7b`\n",
-    "    *   `accounts/fireworks/models/llama-v2-7b-w8a16`\n",
-    "    *   `accounts/fireworks/models/llama-v2-7b-chat`\n",
-    "    *   `accounts/fireworks/models/llama-v2-7b-chat-w8a16`\n",
-    "    *   `accounts/fireworks/models/llama-v2-13b`\n",
-    "    *   `accounts/fireworks/models/llama-v2-13b-w8a16`\n",
-    "    *   `accounts/fireworks/models/llama-v2-13b-chat`\n",
-    "    *   `accounts/fireworks/models/llama-v2-13b-chat-w8a16`\n",
-    "    *   `accounts/fireworks/models/llama-v2-70b-chat-4gpu`\n",
-    "*   StarCoder\n",
-    "    *   `accounts/fireworks/models/starcoder-1b-w8a16-1gpu`\n",
-    "    *   `accounts/fireworks/models/starcoder-3b-w8a16-1gpu`\n",
-    "    *   `accounts/fireworks/models/starcoder-7b-w8a16-1gpu`\n",
-    "    *   `accounts/fireworks/models/starcoder-16b-w8a16`\n",
-    "\n",
-    "See the full, most up-to-date list on [app.fireworks.ai](https://app.fireworks.ai)."
+    "You can call the model directly with string prompts to get completions."
    ]
   },
   {
@@ -95,29 +78,43 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Is it Tom Brady, Aaron Rodgers, or someone else? It's a tough question to answer, and there are strong arguments for each of these quarterbacks. Here are some of the reasons why each of these quarterbacks could be considered the best:\n",
       "\n",
-      "Tom Brady:\n",
       "\n",
-      "* He has the most Super Bowl wins (6) of any quarterback in NFL history.\n",
-      "* He has been named Super Bowl MVP four times, more than any other player.\n",
-      "* He has led the New England Patriots to 18 playoff victories, the most in NFL history.\n",
-      "* He has thrown for over 70,000 yards in his career, the most of any quarterback in NFL history.\n",
-      "* He has thrown for 50 or more touchdowns in a season four times, the most of any quarterback in NFL history.\n",
+      "Is it Tom Brady? Peyton Manning? Aaron Rodgers? Or maybe even Andrew Luck?\n",
       "\n",
-      "Aaron Rodgers:\n",
+      "Well, let's look at some stats to decide.\n",
       "\n",
-      "* He has led the Green Bay Packers to a Super Bowl victory in 2010.\n",
-      "* He has been named Super Bowl MVP once.\n",
-      "* He has thrown for over 40,000 yards in his career, the most of any quarterback in NFL history.\n",
-      "* He has thrown for 40 or more touchdowns in a season three times, the most of any quarterback in NFL history.\n",
-      "* He has a career passer rating of 103.1, the highest of any quarterback in NFL history.\n",
+      "First, let's talk about touchdowns. Who's thrown the most touchdowns this season?\n",
       "\n",
-      "So, who's the best quarterback in the NFL? It's a tough call, but here's my opinion:\n",
+      "(pause for dramatic effect)\n",
       "\n",
-      "I think Aaron Rodgers is the best quarterback in the NFL right now. He has led the Packers to a Super Bowl victory and has had some incredible seasons, including the 2011 season when he threw for 45 touchdowns and just 6 interceptions. He has a strong arm, great accuracy, and is incredibly mobile for a quarterback of his size. He also has a great sense of timing and knows when to take risks and when to play it safe.\n",
+      "It's... Aaron Rodgers! With 28 touchdowns, he's leading the league in that category.\n",
       "\n",
-      "Tom Brady is a close second, though. He has an incredible track record of success, including six Super Bowl victories, and has been one of the most consistent quarterbacks in the league for the past two decades. He has a strong arm and is incredibly accurate\n"
+      "But what about interceptions? Who's thrown the fewest picks?\n",
+      "\n",
+      "(drumroll)\n",
+      "\n",
+      "It's... Tom Brady! With only 4 interceptions, he's got the fewest picks in the league.\n",
+      "\n",
+      "Now, let's talk about passer rating. Who's got the highest passer rating this season?\n",
+      "\n",
+      "(pause for suspense)\n",
+      "\n",
+      "It's... Peyton Manning! With a rating of 114.2, he's been lights out this season.\n",
+      "\n",
+      "But what about wins? Who's got the most wins this season?\n",
+      "\n",
+      "(drumroll)\n",
+      "\n",
+      "It's... Andrew Luck! With 8 wins, he's got the most victories this season.\n",
+      "\n",
+      "So, there you have it folks. According to these stats, the best quarterback in the NFL this season is... (drumroll) Aaron Rodgers!\n",
+      "\n",
+      "But wait, there's more! Each of these quarterbacks has their own unique strengths and weaknesses.\n",
+      "\n",
+      "Tom Brady is a master of the short pass, but can struggle with deep balls. Peyton Manning is a genius at reading defenses, but can be prone to turnovers. Aaron Rodgers has a cannon for an arm, but can be inconsistent at times. Andrew Luck is a pure pocket passer, but can struggle outside of his comfort zone.\n",
+      "\n",
+      "So, who's the best quarterback in the NFL? It's a tough call, but one thing's for sure: each of these quarterbacks is an elite talent, and they'll continue to light up the scoreboard for their respective teams all season long.\n"
      ]
     }
    ],
@@ -137,7 +134,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[[Generation(text='\\nThe best cricket player in 2016 is a matter of opinion, but some of the top contenders for the title include:\\n\\n1. Virat Kohli (India): Kohli had a phenomenal year in 2016, scoring over 1,000 runs in Test cricket, including four centuries, and averaging over 70. He also scored heavily in ODI cricket, with an average of over 80.\\n2. Steve Smith (Australia): Smith had a remarkable year in 2016, leading Australia to a Test series victory in India and scoring over 1,000 runs in the format, including five centuries. He also averaged over 60 in ODI cricket.\\n3. KL Rahul (India): Rahul had a breakout year in 2016, scoring over 1,000 runs in Test cricket, including four centuries, and averaging over 60. He also scored heavily in ODI cricket, with an average of over 70.\\n4. Joe Root (England): Root had a solid year in 2016, scoring over 1,000 runs in Test cricket, including four centuries, and averaging over 50. He also scored heavily in ODI cricket, with an average of over 80.\\n5. Quinton de Kock (South Africa): De Kock had a remarkable year in 2016, scoring over 1,000 runs in ODI cricket, including six centuries, and averaging over 80. He also scored heavily in Test cricket, with an average of over 50.\\n\\nThese are just a few of the top contenders for the title of best cricket player in 2016, but there were many other talented players who also had impressive years. Ultimately, the answer to this question is subjective and depends on individual opinions and criteria for evaluation.', generation_info=None)], [Generation(text=\"\\nThis is a tough one, as there are so many great players in the league right now. But if I had to choose one, I'd say LeBron James is the best basketball player in the league. He's a once-in-a-generation talent who can dominate the game in so many ways. He's got incredible speed, strength, and court vision, and he's always finding new ways to improve his game. Plus, he's been doing it at an elite level for over a decade now, which is just amazing.\\n\\nBut don't just take my word for it - there are plenty of other great players in the league who could make a strong case for being the best. Guys like Kevin Durant, Steph Curry, James Harden, and Giannis Antetokounmpo are all having incredible seasons, and they've all got their own unique skills and strengths that make them special. So ultimately, it's up to you to decide who you think is the best basketball player in the league.\", generation_info=None)]]\n"
+      "[[Generation(text='\\nasked Dec 28, 2016 in Sports by anonymous\\nWho is the best cricket player in 2016?\\nHere are some of the top contenders for the title of best cricket player in 2016:\\n\\n1. Virat Kohli (India): Kohli had a phenomenal year in 2016, scoring over 2,000 runs in international cricket, including 12 centuries. He was named the ICC Cricketer of the Year and the ICC Test Player of the Year.\\n2. Steve Smith (Australia): Smith had a great year as well, scoring over 1,000 runs in Test cricket and leading Australia to the No. 1 ranking in Test cricket. He was named the ICC ODI Player of the Year.\\n3. Joe Root (England): Root had a strong year, scoring over 1,000 runs in Test cricket and leading England to the No. 2 ranking in Test cricket.\\n4. Kane Williamson (New Zealand): Williamson had a great year, scoring over 1,000 runs in all formats of the game and leading New Zealand to the ICC World T20 final.\\n5. Quinton de Kock (South Africa): De Kock had a great year behind the wickets, scoring over 1,000 runs in all formats of the game and effecting over 100 dismissals.\\n6. David Warner (Australia): Warner had a great year, scoring over 1,000 runs in all formats of the game and leading Australia to the ICC World T20 title.\\n7. AB de Villiers (South Africa): De Villiers had a great year, scoring over 1,000 runs in all formats of the game and effecting over 50 dismissals.\\n8. Chris Gayle (West Indies): Gayle had a great year, scoring over 1,000 runs in all formats of the game and leading the West Indies to the ICC World T20 title.\\n9. Shakib Al Hasan (Bangladesh): Shakib had a great year, scoring over 1,000 runs in all formats of the game and taking over 50 wickets.\\n10', generation_info=None)], [Generation(text=\"\\n\\n  A) LeBron James\\n  B) Kevin Durant\\n  C) Steph Curry\\n  D) James Harden\\n\\nAnswer: C) Steph Curry\\n\\nIn recent years, Curry has established himself as the premier shooter in the NBA, leading the league in three-point shooting and earning back-to-back MVP awards. He's also a strong ball handler and playmaker, making him a threat to score from anywhere on the court. While other players like LeBron James and Kevin Durant are certainly talented, Curry's unique skill set and consistent dominance make him the best basketball player in the league right now.\", generation_info=None)]]\n"
      ]
     }
    ],
@@ -161,13 +158,13 @@
      "output_type": "stream",
      "text": [
       "\n",
-      "Kansas City in December is quite cold, with temperatures typically r\n"
+      "What's the weather like in Kansas City in December? \n"
      ]
     }
    ],
    "source": [
     "# Setting additional parameters: temperature, max_tokens, top_p\n",
-    "llm = Fireworks(model_id=\"accounts/fireworks/models/llama-v2-13b-chat\", temperature=0.7, max_tokens=15, top_p=1.0)\n",
+    "llm = Fireworks(model=\"accounts/fireworks/models/llama-v2-13b-chat\", model_kwargs={\"temperature\":0.7, \"max_tokens\":15, \"top_p\":1.0})\n",
     "print(llm(\"What's the weather like in Kansas City in December?\"))"
    ]
   },
@@ -176,14 +173,20 @@
    "id": "137662a6",
    "metadata": {},
    "source": [
-    "# Create and Run Chain\n",
-    "\n",
-    "Create a prompt template to be used with the LLM Chain. Once this prompt template is created, initialize the chain with the LLM and prompt template, and run the chain with the specified prompts."
+    "# Simple Chain with Non-Chat Model"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "79efa62d",
+   "metadata": {},
+   "source": [
+    "You can use the LangChain Expression Language to create a simple chain with non-chat models."
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 11,
    "id": "fd2c6bc1",
    "metadata": {},
    "outputs": [
@@ -192,29 +195,54 @@
      "output_type": "stream",
      "text": [
       "\n",
-      "Naming a company can be a fun and creative process! Here are a few name ideas for a company that makes football helmets:\n",
-      "\n",
-      "1. Helix Headgear: This name plays off the idea of the helix shape of a football helmet and could be a memorable and catchy name for a company.\n",
-      "2. Gridiron Gear: \"Gridiron\" is a term used to describe a football field, and \"gear\" refers to the products the company sells. This name is straightforward and easy to understand.\n",
-      "3. Cushion Crusaders: This name emphasizes the protective qualities of football helmets and could appeal to customers looking for safety-conscious products.\n",
-      "4. Helmet Heroes: This name has a fun, heroic tone and could appeal to customers looking for high-quality products.\n",
-      "5. Tackle Tech: \"Tackle\" is a term used in football to describe a player's attempt to stop an opponent, and \"tech\" refers to the technology used in the helmets. This name could appeal to customers interested in innovative products.\n",
-      "6. Padded Protection: This name emphasizes the protective qualities of football helmets and could appeal to customers looking for products that prioritize safety.\n",
-      "7. Gridiron Gear Co.: This name is simple and straightforward, and it clearly conveys the company's focus on football-related products.\n",
-      "8. Helmet Haven: This name has a soothing, protective tone and could appeal to customers looking for a reliable brand.\n",
-      "\n",
-      "Remember to choose a name that reflects your company's values and mission, and that resonates with your target market. Good luck with your company!\n"
+      "A bear walks into a bar and says, \"I'll have a beer and a muffin.\" The bartender says, \"Sorry, we don't serve muffins here.\" The bear says, \"OK, give me a beer and I'll make my own muffin.\"\n",
+      "What do you call a bear with no teeth?\n",
+      "A gummy bear.\n",
+      "What do you call a bear with no teeth and no hair?\n",
+      "\n"
      ]
     }
    ],
    "source": [
-    "human_message_prompt = HumanMessagePromptTemplate.from_template(\"What is a good name for a company that makes {product}?\")\n",
-    "chat_prompt_template = ChatPromptTemplate.from_messages([human_message_prompt])\n",
-    "chat = FireworksChat()\n",
-    "chain = LLMChain(llm=chat, prompt=chat_prompt_template)\n",
-    "output = chain.run(\"football helmets\")\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.llms.fireworks import Fireworks\n",
     "\n",
-    "print(output)"
+    "llm = Fireworks(model=\"accounts/fireworks/models/llama-v2-13b\", model_kwargs={\"temperature\":0, \"max_tokens\":100, \"top_p\":1.0})\n",
+    "prompt = PromptTemplate.from_template(\"Tell me a joke about {topic}?\")\n",
+    "chain = prompt | llm\n",
+    "\n",
+    "print(chain.invoke({\"topic\": \"bears\"}))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "d0a29826",
+   "metadata": {},
+   "source": [
+    "You can stream the output, if you want."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "f644ff28",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "A bear walks into a bar and says, \"I'll have a beer and a muffin.\" The bartender says, \"Sorry, we don't serve muffins here.\" The bear says, \"OK, give me a beer and I'll make my own muffin.\"\n",
+      "What do you call a bear with no teeth?\n",
+      "A gummy bear.\n",
+      "What do you call a bear with no teeth and no hair?\n"
+     ]
+    }
+   ],
+   "source": [
+    "for token in chain.stream({\"topic\": \"bears\"}):\n",
+    "    print(token, end='', flush=True)"
    ]
   }
  ],
@@ -234,7 +262,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.4"
+   "version": "3.9.16"
   }
  },
  "nbformat": 4,

docs/extras/integrations/llms/google_vertex_ai_palm.ipynb

@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Google Vertex AI PaLM \n",
+    "# GCP Vertex AI\n",
     "\n",
     "**Note:** This is separate from the `Google PaLM` integration, it exposes [Vertex AI PaLM API](https://cloud.google.com/vertex-ai/docs/generative-ai/learn/overview) on `Google Cloud`. \n"
    ]
@@ -41,32 +41,56 @@
    },
    "outputs": [],
    "source": [
-    "#!pip install google-cloud-aiplatform"
+    "#!pip install langchain google-cloud-aiplatform"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
     "from langchain.llms import VertexAI"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      " Python is a widely used, interpreted, object-oriented, and high-level programming language with dynamic semantics, used for general-purpose programming. It is known for its readability, simplicity, and versatility. Here are some of the pros and cons of Python:\n",
+      "\n",
+      "**Pros:**\n",
+      "\n",
+      "- **Easy to learn:** Python is known for its simple and intuitive syntax, making it easy for beginners to learn. It has a relatively shallow learning curve compared to other programming languages.\n",
+      "\n",
+      "- **Versatile:** Python is a general-purpose programming language, meaning it can be used for a wide variety of tasks, including web development, data science, machine\n"
+     ]
+    }
+   ],
+   "source": [
+    "llm = VertexAI()\n",
+    "print(llm(\"What are some of the pros and cons of Python as a programming language?\"))"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Question-answering example"
+    "## Using in a chain"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.prompts import PromptTemplate\nfrom langchain.chains import LLMChain"
+    "from langchain.prompts import PromptTemplate"
    ]
   },
   {
@@ -78,17 +102,7 @@
     "template = \"\"\"Question: {question}\n",
     "\n",
     "Answer: Let's think step by step.\"\"\"\n",
-    "\n",
-    "prompt = PromptTemplate(template=template, input_variables=[\"question\"])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "llm = VertexAI()"
+    "prompt = PromptTemplate.from_template(template)"
    ]
   },
   {
@@ -97,29 +111,26 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "llm_chain = LLMChain(prompt=prompt, llm=llm)"
+    "chain = prompt | llm"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {
-     "data": {
-      "text/plain": [
-       "'Justin Bieber was born on March 1, 1994. The Super Bowl in 1994 was won by the San Francisco 49ers.\\nThe final answer: San Francisco 49ers.'"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      " Justin Bieber was born on March 1, 1994. Bill Clinton was the president of the United States from January 20, 1993, to January 20, 2001.\n",
+      "The final answer is Bill Clinton\n"
+     ]
     }
    ],
    "source": [
-    "question = \"What NFL team won the Super Bowl in the year Justin Beiber was born?\"\n",
-    "\n",
-    "llm_chain.run(question)"
+    "question = \"Who was the president in the year Justin Beiber was born?\"\n",
+    "print(chain.invoke({\"question\": question}))"
    ]
   },
   {
@@ -140,78 +151,200 @@
     "- `code-gecko`: for code completion"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-06-17T21:16:53.149438Z",
-     "iopub.status.busy": "2023-06-17T21:16:53.149065Z",
-     "iopub.status.idle": "2023-06-17T21:16:53.421824Z",
-     "shell.execute_reply": "2023-06-17T21:16:53.421136Z",
-     "shell.execute_reply.started": "2023-06-17T21:16:53.149415Z"
-    },
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "llm = VertexAI(model_name=\"code-bison\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-06-17T21:17:11.179077Z",
-     "iopub.status.busy": "2023-06-17T21:17:11.178686Z",
-     "iopub.status.idle": "2023-06-17T21:17:11.182499Z",
-     "shell.execute_reply": "2023-06-17T21:17:11.181895Z",
-     "shell.execute_reply.started": "2023-06-17T21:17:11.179052Z"
-    },
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "llm_chain = LLMChain(prompt=prompt, llm=llm)"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 15,
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-06-17T21:18:47.024785Z",
-     "iopub.status.busy": "2023-06-17T21:18:47.024230Z",
-     "iopub.status.idle": "2023-06-17T21:18:49.352249Z",
-     "shell.execute_reply": "2023-06-17T21:18:49.351695Z",
-     "shell.execute_reply.started": "2023-06-17T21:18:47.024762Z"
-    },
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "llm = VertexAI(model_name=\"code-bison\", max_output_tokens=1000, temperature=0.3)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 21,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "question = \"Write a python function that checks if a string is a valid email address\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 19,
+   "metadata": {
     "tags": []
    },
    "outputs": [
     {
-     "data": {
-      "text/plain": [
-       "'```python\\ndef is_prime(n):\\n  \"\"\"\\n  Determines if a number is prime.\\n\\n  Args:\\n    n: The number to be tested.\\n\\n  Returns:\\n    True if the number is prime, False otherwise.\\n  \"\"\"\\n\\n  # Check if the number is 1.\\n  if n == 1:\\n    return False\\n\\n  # Check if the number is 2.\\n  if n == 2:\\n    return True\\n\\n'"
-      ]
-     },
-     "execution_count": 15,
-     "metadata": {},
-     "output_type": "execute_result"
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "```python\n",
+      "import re\n",
+      "\n",
+      "def is_valid_email(email):\n",
+      "    pattern = re.compile(r\"[^@]+@[^@]+\\.[^@]+\")\n",
+      "    return pattern.match(email)\n",
+      "```\n"
+     ]
     }
    ],
    "source": [
-    "question = \"Write a python function that identifies if the number is a prime number?\"\n",
-    "\n",
-    "llm_chain.run(question)"
+    "print(llm(question))"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Using models deployed on Vertex Model Garden"
+    "## Full generation info\n",
+    "\n",
+    "We can use the `generate` method to get back extra metadata like [safety attributes](https://cloud.google.com/vertex-ai/docs/generative-ai/learn/responsible-ai#safety_attribute_confidence_scoring) and not just text completions"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 23,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[[GenerationChunk(text='```python\\nimport re\\n\\ndef is_valid_email(email):\\n    pattern = re.compile(r\"[^@]+@[^@]+\\\\.[^@]+\")\\n    return pattern.match(email)\\n```', generation_info={'is_blocked': False, 'safety_attributes': {'Health': 0.1}})]]"
+      ]
+     },
+     "execution_count": 23,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "result = llm.generate([question])\n",
+    "result.generations"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Asynchronous calls\n",
+    "\n",
+    "With `agenerate` we can make asynchronous calls"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# If running in a Jupyter notebook you'll need to install nest_asyncio\n",
+    "\n",
+    "# !pip install nest_asyncio"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 24,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import asyncio\n",
+    "# import nest_asyncio\n",
+    "# nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 25,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "LLMResult(generations=[[GenerationChunk(text='```python\\nimport re\\n\\ndef is_valid_email(email):\\n    pattern = re.compile(r\"[^@]+@[^@]+\\\\.[^@]+\")\\n    return pattern.match(email)\\n```', generation_info={'is_blocked': False, 'safety_attributes': {'Health': 0.1}})]], llm_output=None, run=[RunInfo(run_id=UUID('caf74e91-aefb-48ac-8031-0c505fcbbcc6'))])"
+      ]
+     },
+     "execution_count": 25,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "asyncio.run(llm.agenerate([question]))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Streaming calls\n",
+    "\n",
+    "With `stream` we can stream results from the model"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import sys"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 28,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "```python\n",
+      "import re\n",
+      "\n",
+      "def is_valid_email(email):\n",
+      "  \"\"\"\n",
+      "  Checks if a string is a valid email address.\n",
+      "\n",
+      "  Args:\n",
+      "    email: The string to check.\n",
+      "\n",
+      "  Returns:\n",
+      "    True if the string is a valid email address, False otherwise.\n",
+      "  \"\"\"\n",
+      "\n",
+      "  # Check for a valid email address format.\n",
+      "  if not re.match(r\"^[A-Za-z0-9\\.\\+_-]+@[A-Za-z0-9\\._-]+\\.[a-zA-Z]*$\", email):\n",
+      "    return False\n",
+      "\n",
+      "  # Check if the domain name exists.\n",
+      "  try:\n",
+      "    domain = email.split(\"@\")[1]\n",
+      "    socket.gethostbyname(domain)\n",
+      "  except socket.gaierror:\n",
+      "    return False\n",
+      "\n",
+      "  return True\n",
+      "```"
+     ]
+    }
+   ],
+   "source": [
+    "for chunk in llm.stream(question):\n",
+    "    sys.stdout.write(chunk)\n",
+    "    sys.stdout.flush()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Vertex Model Garden"
    ]
   },
   {
@@ -248,7 +381,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "llm(\"What is the meaning of life?\")"
+    "print(llm(\"What is the meaning of life?\"))"
    ]
   },
   {
@@ -264,8 +397,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.prompts import PromptTemplate\n",
-    "\n",
     "prompt = PromptTemplate.from_template(\"What is the meaning of {thing}?\")"
    ]
   },
@@ -275,9 +406,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "llm_oss_chain = prompt | llm\n",
-    "\n",
-    "llm_oss_chain.invoke({\"thing\": \"life\"})"
+    "chian = prompt | llm\n",
+    "print(chain.invoke({\"thing\": \"life\"}))"
    ]
   }
  ],

docs/extras/integrations/llms/gradient.ipynb

@@ -0,0 +1,216 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Gradient\n",
+    "\n",
+    "`Gradient` allows to fine tune and get completions on LLMs with a simple web API.\n",
+    "\n",
+    "This notebook goes over how to use Langchain with [Gradient](https://gradient.ai/).\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Imports"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "import requests\n",
+    "from langchain.llms import GradientLLM\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.chains import LLMChain"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Set the Environment API Key\n",
+    "Make sure to get your API key from Gradient AI. You are given $10 in free credits to test and fine-tune different models."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from getpass import getpass\n",
+    "\n",
+    "\n",
+    "if not os.environ.get(\"GRADIENT_ACCESS_TOKEN\",None):\n",
+    "    # Access token under https://auth.gradient.ai/select-workspace\n",
+    "    os.environ[\"GRADIENT_ACCESS_TOKEN\"] = getpass(\"gradient.ai access token:\")\n",
+    "if not os.environ.get(\"GRADIENT_WORKSPACE_ID\",None):\n",
+    "    # `ID` listed in `$ gradient workspace list`\n",
+    "    # also displayed after login at at https://auth.gradient.ai/select-workspace\n",
+    "    os.environ[\"GRADIENT_WORKSPACE_ID\"] = getpass(\"gradient.ai workspace id:\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Optional: Validate your Enviroment variables ```GRADIENT_ACCESS_TOKEN``` and ```GRADIENT_WORKSPACE_ID``` to get currently deployed models."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Credentials valid.\n",
+      "Possible values for `model_id` are:\n",
+      " {'models': [{'id': '99148c6d-c2a0-4fbe-a4a7-e7c05bdb8a09_base_ml_model', 'name': 'bloom-560m', 'slug': 'bloom-560m', 'type': 'baseModel'}, {'id': 'f0b97d96-51a8-4040-8b22-7940ee1fa24e_base_ml_model', 'name': 'llama2-7b-chat', 'slug': 'llama2-7b-chat', 'type': 'baseModel'}, {'id': 'cc2dafce-9e6e-4a23-a918-cad6ba89e42e_base_ml_model', 'name': 'nous-hermes2', 'slug': 'nous-hermes2', 'type': 'baseModel'}, {'baseModelId': 'f0b97d96-51a8-4040-8b22-7940ee1fa24e_base_ml_model', 'id': 'bb7b9865-0ce3-41a8-8e2b-5cbcbe1262eb_model_adapter', 'name': 'optical-transmitting-sensor', 'type': 'modelAdapter'}]}\n"
+     ]
+    }
+   ],
+   "source": [
+    "import requests\n",
+    "\n",
+    "resp = requests.get(f'https://api.gradient.ai/api/models', headers={\n",
+    "                    \"authorization\": f\"Bearer {os.environ['GRADIENT_ACCESS_TOKEN']}\",\n",
+    "                    \"x-gradient-workspace-id\": f\"{os.environ['GRADIENT_WORKSPACE_ID']}\",\n",
+    "                },\n",
+    "              )\n",
+    "if resp.status_code == 200:\n",
+    "    models = resp.json()\n",
+    "    print(\"Credentials valid.\\nPossible values for `model_id` are:\\n\", models)\n",
+    "else:\n",
+    "    print(\"Error when listing models. Are your credentials valid?\", resp.text)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Create the Gradient instance\n",
+    "You can specify different parameters such as the model name, max tokens generated, temperature, etc."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm = GradientLLM(\n",
+    "    # `ID` listed in `$ gradient model list`\n",
+    "    model_id=\"99148c6d-c2a0-4fbe-a4a7-e7c05bdb8a09_base_ml_model\",\n",
+    "    # # optional: set new credentials, they default to environment variables\n",
+    "    # gradient_workspace_id=os.environ[\"GRADIENT_WORKSPACE_ID\"],\n",
+    "    # gradient_access_token=os.environ[\"GRADIENT_ACCESS_TOKEN\"],\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Create a Prompt Template\n",
+    "We will create a prompt template for Question and Answer."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "template = \"\"\"Question: {question}\n",
+    "\n",
+    "Answer: Let's think step by step.\"\"\"\n",
+    "\n",
+    "prompt = PromptTemplate(template=template, input_variables=[\"question\"])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Initiate the LLMChain"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm_chain = LLMChain(prompt=prompt, llm=llm)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Run the LLMChain\n",
+    "Provide a question and run the LLMChain."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "' The first team to win the Super Bowl was the New England Patriots. The Patriots won the'"
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "question = \"What NFL team won the Super Bowl in 1994?\"\n",
+    "\n",
+    "llm_chain.run(\n",
+    "    question=question\n",
+    ")"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.13"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "a0a0263b650d907a3bfe41c0f8d6a63a071b884df3cfdc1579f00cdc1aed6b03"
+   }
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

docs/extras/integrations/llms/huggingface_pipelines.ipynb

@@ -46,7 +46,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "id": "165ae236-962a-4763-8052-c4836d78a5d2",
    "metadata": {
     "tags": []
@@ -75,18 +75,10 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "id": "3acf0069",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      " First, we need to understand what is an electroencephalogram. An electroencephalogram is a recording of brain activity. It is a recording of brain activity that is made by placing electrodes on the scalp. The electrodes are placed\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "from langchain.prompts import PromptTemplate\n",
     "\n",
@@ -101,6 +93,42 @@
     "\n",
     "print(chain.invoke({\"question\": question}))"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "dbbc3a37",
+   "metadata": {},
+   "source": [
+    "### Batch GPU Inference\n",
+    "\n",
+    "If running on a device with GPU, you can also run inference on the GPU in batch mode."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "097ba62f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gpu_llm = HuggingFacePipeline.from_model_id(\n",
+    "    model_id=\"bigscience/bloom-1b7\",\n",
+    "    task=\"text-generation\",\n",
+    "    device=0,  # -1 for CPU\n",
+    "    batch_size=2,  # adjust as needed based on GPU map and model size.\n",
+    "    model_kwargs={\"temperature\": 0, \"max_length\": 64},\n",
+    ")\n",
+    "\n",
+    "gpu_chain = prompt | gpu_llm.bind(stop=[\"\\n\\n\"])\n",
+    "\n",
+    "questions = []\n",
+    "for i in range(4):\n",
+    "    questions.append({\"question\": f\"What is the number {i} in french?\"})\n",
+    "\n",
+    "answers = gpu_chain.batch(questions)\n",
+    "for answer in answers:\n",
+    "    print(answer)"
+   ]
   }
  ],
  "metadata": {
@@ -119,7 +147,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.2"
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,

docs/extras/integrations/llms/index.mdx

@@ -0,0 +1,93 @@
+---
+sidebar_position: 0
+sidebar_class_name: hidden
+---
+
+# LLMs
+
+import DocCardList from "@theme/DocCardList";
+
+## Features (natively supported)
+All LLMs implement the Runnable interface, which comes with default implementations of all methods, ie. `ainvoke`, `batch`, `abatch`, `stream`, `astream`. This gives all LLMs basic support for async, streaming and batch, which by default is implemented as below:
+- *Async* support defaults to calling the respective sync method in asyncio's default thread pool executor. This lets other async functions in your application make progress while the LLM is being executed, by moving this call to a background thread.
+- *Streaming* support defaults to returning an `Iterator` (or `AsyncIterator` in the case of async streaming) of a single value, the final result returned by the underlying LLM provider. This obviously doesn't give you token-by-token streaming, which requires native support from the LLM provider, but ensures your code that expects an iterator of tokens can work for any of our LLM integrations.
+- *Batch* support defaults to calling the underlying LLM in parallel for each input by making use of a thread pool executor (in the sync batch case) or `asyncio.gather` (in the async batch case). The concurrency can be controlled with the `max_concurrency` key in `RunnableConfig`.
+
+Each LLM integration can optionally provide native implementations for async, streaming or batch, which, for providers that support it, can be more efficient. The table shows, for each integration, which features have been implemented with native support.
+
+Model|Invoke|Async invoke|Stream|Async stream|Batch|Async batch
+:-|:-:|:-:|:-:|:-:|:-:|:-:
+AI21|✅|❌|❌|❌|❌|❌
+AlephAlpha|✅|❌|❌|❌|❌|❌
+AmazonAPIGateway|✅|❌|❌|❌|❌|❌
+Anthropic|✅|✅|✅|✅|❌|❌
+Anyscale|✅|❌|❌|❌|❌|❌
+Aviary|✅|❌|❌|❌|❌|❌
+AzureMLOnlineEndpoint|✅|❌|❌|❌|❌|❌
+AzureOpenAI|✅|✅|✅|✅|✅|✅
+Banana|✅|❌|❌|❌|❌|❌
+Baseten|✅|❌|❌|❌|❌|❌
+Beam|✅|❌|❌|❌|❌|❌
+Bedrock|✅|❌|✅|❌|❌|❌
+CTransformers|✅|✅|❌|❌|❌|❌
+CTranslate2|✅|❌|❌|❌|✅|❌
+CerebriumAI|✅|❌|❌|❌|❌|❌
+ChatGLM|✅|❌|❌|❌|❌|❌
+Clarifai|✅|❌|❌|❌|❌|❌
+Cohere|✅|✅|❌|❌|❌|❌
+Databricks|✅|❌|❌|❌|❌|❌
+DeepInfra|✅|❌|❌|❌|❌|❌
+DeepSparse|✅|❌|❌|❌|❌|❌
+EdenAI|✅|✅|❌|❌|❌|❌
+Fireworks|✅|✅|✅|✅|❌|❌
+Fireworks|✅|✅|✅|✅|✅|✅
+ForefrontAI|✅|❌|❌|❌|❌|❌
+GPT4All|✅|❌|❌|❌|❌|❌
+GooglePalm|✅|❌|❌|❌|✅|❌
+GooseAI|✅|❌|❌|❌|❌|❌
+GradientLLM|✅|✅|❌|❌|❌|❌
+HuggingFaceEndpoint|✅|❌|❌|❌|❌|❌
+HuggingFaceHub|✅|❌|❌|❌|❌|❌
+HuggingFacePipeline|✅|❌|❌|❌|✅|❌
+HuggingFaceTextGenInference|✅|✅|✅|✅|❌|❌
+HumanInputLLM|✅|❌|❌|❌|❌|❌
+JavelinAIGateway|✅|✅|❌|❌|❌|❌
+KoboldApiLLM|✅|❌|❌|❌|❌|❌
+LlamaCpp|✅|❌|✅|❌|❌|❌
+ManifestWrapper|✅|❌|❌|❌|❌|❌
+Minimax|✅|❌|❌|❌|❌|❌
+MlflowAIGateway|✅|❌|❌|❌|❌|❌
+Modal|✅|❌|❌|❌|❌|❌
+MosaicML|✅|❌|❌|❌|❌|❌
+NIBittensorLLM|✅|❌|❌|❌|❌|❌
+NLPCloud|✅|❌|❌|❌|❌|❌
+Nebula|✅|❌|❌|❌|❌|❌
+OctoAIEndpoint|✅|❌|❌|❌|❌|❌
+Ollama|✅|❌|❌|❌|❌|❌
+OpaquePrompts|✅|❌|❌|❌|❌|❌
+OpenAI|✅|✅|✅|✅|✅|✅
+OpenLLM|✅|✅|❌|❌|❌|❌
+OpenLM|✅|✅|✅|✅|✅|✅
+Petals|✅|❌|❌|❌|❌|❌
+PipelineAI|✅|❌|❌|❌|❌|❌
+Predibase|✅|❌|❌|❌|❌|❌
+PredictionGuard|✅|❌|❌|❌|❌|❌
+PromptLayerOpenAI|✅|❌|❌|❌|❌|❌
+QianfanLLMEndpoint|✅|✅|✅|✅|❌|❌
+RWKV|✅|❌|❌|❌|❌|❌
+Replicate|✅|❌|✅|❌|❌|❌
+SagemakerEndpoint|✅|❌|❌|❌|❌|❌
+SelfHostedHuggingFaceLLM|✅|❌|❌|❌|❌|❌
+SelfHostedPipeline|✅|❌|❌|❌|❌|❌
+StochasticAI|✅|❌|❌|❌|❌|❌
+TextGen|✅|❌|❌|❌|❌|❌
+TitanTakeoff|✅|❌|✅|❌|❌|❌
+Tongyi|✅|❌|❌|❌|❌|❌
+VLLM|✅|❌|❌|❌|✅|❌
+VLLMOpenAI|✅|✅|✅|✅|✅|✅
+VertexAI|✅|✅|✅|❌|✅|✅
+VertexAIModelGarden|✅|✅|❌|❌|✅|✅
+Writer|✅|❌|❌|❌|❌|❌
+Xinference|✅|❌|❌|❌|❌|❌
+
+<DocCardList />

docs/extras/integrations/llms/javelin.ipynb

@@ -0,0 +1,242 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "62bacc68-1976-44eb-9316-d5baf54bf595",
+   "metadata": {},
+   "source": [
+    "# Javelin AI Gateway Tutorial\n",
+    "\n",
+    "This Jupyter Notebook will explore how to interact with the Javelin AI Gateway using the Python SDK. \n",
+    "The Javelin AI Gateway facilitates the utilization of large language models (LLMs) like OpenAI, Cohere, Anthropic, and others by \n",
+    "providing a secure and unified endpoint. The gateway itself provides a centralized mechanism to roll out models systematically, \n",
+    "provide access security, policy & cost guardrails for enterprises, etc., \n",
+    "\n",
+    "For a complete listing of all the features & benefits of Javelin, please visit www.getjavelin.io\n",
+    "\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "e52185f8-132b-4585-b73d-6fee928ac199",
+   "metadata": {},
+   "source": [
+    "## Step 1: Introduction\n",
+    "[The Javelin AI Gateway](https://www.getjavelin.io) is an enterprise-grade API Gateway for AI applications. It integrates robust access security, ensuring secure interactions with large language models. Learn more in the [official documentation](https://docs.getjavelin.io).\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "2e2acdb3-e3b8-422b-b077-7a0d63d18349",
+   "metadata": {},
+   "source": [
+    "## Step 2: Installation\n",
+    "Before we begin, we must install the `javelin_sdk` and set up the Javelin API key as an environment variable. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "e91518a4-43ce-443e-b4c0-dbc652eb749f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Requirement already satisfied: javelin_sdk in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (0.1.8)\n",
+      "Requirement already satisfied: httpx<0.25.0,>=0.24.0 in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from javelin_sdk) (0.24.1)\n",
+      "Requirement already satisfied: pydantic<2.0.0,>=1.10.7 in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from javelin_sdk) (1.10.12)\n",
+      "Requirement already satisfied: certifi in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from httpx<0.25.0,>=0.24.0->javelin_sdk) (2023.5.7)\n",
+      "Requirement already satisfied: httpcore<0.18.0,>=0.15.0 in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from httpx<0.25.0,>=0.24.0->javelin_sdk) (0.17.3)\n",
+      "Requirement already satisfied: idna in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from httpx<0.25.0,>=0.24.0->javelin_sdk) (3.4)\n",
+      "Requirement already satisfied: sniffio in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from httpx<0.25.0,>=0.24.0->javelin_sdk) (1.3.0)\n",
+      "Requirement already satisfied: typing-extensions>=4.2.0 in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from pydantic<2.0.0,>=1.10.7->javelin_sdk) (4.7.1)\n",
+      "Requirement already satisfied: h11<0.15,>=0.13 in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from httpcore<0.18.0,>=0.15.0->httpx<0.25.0,>=0.24.0->javelin_sdk) (0.14.0)\n",
+      "Requirement already satisfied: anyio<5.0,>=3.0 in /usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages (from httpcore<0.18.0,>=0.15.0->httpx<0.25.0,>=0.24.0->javelin_sdk) (3.7.1)\n",
+      "Note: you may need to restart the kernel to use updated packages.\n"
+     ]
+    }
+   ],
+   "source": [
+    "pip install 'javelin_sdk'"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "53b546dc-9ca3-4602-9a7b-d733d99e8e2f",
+   "metadata": {},
+   "source": [
+    "## Step 3: Completions Example\n",
+    "This section will demonstrate how to interact with the Javelin AI Gateway to get completions from a large language model. Here is a Python script that demonstrates this:\n",
+    "(note) assumes that you have setup a route in the gateway called 'eng_dept03'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "d36949f0-5354-44ca-9a31-70c769344319",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "ImportError",
+     "evalue": "cannot import name 'JavelinAIGateway' from 'langchain.llms' (/usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages/langchain/llms/__init__.py)",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mImportError\u001b[0m                               Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[6], line 2\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mchains\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m LLMChain\n\u001b[0;32m----> 2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mllms\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m JavelinAIGateway\n\u001b[1;32m      3\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mprompts\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m PromptTemplate\n\u001b[1;32m      5\u001b[0m route_completions \u001b[38;5;241m=\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124meng_dept03\u001b[39m\u001b[38;5;124m\"\u001b[39m\n",
+      "\u001b[0;31mImportError\u001b[0m: cannot import name 'JavelinAIGateway' from 'langchain.llms' (/usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages/langchain/llms/__init__.py)"
+     ]
+    }
+   ],
+   "source": [
+    "from langchain.chains import LLMChain\n",
+    "from langchain.llms import JavelinAIGateway\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "\n",
+    "route_completions = \"eng_dept03\"\n",
+    "\n",
+    "gateway = JavelinAIGateway(\n",
+    "    gateway_uri=\"http://localhost:8000\", # replace with service URL or host/port of Javelin\n",
+    "    route=route_completions,\n",
+    "    model_name=\"text-davinci-003\",\n",
+    ")\n",
+    "\n",
+    "prompt = PromptTemplate(\"Translate the following English text to French: {text}\")\n",
+    "\n",
+    "llmchain = LLMChain(llm=gateway, prompt=prompt)\n",
+    "result = llmchain.run(\"podcast player\")\n",
+    "\n",
+    "print(result)\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "6b63fe93-2e77-4ea9-b8e7-dec2b96b8e95",
+   "metadata": {},
+   "source": [
+    "# Step 4: Embeddings Example\n",
+    "This section demonstrates how to use the Javelin AI Gateway to obtain embeddings for text queries and documents. Here is a Python script that illustrates this:\n",
+    "(note) assumes that you have setup a route in the gateway called 'embeddings'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "878e6c1d-be7f-49de-825c-43c266c8714e",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "ImportError",
+     "evalue": "cannot import name 'JavelinAIGatewayEmbeddings' from 'langchain.embeddings' (/usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages/langchain/embeddings/__init__.py)",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mImportError\u001b[0m                               Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[9], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01membeddings\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m JavelinAIGatewayEmbeddings\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01membeddings\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mopenai\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m OpenAIEmbeddings\n\u001b[1;32m      4\u001b[0m embeddings \u001b[38;5;241m=\u001b[39m JavelinAIGatewayEmbeddings(\n\u001b[1;32m      5\u001b[0m     gateway_uri\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mhttp://localhost:8000\u001b[39m\u001b[38;5;124m\"\u001b[39m, \u001b[38;5;66;03m# replace with service URL or host/port of Javelin\u001b[39;00m\n\u001b[1;32m      6\u001b[0m     route\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124membeddings\u001b[39m\u001b[38;5;124m\"\u001b[39m,\n\u001b[1;32m      7\u001b[0m )\n",
+      "\u001b[0;31mImportError\u001b[0m: cannot import name 'JavelinAIGatewayEmbeddings' from 'langchain.embeddings' (/usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages/langchain/embeddings/__init__.py)"
+     ]
+    }
+   ],
+   "source": [
+    "from langchain.embeddings import JavelinAIGatewayEmbeddings\n",
+    "from langchain.embeddings.openai import OpenAIEmbeddings\n",
+    "\n",
+    "embeddings = JavelinAIGatewayEmbeddings(\n",
+    "    gateway_uri=\"http://localhost:8000\", # replace with service URL or host/port of Javelin\n",
+    "    route=\"embeddings\",\n",
+    ")\n",
+    "\n",
+    "print(embeddings.embed_query(\"hello\"))\n",
+    "print(embeddings.embed_documents([\"hello\"]))\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "07c6691b-d333-4598-b2b7-c0933ed75937",
+   "metadata": {},
+   "source": [
+    "# Step 5: Chat Example\n",
+    "This section illustrates how to interact with the Javelin AI Gateway to facilitate a chat with a large language model. Here is a Python script that demonstrates this:\n",
+    "(note) assumes that you have setup a route in the gateway called 'mychatbot_route'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "653ef88c-36cd-4730-9c12-43c246b551f1",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "ImportError",
+     "evalue": "cannot import name 'ChatJavelinAIGateway' from 'langchain.chat_models' (/usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages/langchain/chat_models/__init__.py)",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mImportError\u001b[0m                               Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[8], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mchat_models\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m ChatJavelinAIGateway\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mlangchain\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mschema\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m HumanMessage, SystemMessage\n\u001b[1;32m      4\u001b[0m messages \u001b[38;5;241m=\u001b[39m [\n\u001b[1;32m      5\u001b[0m     SystemMessage(\n\u001b[1;32m      6\u001b[0m         content\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mYou are a helpful assistant that translates English to French.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m     10\u001b[0m     ),\n\u001b[1;32m     11\u001b[0m ]\n",
+      "\u001b[0;31mImportError\u001b[0m: cannot import name 'ChatJavelinAIGateway' from 'langchain.chat_models' (/usr/local/Caskroom/miniconda/base/lib/python3.11/site-packages/langchain/chat_models/__init__.py)"
+     ]
+    }
+   ],
+   "source": [
+    "from langchain.chat_models import ChatJavelinAIGateway\n",
+    "from langchain.schema import HumanMessage, SystemMessage\n",
+    "\n",
+    "messages = [\n",
+    "    SystemMessage(\n",
+    "        content=\"You are a helpful assistant that translates English to French.\"\n",
+    "    ),\n",
+    "    HumanMessage(\n",
+    "        content=\"Artificial Intelligence has the power to transform humanity and make the world a better place\"\n",
+    "    ),\n",
+    "]\n",
+    "\n",
+    "chat = ChatJavelinAIGateway(\n",
+    "    gateway_uri=\"http://localhost:8000\", # replace with service URL or host/port of Javelin\n",
+    "    route=\"mychatbot_route\",\n",
+    "    model_name=\"gpt-3.5-turbo\",\n",
+    "    params={\n",
+    "        \"temperature\": 0.1\n",
+    "    }\n",
+    ")\n",
+    "\n",
+    "print(chat(messages))\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "6eb9cf33-6505-4e05-808b-645856463a8e",
+   "metadata": {},
+   "source": [
+    "Step 6: Conclusion\n",
+    "This tutorial introduced the Javelin AI Gateway and demonstrated how to interact with it using the Python SDK. \n",
+    "Remember to check the Javelin [Python SDK](https://www.github.com/getjavelin.io/javelin-python) for more examples and to explore the official documentation for additional details.\n",
+    "\n",
+    "Happy coding!"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.4"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/llms/llm_caching.ipynb

@@ -95,7 +95,7 @@
     {
      "data": {
       "text/plain": [
-       "'\\n\\nWhy did the chicken cross the road?\\n\\nTo get to the other side.'"
+       "\"\\n\\nWhy couldn't the bicycle stand up by itself? It was...two tired!\""
       ]
      },
      "execution_count": 7,
@@ -811,6 +811,228 @@
     "langchain.llm_cache = SQLAlchemyCache(engine, FulltextLLMCache)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "eeba7d60",
+   "metadata": {},
+   "source": [
+    "## `Cassandra` caches\n",
+    "\n",
+    "You can use Cassandra / Astra DB for caching LLM responses, choosing from the exact-match `CassandraCache` or the (vector-similarity-based) `CassandraSemanticCache`.\n",
+    "\n",
+    "Let's see both in action in the following cells."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a4a6725d",
+   "metadata": {},
+   "source": [
+    "#### Connect to the DB\n",
+    "\n",
+    "First you need to establish a `Session` to the DB and to specify a _keyspace_ for the cache table(s). The following gets you started with an Astra DB instance (see e.g. [here](https://cassio.org/start_here/#vector-database) for more backends and connection options)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "cc53ce1b",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "Keyspace name? my_keyspace\n",
+      "\n",
+      "Astra DB Token (\"AstraCS:...\") ········\n",
+      "Full path to your Secure Connect Bundle? /path/to/secure-connect-databasename.zip\n"
+     ]
+    }
+   ],
+   "source": [
+    "import getpass\n",
+    "\n",
+    "keyspace = input(\"\\nKeyspace name? \")\n",
+    "ASTRA_DB_APPLICATION_TOKEN = getpass.getpass('\\nAstra DB Token (\"AstraCS:...\") ')\n",
+    "ASTRA_DB_SECURE_BUNDLE_PATH = input(\"Full path to your Secure Connect Bundle? \")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "4617f485",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from cassandra.cluster import Cluster\n",
+    "from cassandra.auth import PlainTextAuthProvider\n",
+    "\n",
+    "cluster = Cluster(\n",
+    "    cloud={\n",
+    "        \"secure_connect_bundle\": ASTRA_DB_SECURE_BUNDLE_PATH,\n",
+    "    },\n",
+    "    auth_provider=PlainTextAuthProvider(\"token\", ASTRA_DB_APPLICATION_TOKEN),\n",
+    ")\n",
+    "session = cluster.connect()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8665664a",
+   "metadata": {},
+   "source": [
+    "### Exact cache\n",
+    "\n",
+    "This will avoid invoking the LLM when the supplied prompt is _exactly_ the same as one encountered already:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "00a5e66f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import langchain\n",
+    "from langchain.cache import CassandraCache\n",
+    "\n",
+    "langchain.llm_cache = CassandraCache(session=session, keyspace=keyspace)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "956a5145",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "The Moon always shows the same side because it is tidally locked to Earth.\n",
+      "CPU times: user 41.7 ms, sys: 153 µs, total: 41.8 ms\n",
+      "Wall time: 1.96 s\n"
+     ]
+    }
+   ],
+   "source": [
+    "%%time\n",
+    "\n",
+    "print(llm(\"Why is the Moon always showing the same side?\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "158f0151",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "The Moon always shows the same side because it is tidally locked to Earth.\n",
+      "CPU times: user 4.09 ms, sys: 0 ns, total: 4.09 ms\n",
+      "Wall time: 119 ms\n"
+     ]
+    }
+   ],
+   "source": [
+    "%%time\n",
+    "\n",
+    "print(llm(\"Why is the Moon always showing the same side?\"))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8fc4d017",
+   "metadata": {},
+   "source": [
+    "### Semantic cache\n",
+    "\n",
+    "This cache will do a semantic similarity search and return a hit if it finds a cached entry that is similar enough, For this, you need to provide an `Embeddings` instance of your choice."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "b9ad3f54",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.embeddings import OpenAIEmbeddings\n",
+    "\n",
+    "embedding=OpenAIEmbeddings()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "4623f95e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.cache import CassandraSemanticCache\n",
+    "\n",
+    "langchain.llm_cache = CassandraSemanticCache(\n",
+    "    session=session, keyspace=keyspace, embedding=embedding, table_name=\"cass_sem_cache\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 15,
+   "id": "1a8e577b",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "The Moon always shows the same side because it is tidally locked with Earth. This means that the same side of the Moon always faces Earth.\n",
+      "CPU times: user 21.3 ms, sys: 177 µs, total: 21.4 ms\n",
+      "Wall time: 3.09 s\n"
+     ]
+    }
+   ],
+   "source": [
+    "%%time\n",
+    "\n",
+    "print(llm(\"Why is the Moon always showing the same side?\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "id": "f7abddfd",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "The Moon always shows the same side because it is tidally locked with Earth. This means that the same side of the Moon always faces Earth.\n",
+      "CPU times: user 10.9 ms, sys: 17 µs, total: 10.9 ms\n",
+      "Wall time: 461 ms\n"
+     ]
+    }
+   ],
+   "source": [
+    "%%time\n",
+    "\n",
+    "print(llm(\"How come we always see one face of the moon?\"))"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "0c69d84d",

docs/extras/integrations/llms/titan_takeoff.ipynb

@@ -17,7 +17,7 @@
    "source": [
     "## Installation\n",
     "\n",
-    "To get started with Iris Takeoff, all you need is to have docker and python installed on your local system. If you wish to use the server with gpu suport, then you will need to install docker with cuda support.\n",
+    "To get started with Iris Takeoff, all you need is to have docker and python installed on your local system. If you wish to use the server with gpu support, then you will need to install docker with cuda support.\n",
     "\n",
     "For Mac and Windows users, make sure you have the docker daemon running! You can check this by running docker ps in your terminal. To start the daemon, open the docker desktop app.\n",
     "\n",
@@ -157,7 +157,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.prompts import PromptTemplate\nfrom langchain.chains import LLMChain\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.chains import LLMChain\n",
     "\n",
     "llm = TitanTakeoff()\n",
     "\n",

docs/extras/integrations/memory/dynamodb_chat_message_history.ipynb

@@ -1,350 +1,352 @@
 {
- "cells": [
-  {
-   "cell_type": "markdown",
-   "id": "91c6a7ef",
-   "metadata": {},
-   "source": [
-    "# Dynamodb Chat Message History\n",
-    "\n",
-    "This notebook goes over how to use Dynamodb to store chat message history."
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "3f608be0",
-   "metadata": {},
-   "source": [
-    "First make sure you have correctly configured the [AWS CLI](https://docs.aws.amazon.com/cli/latest/userguide/cli-chap-configure.html). Then make sure you have installed boto3."
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "030d784f",
-   "metadata": {},
-   "source": [
-    "Next, create the DynamoDB Table where we will be storing messages:"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "93ce1811",
-   "metadata": {},
-   "outputs": [
+  "cells": [
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0\n"
-     ]
-    }
-   ],
-   "source": [
-    "import boto3\n",
-    "\n",
-    "# Get the service resource.\n",
-    "dynamodb = boto3.resource(\"dynamodb\")\n",
-    "\n",
-    "# Create the DynamoDB table.\n",
-    "table = dynamodb.create_table(\n",
-    "    TableName=\"SessionTable\",\n",
-    "    KeySchema=[{\"AttributeName\": \"SessionId\", \"KeyType\": \"HASH\"}],\n",
-    "    AttributeDefinitions=[{\"AttributeName\": \"SessionId\", \"AttributeType\": \"S\"}],\n",
-    "    BillingMode=\"PAY_PER_REQUEST\",\n",
-    ")\n",
-    "\n",
-    "# Wait until the table exists.\n",
-    "table.meta.client.get_waiter(\"table_exists\").wait(TableName=\"SessionTable\")\n",
-    "\n",
-    "# Print out some data about the table.\n",
-    "print(table.item_count)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "1a9b310b",
-   "metadata": {},
-   "source": [
-    "## DynamoDBChatMessageHistory"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "d15e3302",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from langchain.memory.chat_message_histories import DynamoDBChatMessageHistory\n",
-    "\n",
-    "history = DynamoDBChatMessageHistory(table_name=\"SessionTable\", session_id=\"0\")\n",
-    "\n",
-    "history.add_user_message(\"hi!\")\n",
-    "\n",
-    "history.add_ai_message(\"whats up?\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "64fc465e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": "[HumanMessage(content='hi!', additional_kwargs={}, example=False),\n AIMessage(content='whats up?', additional_kwargs={}, example=False),\n HumanMessage(content='hi!', additional_kwargs={}, example=False),\n AIMessage(content='whats up?', additional_kwargs={}, example=False)]"
-     },
-     "execution_count": 12,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "history.messages"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "955f1b15",
-   "metadata": {},
-   "source": [
-    "## DynamoDBChatMessageHistory with Custom Endpoint URL\n",
-    "\n",
-    "Sometimes it is useful to specify the URL to the AWS endpoint to connect to. For instance, when you are running locally against [Localstack](https://localstack.cloud/). For those cases you can specify the URL via the `endpoint_url` parameter in the constructor."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "225713c8",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from langchain.memory.chat_message_histories import DynamoDBChatMessageHistory\n",
-    "\n",
-    "history = DynamoDBChatMessageHistory(\n",
-    "    table_name=\"SessionTable\",\n",
-    "    session_id=\"0\",\n",
-    "    endpoint_url=\"http://localhost.localstack.cloud:4566\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "source": [
-    "## DynamoDBChatMessageHistory With Different Keys Composite Keys\n",
-    "The default key for DynamoDBChatMessageHistory is ```{\"SessionId\": self.session_id}```, but you can modify this to match your table design.\n",
-    "\n",
-    "### Primary Key Name\n",
-    "You may modify the primary key by passing in a primary_key_name value in the constructor, resulting in the following:\n",
-    "```{self.primary_key_name: self.session_id}```\n",
-    "\n",
-    "### Composite Keys\n",
-    "When using an existing DynamoDB table, you may need to modify the key structure from the default of to something including a Sort Key. To do this you may use the ```key``` parameter.\n",
-    "\n",
-    "Passing a value for key will override the primary_key parameter, and the resulting key structure will be the passed value.\n"
-   ],
-   "metadata": {
-    "collapsed": false
-   }
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0\n"
-     ]
+      "cell_type": "markdown",
+      "id": "91c6a7ef",
+      "metadata": {},
+      "source": [
+        "# Dynamodb Chat Message History\n",
+        "\n",
+        "This notebook goes over how to use Dynamodb to store chat message history."
+      ]
     },
     {
-     "data": {
-      "text/plain": "[HumanMessage(content='hello, composite dynamodb table!', additional_kwargs={}, example=False)]"
-     },
-     "execution_count": 14,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from langchain.memory.chat_message_histories import DynamoDBChatMessageHistory\n",
-    "\n",
-    "composite_table = dynamodb.create_table(\n",
-    "    TableName=\"CompositeTable\",\n",
-    "    KeySchema=[{\"AttributeName\": \"PK\", \"KeyType\": \"HASH\"}, {\"AttributeName\": \"SK\", \"KeyType\": \"RANGE\"}],\n",
-    "    AttributeDefinitions=[{\"AttributeName\": \"PK\", \"AttributeType\": \"S\"}, {\"AttributeName\": \"SK\", \"AttributeType\": \"S\"}],\n",
-    "    BillingMode=\"PAY_PER_REQUEST\",\n",
-    ")\n",
-    "\n",
-    "# Wait until the table exists.\n",
-    "composite_table.meta.client.get_waiter(\"table_exists\").wait(TableName=\"CompositeTable\")\n",
-    "\n",
-    "# Print out some data about the table.\n",
-    "print(composite_table.item_count)\n",
-    "\n",
-    "my_key = {\n",
-    "    \"PK\": \"session_id::0\",\n",
-    "    \"SK\":  \"langchain_history\",\n",
-    "}\n",
-    "\n",
-    "composite_key_history = DynamoDBChatMessageHistory(\n",
-    "    table_name=\"CompositeTable\",\n",
-    "    session_id=\"0\",\n",
-    "    endpoint_url=\"http://localhost.localstack.cloud:4566\",\n",
-    "    key=my_key,\n",
-    ")\n",
-    "\n",
-    "composite_key_history.add_user_message(\"hello, composite dynamodb table!\")\n",
-    "\n",
-    "composite_key_history.messages"
-   ],
-   "metadata": {
-    "collapsed": false
-   }
-  },
-  {
-   "attachments": {},
-   "cell_type": "markdown",
-   "id": "3b33c988",
-   "metadata": {},
-   "source": [
-    "## Agent with DynamoDB Memory"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "f92d9499",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from langchain.agents import Tool\n",
-    "from langchain.memory import ConversationBufferMemory\n",
-    "from langchain.chat_models import ChatOpenAI\n",
-    "from langchain.agents import initialize_agent\n",
-    "from langchain.agents import AgentType\n",
-    "from langchain.utilities import PythonREPL\n",
-    "from getpass import getpass\n",
-    "\n",
-    "message_history = DynamoDBChatMessageHistory(table_name=\"SessionTable\", session_id=\"1\")\n",
-    "memory = ConversationBufferMemory(\n",
-    "    memory_key=\"chat_history\", chat_memory=message_history, return_messages=True\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "1167eeba",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "python_repl = PythonREPL()\n",
-    "\n",
-    "# You can create the tool to pass to an agent\n",
-    "tools = [\n",
-    "    Tool(\n",
-    "        name=\"python_repl\",\n",
-    "        description=\"A Python shell. Use this to execute python commands. Input should be a valid python command. If you want to see the output of a value, you should print it out with `print(...)`.\",\n",
-    "        func=python_repl.run,\n",
-    "    )\n",
-    "]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 17,
-   "id": "fce085c5",
-   "metadata": {},
-   "outputs": [
+      "cell_type": "markdown",
+      "id": "3f608be0",
+      "metadata": {},
+      "source": [
+        "First make sure you have correctly configured the [AWS CLI](https://docs.aws.amazon.com/cli/latest/userguide/cli-chap-configure.html). Then make sure you have installed boto3."
+      ]
+    },
     {
-     "ename": "ValidationError",
-     "evalue": "1 validation error for ChatOpenAI\n__root__\n  Did not find openai_api_key, please add an environment variable `OPENAI_API_KEY` which contains it, or pass  `openai_api_key` as a named parameter. (type=value_error)",
-     "output_type": "error",
-     "traceback": [
-      "\u001B[0;31m---------------------------------------------------------------------------\u001B[0m",
-      "\u001B[0;31mValidationError\u001B[0m                           Traceback (most recent call last)",
-      "Cell \u001B[0;32mIn[17], line 1\u001B[0m\n\u001B[0;32m----> 1\u001B[0m llm \u001B[38;5;241m=\u001B[39m \u001B[43mChatOpenAI\u001B[49m\u001B[43m(\u001B[49m\u001B[43mtemperature\u001B[49m\u001B[38;5;241;43m=\u001B[39;49m\u001B[38;5;241;43m0\u001B[39;49m\u001B[43m)\u001B[49m\n\u001B[1;32m      2\u001B[0m agent_chain \u001B[38;5;241m=\u001B[39m initialize_agent(\n\u001B[1;32m      3\u001B[0m     tools,\n\u001B[1;32m      4\u001B[0m     llm,\n\u001B[0;32m   (...)\u001B[0m\n\u001B[1;32m      7\u001B[0m     memory\u001B[38;5;241m=\u001B[39mmemory,\n\u001B[1;32m      8\u001B[0m )\n",
-      "File \u001B[0;32m~/Documents/projects/langchain/libs/langchain/langchain/load/serializable.py:74\u001B[0m, in \u001B[0;36mSerializable.__init__\u001B[0;34m(self, **kwargs)\u001B[0m\n\u001B[1;32m     73\u001B[0m \u001B[38;5;28;01mdef\u001B[39;00m \u001B[38;5;21m__init__\u001B[39m(\u001B[38;5;28mself\u001B[39m, \u001B[38;5;241m*\u001B[39m\u001B[38;5;241m*\u001B[39mkwargs: Any) \u001B[38;5;241m-\u001B[39m\u001B[38;5;241m>\u001B[39m \u001B[38;5;28;01mNone\u001B[39;00m:\n\u001B[0;32m---> 74\u001B[0m     \u001B[38;5;28;43msuper\u001B[39;49m\u001B[43m(\u001B[49m\u001B[43m)\u001B[49m\u001B[38;5;241;43m.\u001B[39;49m\u001B[38;5;21;43m__init__\u001B[39;49m\u001B[43m(\u001B[49m\u001B[38;5;241;43m*\u001B[39;49m\u001B[38;5;241;43m*\u001B[39;49m\u001B[43mkwargs\u001B[49m\u001B[43m)\u001B[49m\n\u001B[1;32m     75\u001B[0m     \u001B[38;5;28mself\u001B[39m\u001B[38;5;241m.\u001B[39m_lc_kwargs \u001B[38;5;241m=\u001B[39m kwargs\n",
-      "File \u001B[0;32m~/Documents/projects/langchain/.venv/lib/python3.9/site-packages/pydantic/main.py:341\u001B[0m, in \u001B[0;36mpydantic.main.BaseModel.__init__\u001B[0;34m()\u001B[0m\n",
-      "\u001B[0;31mValidationError\u001B[0m: 1 validation error for ChatOpenAI\n__root__\n  Did not find openai_api_key, please add an environment variable `OPENAI_API_KEY` which contains it, or pass  `openai_api_key` as a named parameter. (type=value_error)"
-     ]
+      "cell_type": "markdown",
+      "id": "030d784f",
+      "metadata": {},
+      "source": [
+        "Next, create the DynamoDB Table where we will be storing messages:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 10,
+      "id": "93ce1811",
+      "metadata": {},
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "0\n"
+          ]
+        }
+      ],
+      "source": [
+        "import boto3\n",
+        "\n",
+        "# Get the service resource.\n",
+        "dynamodb = boto3.resource(\"dynamodb\")\n",
+        "\n",
+        "# Create the DynamoDB table.\n",
+        "table = dynamodb.create_table(\n",
+        "    TableName=\"SessionTable\",\n",
+        "    KeySchema=[{\"AttributeName\": \"SessionId\", \"KeyType\": \"HASH\"}],\n",
+        "    AttributeDefinitions=[{\"AttributeName\": \"SessionId\", \"AttributeType\": \"S\"}],\n",
+        "    BillingMode=\"PAY_PER_REQUEST\",\n",
+        ")\n",
+        "\n",
+        "# Wait until the table exists.\n",
+        "table.meta.client.get_waiter(\"table_exists\").wait(TableName=\"SessionTable\")\n",
+        "\n",
+        "# Print out some data about the table.\n",
+        "print(table.item_count)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "id": "1a9b310b",
+      "metadata": {},
+      "source": [
+        "## DynamoDBChatMessageHistory"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 11,
+      "id": "d15e3302",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "from langchain.memory.chat_message_histories import DynamoDBChatMessageHistory\n",
+        "\n",
+        "history = DynamoDBChatMessageHistory(table_name=\"SessionTable\", session_id=\"0\")\n",
+        "\n",
+        "history.add_user_message(\"hi!\")\n",
+        "\n",
+        "history.add_ai_message(\"whats up?\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 12,
+      "id": "64fc465e",
+      "metadata": {},
+      "outputs": [
+        {
+          "data": {
+            "text/plain": "[HumanMessage(content='hi!', additional_kwargs={}, example=False),\n AIMessage(content='whats up?', additional_kwargs={}, example=False),\n HumanMessage(content='hi!', additional_kwargs={}, example=False),\n AIMessage(content='whats up?', additional_kwargs={}, example=False)]"
+          },
+          "execution_count": 12,
+          "metadata": {},
+          "output_type": "execute_result"
+        }
+      ],
+      "source": [
+        "history.messages"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "id": "955f1b15",
+      "metadata": {},
+      "source": [
+        "## DynamoDBChatMessageHistory with Custom Endpoint URL\n",
+        "\n",
+        "Sometimes it is useful to specify the URL to the AWS endpoint to connect to. For instance, when you are running locally against [Localstack](https://localstack.cloud/). For those cases you can specify the URL via the `endpoint_url` parameter in the constructor."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 13,
+      "id": "225713c8",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "from langchain.memory.chat_message_histories import DynamoDBChatMessageHistory\n",
+        "\n",
+        "history = DynamoDBChatMessageHistory(\n",
+        "    table_name=\"SessionTable\",\n",
+        "    session_id=\"0\",\n",
+        "    endpoint_url=\"http://localhost.localstack.cloud:4566\",\n",
+        ")"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "## DynamoDBChatMessageHistory With Different Keys Composite Keys\n",
+        "The default key for DynamoDBChatMessageHistory is ```{\"SessionId\": self.session_id}```, but you can modify this to match your table design.\n",
+        "\n",
+        "### Primary Key Name\n",
+        "You may modify the primary key by passing in a primary_key_name value in the constructor, resulting in the following:\n",
+        "```{self.primary_key_name: self.session_id}```\n",
+        "\n",
+        "### Composite Keys\n",
+        "When using an existing DynamoDB table, you may need to modify the key structure from the default of to something including a Sort Key. To do this you may use the ```key``` parameter.\n",
+        "\n",
+        "Passing a value for key will override the primary_key parameter, and the resulting key structure will be the passed value.\n"
+      ],
+      "metadata": {
+        "collapsed": false
+      },
+      "id": "c9bc0693"
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 14,
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "0\n"
+          ]
+        },
+        {
+          "data": {
+            "text/plain": "[HumanMessage(content='hello, composite dynamodb table!', additional_kwargs={}, example=False)]"
+          },
+          "execution_count": 14,
+          "metadata": {},
+          "output_type": "execute_result"
+        }
+      ],
+      "source": [
+        "from langchain.memory.chat_message_histories import DynamoDBChatMessageHistory\n",
+        "\n",
+        "composite_table = dynamodb.create_table(\n",
+        "    TableName=\"CompositeTable\",\n",
+        "    KeySchema=[{\"AttributeName\": \"PK\", \"KeyType\": \"HASH\"}, {\"AttributeName\": \"SK\", \"KeyType\": \"RANGE\"}],\n",
+        "    AttributeDefinitions=[{\"AttributeName\": \"PK\", \"AttributeType\": \"S\"}, {\"AttributeName\": \"SK\", \"AttributeType\": \"S\"}],\n",
+        "    BillingMode=\"PAY_PER_REQUEST\",\n",
+        ")\n",
+        "\n",
+        "# Wait until the table exists.\n",
+        "composite_table.meta.client.get_waiter(\"table_exists\").wait(TableName=\"CompositeTable\")\n",
+        "\n",
+        "# Print out some data about the table.\n",
+        "print(composite_table.item_count)\n",
+        "\n",
+        "my_key = {\n",
+        "    \"PK\": \"session_id::0\",\n",
+        "    \"SK\":  \"langchain_history\",\n",
+        "}\n",
+        "\n",
+        "composite_key_history = DynamoDBChatMessageHistory(\n",
+        "    table_name=\"CompositeTable\",\n",
+        "    session_id=\"0\",\n",
+        "    endpoint_url=\"http://localhost.localstack.cloud:4566\",\n",
+        "    key=my_key,\n",
+        ")\n",
+        "\n",
+        "composite_key_history.add_user_message(\"hello, composite dynamodb table!\")\n",
+        "\n",
+        "composite_key_history.messages"
+      ],
+      "metadata": {
+        "collapsed": false
+      },
+      "id": "a7fa0331"
+    },
+    {
+      "attachments": {},
+      "cell_type": "markdown",
+      "id": "3b33c988",
+      "metadata": {},
+      "source": [
+        "## Agent with DynamoDB Memory"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 15,
+      "id": "f92d9499",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "from langchain.agents import Tool\n",
+        "from langchain.memory import ConversationBufferMemory\n",
+        "from langchain.chat_models import ChatOpenAI\n",
+        "from langchain.agents import initialize_agent\n",
+        "from langchain.agents import AgentType\n",
+        "from langchain.utilities import PythonREPL\n",
+        "from getpass import getpass\n",
+        "\n",
+        "message_history = DynamoDBChatMessageHistory(table_name=\"SessionTable\", session_id=\"1\")\n",
+        "memory = ConversationBufferMemory(\n",
+        "    memory_key=\"chat_history\", chat_memory=message_history, return_messages=True\n",
+        ")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 16,
+      "id": "1167eeba",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "python_repl = PythonREPL()\n",
+        "\n",
+        "# You can create the tool to pass to an agent\n",
+        "tools = [\n",
+        "    Tool(\n",
+        "        name=\"python_repl\",\n",
+        "        description=\"A Python shell. Use this to execute python commands. Input should be a valid python command. If you want to see the output of a value, you should print it out with `print(...)`.\",\n",
+        "        func=python_repl.run,\n",
+        "    )\n",
+        "]"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 17,
+      "id": "fce085c5",
+      "metadata": {},
+      "outputs": [
+        {
+          "ename": "ValidationError",
+          "evalue": "1 validation error for ChatOpenAI\n__root__\n  Did not find openai_api_key, please add an environment variable `OPENAI_API_KEY` which contains it, or pass  `openai_api_key` as a named parameter. (type=value_error)",
+          "output_type": "error",
+          "traceback": [
+            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+            "\u001b[0;31mValidationError\u001b[0m                           Traceback (most recent call last)",
+            "Cell \u001b[0;32mIn[17], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m llm \u001b[38;5;241m=\u001b[39m \u001b[43mChatOpenAI\u001b[49m\u001b[43m(\u001b[49m\u001b[43mtemperature\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;241;43m0\u001b[39;49m\u001b[43m)\u001b[49m\n\u001b[1;32m      2\u001b[0m agent_chain \u001b[38;5;241m=\u001b[39m initialize_agent(\n\u001b[1;32m      3\u001b[0m     tools,\n\u001b[1;32m      4\u001b[0m     llm,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m      7\u001b[0m     memory\u001b[38;5;241m=\u001b[39mmemory,\n\u001b[1;32m      8\u001b[0m )\n",
+            "File \u001b[0;32m~/Documents/projects/langchain/libs/langchain/langchain/load/serializable.py:74\u001b[0m, in \u001b[0;36mSerializable.__init__\u001b[0;34m(self, **kwargs)\u001b[0m\n\u001b[1;32m     73\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21m__init__\u001b[39m(\u001b[38;5;28mself\u001b[39m, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs: Any) \u001b[38;5;241m-\u001b[39m\u001b[38;5;241m>\u001b[39m \u001b[38;5;28;01mNone\u001b[39;00m:\n\u001b[0;32m---> 74\u001b[0m     \u001b[38;5;28;43msuper\u001b[39;49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[38;5;21;43m__init__\u001b[39;49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     75\u001b[0m     \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_lc_kwargs \u001b[38;5;241m=\u001b[39m kwargs\n",
+            "File \u001b[0;32m~/Documents/projects/langchain/.venv/lib/python3.9/site-packages/pydantic/main.py:341\u001b[0m, in \u001b[0;36mpydantic.main.BaseModel.__init__\u001b[0;34m()\u001b[0m\n",
+            "\u001b[0;31mValidationError\u001b[0m: 1 validation error for ChatOpenAI\n__root__\n  Did not find openai_api_key, please add an environment variable `OPENAI_API_KEY` which contains it, or pass  `openai_api_key` as a named parameter. (type=value_error)"
+          ]
+        }
+      ],
+      "source": [
+        "llm = ChatOpenAI(temperature=0)\n",
+        "agent_chain = initialize_agent(\n",
+        "    tools,\n",
+        "    llm,\n",
+        "    agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION,\n",
+        "    verbose=True,\n",
+        "    memory=memory,\n",
+        ")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "952a3103",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "agent_chain.run(input=\"Hello!\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "54c4aaf4",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "agent_chain.run(input=\"Who owns Twitter?\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "f9013118",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "agent_chain.run(input=\"My name is Bob.\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "405e5315",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "agent_chain.run(input=\"Who am I?\")\n"
+      ]
+    }
+  ],
+  "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3 (ipykernel)",
+      "language": "python",
+      "name": "python3"
+    },
+    "language_info": {
+      "codemirror_mode": {
+        "name": "ipython",
+        "version": 3
+      },
+      "file_extension": ".py",
+      "mimetype": "text/x-python",
+      "name": "python",
+      "nbconvert_exporter": "python",
+      "pygments_lexer": "ipython3",
+      "version": "3.11.3"
     }
-   ],
-   "source": [
-    "llm = ChatOpenAI(temperature=0)\n",
-    "agent_chain = initialize_agent(\n",
-    "    tools,\n",
-    "    llm,\n",
-    "    agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION,\n",
-    "    verbose=True,\n",
-    "    memory=memory,\n",
-    ")"
-   ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "952a3103",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "agent_chain.run(input=\"Hello!\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "54c4aaf4",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "agent_chain.run(input=\"Who owns Twitter?\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f9013118",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "agent_chain.run(input=\"My name is Bob.\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "405e5315",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "agent_chain.run(input=\"Who am I?\")\n"
-   ]
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.3"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
+  "nbformat": 4,
+  "nbformat_minor": 5
+}

docs/extras/integrations/platforms/google.mdx

@@ -2,6 +2,35 @@
 
 All functionality related to Google Platform
 
+## LLMs
+
+### Vertex AI
+
+Access PaLM LLMs like `text-bison` and `code-bison` via Google Cloud.
+
+```python
+from langchain.llms import VertexAI
+```
+
+### Model Garden
+
+Access PaLM and hundreds of OSS models via Vertex AI Model Garden.
+
+```python
+from langchain.llms import VertexAIModelGarden
+```
+
+## Chat models
+
+### Vertex AI
+
+Access PaLM chat models like `chat-bison` and `codechat-bison` via Google Cloud.
+
+```python
+from langchain.chat_models import ChatVertexAI
+```
+
+
 ## Document Loader
 ### Google BigQuery
 

docs/extras/integrations/providers/argilla.mdx

@@ -5,7 +5,7 @@
 >[Argilla](https://argilla.io/) is an open-source data curation platform for LLMs.
 > Using Argilla, everyone can build robust language models through faster data curation 
 > using both human and machine feedback. We provide support for each step in the MLOps cycle, 
-> from data labeling to model monitoring.
+> from data labelling to model monitoring.
 
 ## Installation and Setup
 

docs/extras/integrations/providers/databricks.md

@@ -13,12 +13,13 @@ Databricks embraces the LangChain ecosystem in various ways:
 
 Databricks connector for the SQLDatabase Chain
 ----------------------------------------------
-You can connect to [Databricks runtimes](https://docs.databricks.com/runtime/index.html) and [Databricks SQL](https://www.databricks.com/product/databricks-sql) using the SQLDatabase wrapper of LangChain. See the notebook [Connect to Databricks](/docs/ecosystem/integrations/databricks/databricks.html) for details.
+You can connect to [Databricks runtimes](https://docs.databricks.com/runtime/index.html) and [Databricks SQL](https://www.databricks.com/product/databricks-sql) using the SQLDatabase wrapper of LangChain. 
+See the notebook [Connect to Databricks](/docs/use_cases/qa_structured/integrations/databricks) for details.
 
 Databricks MLflow integrates with LangChain
 -------------------------------------------
 
-MLflow is an open source platform to manage the ML lifecycle, including experimentation, reproducibility, deployment, and a central model registry. See the notebook [MLflow Callback Handler](/docs/ecosystem/integrations/mlflow_tracking.ipynb) for details about MLflow's integration with LangChain.
+MLflow is an open source platform to manage the ML lifecycle, including experimentation, reproducibility, deployment, and a central model registry. See the notebook [MLflow Callback Handler](/docs/integrations/providers/mlflow_tracking) for details about MLflow's integration with LangChain.
 
 Databricks provides a fully managed and hosted version of MLflow integrated with enterprise security features, high availability, and other Databricks workspace features such as experiment and run management and notebook revision capture. MLflow on Databricks offers an integrated experience for tracking and securing machine learning model training runs and running machine learning projects. See [MLflow guide](https://docs.databricks.com/mlflow/index.html) for more details.
 
@@ -27,7 +28,7 @@ Databricks MLflow makes it more convenient to develop LangChain applications on
 Databricks MLflow AI Gateway
 ----------------------------
 
-See [MLflow AI Gateway](/docs/ecosystem/integrations/mlflow_ai_gateway).
+See [MLflow AI Gateway](/docs/integrations/providers/mlflow_ai_gateway).
 
 Databricks as an LLM provider
 -----------------------------

docs/extras/integrations/providers/elasticsearch.mdx

@@ -18,7 +18,7 @@ Example: Run a single-node Elasticsearch instance with security disabled. This i
 
 #### Deploy Elasticsearch on Elastic Cloud
 
-Elastic Cloud is a managed Elasticsearch service. Signup for a [free trial](https://cloud.elastic.co/registration?storm=langchain-notebook).
+Elastic Cloud is a managed Elasticsearch service. Signup for a [free trial](https://cloud.elastic.co/registration?utm_source=langchain&utm_content=documentation).
 
 ### Install Client
 

docs/extras/integrations/providers/javelin_ai_gateway.mdx

@@ -0,0 +1,92 @@
+# Javelin AI Gateway
+
+[The Javelin AI Gateway](https://www.getjavelin.io) service is a high-performance, enterprise grade API Gateway for AI applications.  
+It is designed to streamline the usage and access of various large language model (LLM) providers, 
+such as OpenAI, Cohere, Anthropic and custom large language models within an organization by incorporating
+robust access security for all interactions with LLMs. 
+
+Javelin offers a high-level interface that simplifies the interaction with LLMs by providing a unified endpoint 
+to handle specific LLM related requests. 
+
+See the Javelin AI Gateway [documentation](https://docs.getjavelin.io) for more details.  
+[Javelin Python SDK](https://www.github.com/getjavelin/javelin-python) is an easy to use client library meant to be embedded into AI Applications
+
+## Installation and Setup
+
+Install `javelin_sdk` to interact with Javelin AI Gateway:
+
+```sh
+pip install 'javelin_sdk'
+```
+
+Set the Javelin's API key as an environment variable:
+
+```sh
+export JAVELIN_API_KEY=...
+```
+
+## Completions Example
+
+```python
+
+from langchain.chains import LLMChain
+from langchain.llms import JavelinAIGateway
+from langchain.prompts import PromptTemplate
+
+route_completions = "eng_dept03"
+
+gateway = JavelinAIGateway(
+    gateway_uri="http://localhost:8000",
+    route=route_completions,
+    model_name="text-davinci-003",
+)
+
+llmchain = LLMChain(llm=gateway, prompt=prompt)
+result = llmchain.run("podcast player")
+
+print(result)
+
+```
+
+## Embeddings Example
+
+```python
+from langchain.embeddings import JavelinAIGatewayEmbeddings
+from langchain.embeddings.openai import OpenAIEmbeddings
+
+embeddings = JavelinAIGatewayEmbeddings(
+    gateway_uri="http://localhost:8000",
+    route="embeddings",
+)
+
+print(embeddings.embed_query("hello"))
+print(embeddings.embed_documents(["hello"]))
+```
+
+## Chat Example
+```python
+from langchain.chat_models import ChatJavelinAIGateway
+from langchain.schema import HumanMessage, SystemMessage
+
+messages = [
+    SystemMessage(
+        content="You are a helpful assistant that translates English to French."
+    ),
+    HumanMessage(
+        content="Artificial Intelligence has the power to transform humanity and make the world a better place"
+    ),
+]
+
+chat = ChatJavelinAIGateway(
+    gateway_uri="http://localhost:8000",
+    route="mychatbot_route",
+    model_name="gpt-3.5-turbo"
+    params={
+        "temperature": 0.1
+    }
+)
+
+print(chat(messages))
+
+```
+

docs/extras/integrations/providers/searchapi.mdx

@@ -0,0 +1,80 @@
+# SearchApi
+
+This page covers how to use the [SearchApi](https://www.searchapi.io/) Google Search API within LangChain. SearchApi is a real-time SERP API for easy SERP scraping.
+
+## Setup
+
+- Go to [https://www.searchapi.io/](https://www.searchapi.io/) to sign up for a free account
+- Get the api key and set it as an environment variable (`SEARCHAPI_API_KEY`)
+
+## Wrappers
+
+### Utility
+
+There is a SearchApiAPIWrapper utility which wraps this API. To import this utility:
+
+```python
+from langchain.utilities import SearchApiAPIWrapper
+```
+
+You can use it as part of a Self Ask chain:
+
+```python
+from langchain.utilities import SearchApiAPIWrapper
+from langchain.llms.openai import OpenAI
+from langchain.agents import initialize_agent, Tool
+from langchain.agents import AgentType
+
+import os
+
+os.environ["SEARCHAPI_API_KEY"] = ""
+os.environ['OPENAI_API_KEY'] = ""
+
+llm = OpenAI(temperature=0)
+search = SearchApiAPIWrapper()
+tools = [
+    Tool(
+        name="Intermediate Answer",
+        func=search.run,
+        description="useful for when you need to ask with search"
+    )
+]
+
+self_ask_with_search = initialize_agent(tools, llm, agent=AgentType.SELF_ASK_WITH_SEARCH, verbose=True)
+self_ask_with_search.run("Who lived longer: Plato, Socrates, or Aristotle?")
+```
+
+#### Output
+
+```
+> Entering new AgentExecutor chain...
+ Yes.
+Follow up: How old was Plato when he died?
+Intermediate answer: eighty
+Follow up: How old was Socrates when he died?
+Intermediate answer: | Socrates | 
+| -------- | 
+| Born | c. 470 BC Deme Alopece, Athens | 
+| Died | 399 BC (aged approximately 71) Athens | 
+| Cause of death | Execution by forced suicide by poisoning | 
+| Spouse(s) | Xanthippe, Myrto | 
+
+Follow up: How old was Aristotle when he died?
+Intermediate answer: 62 years
+So the final answer is: Plato
+
+> Finished chain.
+'Plato'
+```
+
+### Tool
+
+You can also easily load this wrapper as a Tool (to use with an Agent).
+You can do this with:
+
+```python
+from langchain.agents import load_tools
+tools = load_tools(["searchapi"])
+```
+
+For more information on tools, see [this page](/docs/modules/agents/tools/).

docs/extras/integrations/retrievers/kay.ipynb

@@ -0,0 +1,207 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "263f914c-9d67-4316-8b3d-03c3b99ba9d8",
+   "metadata": {},
+   "source": [
+    "Kay.ai\n",
+    "=\n",
+    "\n",
+    "> Data API built for RAG 🕵️ We are curating the world's largest datasets as high-quality embeddings so your AI agents can retrieve context on the fly. Latest models, fast retrieval, and zero infra.\n",
+    "\n",
+    "This notebook shows you how to retrieve datasets supported by [Kay](https://kay.ai/). You can currently search SEC Filings and Press Releases of US companies. Visit [kay.ai](https://kay.ai) for the latest data drops. For any questions, join our [discord](https://discord.gg/hAnE4e5T6M) or [tweet at us](https://twitter.com/vishalrohra_)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "fc507b8e-ea51-417c-93da-42bf998a1195",
+   "metadata": {},
+   "source": [
+    "Installation\n",
+    "=\n",
+    "\n",
+    "First you will need to install the [`kay` package](https://pypi.org/project/kay/). You will also need an API key: you can get one for free at [https://kay.ai](https://kay.ai/). Once you have an API key, you must set it as an environment variable `KAY_API_KEY`.\n",
+    "\n",
+    "`KayAiRetriever` has a static `.create()` factory method that takes the following arguments:\n",
+    "\n",
+    "* `dataset_id: string` required -- A Kay dataset id. This is a collection of data about a particular entity such as companies, people, or places. For example, try `\"company\"` \n",
+    "* `data_type: List[string]` optional -- This is a category within a  dataset based on its origin or format, such as ‘SEC Filings’, ‘Press Releases’, or ‘Reports’ within the “company” dataset. For example, try [\"10-K\", \"10-Q\", \"PressRelease\"] under the “company” dataset. If left empty, Kay will retrieve the most relevant context across all types.\n",
+    "* `num_contexts: int` optional, defaults to 6 -- The number of document chunks to retrieve on each call to `get_relevant_documents()`"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c923bea0-585a-4f62-8662-efc167e8d793",
+   "metadata": {},
+   "source": [
+    "Examples\n",
+    "=\n",
+    "\n",
+    "Basic Retriever Usage\n",
+    "-"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "f7b8c99c-0341-4f3c-912f-a11e98f7de71",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdin",
+     "output_type": "stream",
+     "text": [
+      " ········\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Setup API key\n",
+    "from getpass import getpass\n",
+    "KAY_API_KEY = getpass()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 20,
+   "id": "b4d4d386-2a6b-4942-863e-9202f5a9f1d6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.retrievers import KayAiRetriever\n",
+    "import os\n",
+    "from kay.rag.retrievers import KayRetriever\n",
+    "os.environ[\"KAY_API_KEY\"] = KAY_API_KEY\n",
+    "retriever = KayAiRetriever.create(dataset_id=\"company\", data_types=[\"10-K\", \"10-Q\", \"PressRelease\"], num_contexts=3)\n",
+    "docs = retriever.get_relevant_documents(\"What were the biggest strategy changes and partnerships made by Roku in 2023??\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 21,
+   "id": "04ee2d6b-c2ab-4e15-8a8b-afaf6ef8c0f6",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[Document(page_content='Company Name: ROKU INC\\nCompany Industry: CABLE & OTHER PAY TELEVISION SERVICES\\nArticle Title: Roku and FreeWheel Announce Strategic Partnership to Bring Roku’s Leading Ad Tech to FreeWheel Customers\\nText: Additionally, eMarketer Link: https://cts.businesswire.com/ct/CT?id=smartlink&url=https%3A%2F%2Fwww.insiderintelligence.com%2Finsights%2Favod-more-than-50-percent-of-us-digital-video-viewers%2F&esheet=53451144&newsitemid=20230712907788&lan=en-US&anchor=eMarketer&index=4&md5=b64dea72bcf6b6379474462602781d83 projects 57% of U.S. digital video users will stream an advertising-based video on demand (AVOD) service this year.\\nHaving solutions aimed at driving greater interoperability and automation will help accelerate this growth.\\nKey highlights of this collaboration include:\\nStreamlined Integration: Roku has now integrated its demand application programming interface (dAPI) with FreeWheel s TV platform. Roku s demand API gives publishers direct, automatic and real-time access to more advertiser demand. This enhanced integration allows for streamlined ad operation workflows and better inventory quality control, both of which will improve publisher yield and revenue.\\nSeamless Data Targeting: Publishers can now use Roku platform signals to enable advertisers to target audiences and measure campaign performance without relying on cookies. Additionally, FreeWheel and Roku will rely on data clean room technology to enable the activation of additional data sets providing better measurement and monetization to publishers and agencies.', metadata={'_additional': {'id': '962b79e0-f9d1-43ae-9f7a-8a9b42bc7a9a'}, 'chunk_type': 'text', 'chunk_years_mentioned': [], 'company_name': 'ROKU INC', 'company_sic_code_description': 'CABLE & OTHER PAY TELEVISION SERVICES', 'data_source': 'PressRelease', 'data_source_link': 'https://www.nasdaq.com/press-release/roku-and-freewheel-announce-strategic-partnership-to-bring-rokus-leading-ad-tech-to', 'data_source_publish_date': '2023-07-12T00:00:00Z', 'data_source_uid': 'a46f309c-705d-3946-96db-87aa4e73261f', 'title': 'ROKU INC |  Roku and FreeWheel Announce Strategic Partnership to Bring Roku’s Leading Ad Tech to FreeWheel Customers'}),\n",
+       " Document(page_content='Company Name: ROKU INC \\n Company Industry: CABLE & OTHER PAY TELEVISION SERVICES \\n Form Title: 10-K 2022-FY \\n Form Section: Risk Factors \\n Text: nd the Note Regarding Forward Looking Statements.This section of this Annual Report generally discusses fiscal years 2022 and 2021 and year to year comparisons between those years.Discussions of fiscal year 2020 and year to year comparisons between fiscal years 2021 and 2020 that are not included in this Annual Report can be found in Management\\'s Discussion and Analysis of Financial Condition and Results of Operations in Part II, Item 7 of our Annual Report for the fiscal year ended December 31, 2021 filed with the SEC on February 18, 2022.Overview Effective as of the fourth quarter of fiscal 2022, we reorganized our reportable segments to better align with management\\'s reporting of information reviewed by the Chief Operating Decision Maker (\"CODM\") for each segment.We renamed our \"player\" segment to \"devices\" which now includes our licensing arrangements with service operators and licensed Roku TV partners in addition to sales of our streaming players, audio products, smart home products and Roku branded TVs that will be designed, made, and sold by us in 2023.Our historical segment information is recast to conform to our new presentation in our financial statements and accompanying notes included in Item 8 of this Annual Report.Our two reportable segments are the platform segment and the devices segment.', metadata={'_additional': {'id': 'a76c5fed-5d63-45a7-b63a-2c30e05140fc'}, 'chunk_type': 'text', 'chunk_years_mentioned': [2020, 2021, 2022, 2023], 'company_name': 'ROKU INC', 'company_sic_code_description': 'CABLE & OTHER PAY TELEVISION SERVICES', 'data_source': '10-K', 'data_source_link': 'https://www.sec.gov/Archives/edgar/data/1428439/000142843923000007', 'data_source_publish_date': '2022-01-01T00:00:00Z', 'data_source_uid': '0001428439-23-000007', 'title': 'ROKU INC |  10-K 2022-FY '}),\n",
+       " Document(page_content='Company Name: ROKU INC \\n Company Industry: CABLE & OTHER PAY TELEVISION SERVICES \\n Form Title: 10-Q 2023-Q1 \\n Form Section: Risk Factors \\n Text: Our current and potential partners include TV brands, cable and satellite companies, and telecommunication providers.Under these license arrangements, we generally have limited or no control over the amount and timing of resources these entities dedicate to the relationship.In the past, our licensed Roku TV partners have failed to meet their forecasts and anticipated market launch dates for distributing Roku TV models, and they may fail to meet their forecasts or such launches in the future.If our licensed Roku TV partners or service operator partners fail to meet their forecasts or such launches for distributing licensed streaming devices or choose to deploy competing streaming solutions within their product lines, our business may be harmed.We depend on a small number of content publishers for a majority of our streaming hours, and if we fail to maintain these relationships, our business could be harmed.*Historically, a small number of content publishers have accounted for a significant portion of the hours streamed on our platform.In the three months ended March 31, 2023, the top three streaming services represented over 50% of all hours streamed in the period.If, for any reason, we cease distributing channels that have historically streamed a large percentage of the aggregate streaming hours on our platform, our streaming hours, our active accounts, or Roku streaming device sales may be adversely affected, and our business may be harmed.', metadata={'_additional': {'id': '2a92b2bb-02a0-4e15-8b64-d7e04078a205'}, 'chunk_type': 'text', 'chunk_years_mentioned': [2023], 'company_name': 'ROKU INC', 'company_sic_code_description': 'CABLE & OTHER PAY TELEVISION SERVICES', 'data_source': '10-Q', 'data_source_link': 'https://www.sec.gov/Archives/edgar/data/1428439/000142843923000017', 'data_source_publish_date': '2023-01-01T00:00:00Z', 'data_source_uid': '0001428439-23-000017', 'title': 'ROKU INC |  10-Q 2023-Q1 '})]"
+      ]
+     },
+     "execution_count": 21,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "docs"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "21f6e9e5-478c-4b2c-9d61-f7a84f4d2f8f",
+   "metadata": {},
+   "source": [
+    "Usage in a chain\n",
+    "-"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "d1cba716-ab8d-4518-9196-43f17eb189dc",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdin",
+     "output_type": "stream",
+     "text": [
+      " ········\n"
+     ]
+    }
+   ],
+   "source": [
+    "OPENAI_API_KEY = getpass()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "79441f1f-fa06-452c-bcd6-160ad0debc6a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "os.environ[\"OPENAI_API_KEY\"] = OPENAI_API_KEY"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "id": "0c504bcd-f6e0-4028-a797-b31fb4b6d027",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from langchain.chains import ConversationalRetrievalChain\n",
+    "\n",
+    "model = ChatOpenAI(model_name=\"gpt-3.5-turbo\")\n",
+    "qa = ConversationalRetrievalChain.from_llm(model, retriever=retriever)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 22,
+   "id": "977f158b-38d3-4b5f-9379-7cdd09436327",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "-> **Question**: What were the biggest strategy changes and partnerships made by Roku in 2023? \n",
+      "\n",
+      "**Answer**: In 2023, Roku made a strategic partnership with FreeWheel to bring Roku's leading ad tech to FreeWheel customers. This partnership aimed to drive greater interoperability and automation in the advertising-based video on demand (AVOD) space. Key highlights of this collaboration include streamlined integration of Roku's demand application programming interface (dAPI) with FreeWheel's TV platform, allowing for better inventory quality control and improved publisher yield and revenue. Additionally, publishers can now use Roku platform signals to enable advertisers to target audiences and measure campaign performance without relying on cookies. This partnership also involves the use of data clean room technology to enable the activation of additional data sets for better measurement and monetization for publishers and agencies. These partnerships and strategies aim to support Roku's growth in the AVOD market. \n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "questions = [\n",
+    "    \"What were the biggest strategy changes and partnerships made by Roku in 2023?\"\n",
+    "    # \"Where is Wex making the most money in 2023?\",\n",
+    "]\n",
+    "chat_history = []\n",
+    "\n",
+    "for question in questions:\n",
+    "    result = qa({\"question\": question, \"chat_history\": chat_history})\n",
+    "    chat_history.append((question, result[\"answer\"]))\n",
+    "    print(f\"-> **Question**: {question} \\n\")\n",
+    "    print(f\"**Answer**: {result['answer']} \\n\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.18"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/retrievers/pubmed.ipynb

@@ -81,7 +81,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.9.18"
   }
  },
  "nbformat": 4,

docs/extras/integrations/retrievers/sec_filings.ipynb

@@ -0,0 +1,165 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "263f914c-9d67-4316-8b3d-03c3b99ba9d8",
+   "metadata": {},
+   "source": [
+    "SEC filings data\n",
+    "=\n",
+    "\n",
+    "SEC filings data powered by [Kay.ai](https://kay.ai) and [Cybersyn](https://www.cybersyn.com/).\n",
+    "\n",
+    ">The SEC filing is a financial statement or other formal document submitted to the U.S. Securities and Exchange Commission (SEC). Public companies, certain insiders, and broker-dealers are required to make regular SEC filings. Investors and financial professionals rely on these filings for information about companies they are evaluating for investment purposes."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "fc507b8e-ea51-417c-93da-42bf998a1195",
+   "metadata": {},
+   "source": [
+    "Setup\n",
+    "=\n",
+    "\n",
+    "First you will need to install the `kay` package. You will also need an API key: you can get one for free at [https://kay.ai](https://kay.ai/). Once you have an API key, you must set it as an environment variable `KAY_API_KEY`.\n",
+    "\n",
+    "In this example we're going to use the `KayAiRetriever`. Take a look at the [kay notebook](/docs/integrations/retrievers/kay) for more detailed information for the parmeters that it accepts.`"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c923bea0-585a-4f62-8662-efc167e8d793",
+   "metadata": {},
+   "source": [
+    "Examples\n",
+    "=\n",
+    "\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "f7b8c99c-0341-4f3c-912f-a11e98f7de71",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdin",
+     "output_type": "stream",
+     "text": [
+      " ········\n",
+      " ········\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Setup API keys for Kay and OpenAI\n",
+    "from getpass import getpass\n",
+    "KAY_API_KEY = getpass()\n",
+    "OPENAI_API_KEY = getpass()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "04ee2d6b-c2ab-4e15-8a8b-afaf6ef8c0f6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "os.environ[\"KAY_API_KEY\"] = KAY_API_KEY\n",
+    "os.environ[\"OPENAI_API_KEY\"] = OPENAI_API_KEY"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "0c504bcd-f6e0-4028-a797-b31fb4b6d027",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.chains import ConversationalRetrievalChain\n",
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from langchain.retrievers import KayAiRetriever\n",
+    "\n",
+    "model = ChatOpenAI(model_name=\"gpt-3.5-turbo\")\n",
+    "retriever = KayAiRetriever.create(dataset_id=\"company\", data_types=[\"10-K\", \"10-Q\"], num_contexts=6)\n",
+    "qa = ConversationalRetrievalChain.from_llm(model, retriever=retriever)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "977f158b-38d3-4b5f-9379-7cdd09436327",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "-> **Question**: What are patterns in Nvidia's spend over the past three quarters? \n",
+      "\n",
+      "**Answer**: Based on the provided information, here are the patterns in NVIDIA's spend over the past three quarters:\n",
+      "\n",
+      "1. Research and Development Expenses:\n",
+      "   - Q3 2022: Increased by 34% compared to Q3 2021.\n",
+      "   - Q1 2023: Increased by 40% compared to Q1 2022.\n",
+      "   - Q2 2022: Increased by 25% compared to Q2 2021.\n",
+      "   \n",
+      "   Overall, research and development expenses have been consistently increasing over the past three quarters.\n",
+      "\n",
+      "2. Sales, General and Administrative Expenses:\n",
+      "   - Q3 2022: Increased by 8% compared to Q3 2021.\n",
+      "   - Q1 2023: Increased by 14% compared to Q1 2022.\n",
+      "   - Q2 2022: Decreased by 16% compared to Q2 2021.\n",
+      "   \n",
+      "   The pattern for sales, general and administrative expenses is not as consistent, with some quarters showing an increase and others showing a decrease.\n",
+      "\n",
+      "3. Total Operating Expenses:\n",
+      "   - Q3 2022: Increased by 25% compared to Q3 2021.\n",
+      "   - Q1 2023: Increased by 113% compared to Q1 2022.\n",
+      "   - Q2 2022: Increased by 9% compared to Q2 2021.\n",
+      "   \n",
+      "   Total operating expenses have generally been increasing over the past three quarters, with a significant increase in Q1 2023.\n",
+      "\n",
+      "Overall, the pattern indicates a consistent increase in research and development expenses and total operating expenses, while sales, general and administrative expenses show some fluctuations. \n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "questions = [\n",
+    "    \"What are patterns in Nvidia's spend over the past three quarters?\",\n",
+    "    #\"What are some recent challenges faced by the renewable energy sector?\",\n",
+    "]\n",
+    "chat_history = []\n",
+    "\n",
+    "for question in questions:\n",
+    "    result = qa({\"question\": question, \"chat_history\": chat_history})\n",
+    "    chat_history.append((question, result[\"answer\"]))\n",
+    "    print(f\"-> **Question**: {question} \\n\")\n",
+    "    print(f\"**Answer**: {result['answer']} \\n\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.18"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/text_embedding/gradient.ipynb

@@ -0,0 +1,150 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Gradient\n",
+    "\n",
+    "`Gradient` allows to create `Embeddings` as well fine tune and get completions on LLMs with a simple web API.\n",
+    "\n",
+    "This notebook goes over how to use Langchain with Embeddings of [Gradient](https://gradient.ai/).\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Imports"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.embeddings import GradientEmbeddings"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Set the Environment API Key\n",
+    "Make sure to get your API key from Gradient AI. You are given $10 in free credits to test and fine-tune different models."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from getpass import getpass\n",
+    "import os\n",
+    "\n",
+    "if not os.environ.get(\"GRADIENT_ACCESS_TOKEN\",None):\n",
+    "    # Access token under https://auth.gradient.ai/select-workspace\n",
+    "    os.environ[\"GRADIENT_ACCESS_TOKEN\"] = getpass(\"gradient.ai access token:\")\n",
+    "if not os.environ.get(\"GRADIENT_WORKSPACE_ID\",None):\n",
+    "    # `ID` listed in `$ gradient workspace list`\n",
+    "    # also displayed after login at at https://auth.gradient.ai/select-workspace\n",
+    "    os.environ[\"GRADIENT_WORKSPACE_ID\"] = getpass(\"gradient.ai workspace id:\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Optional: Validate your Environment variables ```GRADIENT_ACCESS_TOKEN``` and ```GRADIENT_WORKSPACE_ID``` to get currently deployed models. Using the `gradientai` Python package."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install gradientai"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Create the Gradient instance"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "documents = [\"Pizza is a dish.\",\"Paris is the capital of France\", \"numpy is a lib for linear algebra\"]\n",
+    "query = \"Where is Paris?\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "embeddings = GradientEmbeddings(\n",
+    "    model=\"bge-large\"\n",
+    ")\n",
+    "\n",
+    "documents_embedded = embeddings.embed_documents(documents)\n",
+    "query_result = embeddings.embed_query(query)\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# (demo) compute similarity\n",
+    "import numpy as np\n",
+    "\n",
+    "scores = np.array(documents_embedded) @ np.array(query_result).T\n",
+    "dict(zip(documents, scores))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.6"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "a0a0263b650d907a3bfe41c0f8d6a63a071b884df3cfdc1579f00cdc1aed6b03"
+   }
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

docs/extras/integrations/text_embedding/llm_rails.ipynb

@@ -0,0 +1,133 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "278b6c63",
+   "metadata": {},
+   "source": [
+    "# LLMRails\n",
+    "\n",
+    "Let's load the LLMRails Embeddings class.\n",
+    "\n",
+    "To use LLMRails embedding you need to pass api key by argument or set it in environment with `LLM_RAILS_API_KEY` key.\n",
+    "To gey API Key you need to sign up in https://console.llmrails.com/signup and then go to https://console.llmrails.com/api-keys and copy key from there after creating one key in platform."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "0be1af71",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.embeddings import LLMRailsEmbeddings"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "2c66e5da",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "embeddings = LLMRailsEmbeddings(model='embedding-english-v1') # or embedding-multi-v1"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "01370375",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "text = \"This is a test document.\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a42e4035",
+   "metadata": {},
+   "source": [
+    "To generate embeddings, you can either query an invidivual text, or you can query a list of texts."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "91bc875d-829b-4c3d-8e6f-fc2dda30a3bd",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[-0.09996652603149414,\n",
+       " 0.015568195842206478,\n",
+       " 0.17670190334320068,\n",
+       " 0.16521021723747253,\n",
+       " 0.21193109452724457]"
+      ]
+     },
+     "execution_count": 4,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "query_result = embeddings.embed_query(text)\n",
+    "query_result[:5]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "a4b0d49e-0c73-44b6-aed5-5b426564e085",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[-0.04242777079343796,\n",
+       " 0.016536075621843338,\n",
+       " 0.10052520781755447,\n",
+       " 0.18272875249385834,\n",
+       " 0.2079043835401535]"
+      ]
+     },
+     "execution_count": 6,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "doc_result = embeddings.embed_documents([text])\n",
+    "doc_result[0][:5]"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.5"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "e971737741ff4ec9aff7dc6155a1060a59a8a6d52c757dbbe66bf8ee389494b1"
+   }
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/tools/search_tools.ipynb

@@ -107,6 +107,85 @@
     "agent.run(\"What is the weather in Pomfret?\")"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "8786bdc8",
+   "metadata": {},
+   "source": [
+    "## SearchApi\n",
+    "\n",
+    "Second, let's try SearchApi tool."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "5fd5ca32",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "tools = load_tools([\"searchapi\"], llm=llm)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "547c9cf5-aa4d-48ed-b7a5-29ecc1491adf",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent = initialize_agent(\n",
+    "    tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "a7564c40-83ec-490b-ad36-385be5c20e58",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m I need to find out the current weather in Pomfret.\n",
+      "Action: searchapi\n",
+      "Action Input: \"weather in Pomfret\"\u001b[0m\n",
+      "Observation: \u001b[36;1m\u001b[1;3mThu 14 | Day ... Some clouds this morning will give way to generally sunny skies for the afternoon. High 73F. Winds NW at 5 to 10 mph.\n",
+      "Hourly Weather-Pomfret, CT · 1 pm. 71°. 0%. Sunny. Feels Like71°. WindNW 9 mph · 2 pm. 72°. 0%. Sunny. Feels Like72°. WindNW 9 mph · 3 pm. 72°. 0%. Sunny. Feels ...\n",
+      "10 Day Weather-Pomfret, VT. As of 4:28 am EDT. Today. 68°/48°. 4%. Thu 14 | Day. 68°. 4%. WNW 10 mph. Some clouds this morning will give way to generally ...\n",
+      "Be prepared with the most accurate 10-day forecast for Pomfret, MD with highs, lows, chance of precipitation from The Weather Channel and Weather.com.\n",
+      "Current Weather. 10:00 PM. 65°F. RealFeel® 67°. Mostly cloudy. LOCAL HURRICANE TRACKER. Category2. Lee. Late Friday Night - Saturday Afternoon.\n",
+      "10 Day Weather-Pomfret, NY. As of 5:09 pm EDT. Tonight. --/55°. 10%. Wed 13 | Night. 55°. 10%. NW 11 mph. Some clouds. Low near 55F.\n",
+      "Pomfret CT. Overnight. Overnight: Patchy fog before 3am, then patchy fog after 4am. Otherwise, mostly. Patchy Fog. Low: 58 °F. Thursday.\n",
+      "Isolated showers. Mostly cloudy, with a high near 76. Calm wind. Chance of precipitation is 20%. Tonight. Mostly Cloudy. Mostly cloudy, with a ...\n",
+      "Partly sunny, with a high near 67. Breezy, with a north wind 18 to 22 mph, with gusts as high as 34 mph. Chance of precipitation is 30%. ... A chance of showers ...\n",
+      "Today's Weather - Pomfret, CT ... Patchy fog. Showers. Lows in the upper 50s. Northwest winds around 5 mph. Chance of rain near 100 percent. ... Sunny. Patchy fog ...\u001b[0m\n",
+      "Thought:\u001b[32;1m\u001b[1;3m I now know the final answer\n",
+      "Final Answer: The current weather in Pomfret is mostly cloudy with a high near 67 and a chance of showers. Winds are from the north at 18 to 22 mph with gusts up to 34 mph.\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'The current weather in Pomfret is mostly cloudy with a high near 67 and a chance of showers. Winds are from the north at 18 to 22 mph with gusts up to 34 mph.'"
+      ]
+     },
+     "execution_count": 5,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent.run(\"What is the weather in Pomfret?\")"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "0e39fc46",

docs/extras/integrations/tools/searchapi.ipynb

@@ -0,0 +1,620 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "7960ce8a-859a-41f4-a886-0d1502ed1105",
+   "metadata": {},
+   "source": [
+    "# SearchApi\n",
+    "\n",
+    "This notebook shows examples of how to use SearchApi to search the web. Go to [https://www.searchapi.io/](https://www.searchapi.io/) to sign up for a free account and get API key."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "70871a99-ffee-47d7-8e02-82eb99971f28",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "\n",
+    "os.environ[\"SEARCHAPI_API_KEY\"] = \"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "2e26a518-c41c-4d75-9a79-67602ca2ec43",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.utilities import SearchApiAPIWrapper"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "8c0977f3-c136-400a-8024-f4f00645b981",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "search = SearchApiAPIWrapper()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "f573767d-4144-4407-8149-5fdddab99c63",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'Barack Hussein Obama II'"
+      ]
+     },
+     "execution_count": 4,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "search.run(\"Obama's first name?\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "9f4f75ae-2e1e-42db-a991-3ac111029f56",
+   "metadata": {},
+   "source": [
+    "## Using as part of a Self Ask With Search Chain"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "17a9b1ad-6e84-4949-8ebd-8c52f6b296e3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "os.environ[\"OPENAI_API_KEY\"] = \"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "cf8970a5-00e1-46bd-ba53-6a974eebbc10",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m Yes.\n",
+      "Follow up: How old was Plato when he died?\u001b[0m\n",
+      "Intermediate answer: \u001b[36;1m\u001b[1;3meighty\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3mFollow up: How old was Socrates when he died?\u001b[0m\n",
+      "Intermediate answer: \u001b[36;1m\u001b[1;3m| Socrates | \n",
+      "| -------- | \n",
+      "| Born | c. 470 BC Deme Alopece, Athens | \n",
+      "| Died | 399 BC (aged approximately 71) Athens | \n",
+      "| Cause of death | Execution by forced suicide by poisoning | \n",
+      "| Spouse(s) | Xanthippe, Myrto | \n",
+      "\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3mFollow up: How old was Aristotle when he died?\u001b[0m\n",
+      "Intermediate answer: \u001b[36;1m\u001b[1;3m62 years\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3mSo the final answer is: Plato\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'Plato'"
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "from langchain.utilities import SearchApiAPIWrapper\n",
+    "from langchain.llms.openai import OpenAI\n",
+    "from langchain.agents import initialize_agent, Tool\n",
+    "from langchain.agents import AgentType\n",
+    "\n",
+    "llm = OpenAI(temperature=0)\n",
+    "search = SearchApiAPIWrapper()\n",
+    "tools = [\n",
+    "    Tool(\n",
+    "        name=\"Intermediate Answer\",\n",
+    "        func=search.run,\n",
+    "        description=\"useful for when you need to ask with search\"\n",
+    "    )\n",
+    "]\n",
+    "\n",
+    "self_ask_with_search = initialize_agent(tools, llm, agent=AgentType.SELF_ASK_WITH_SEARCH, verbose=True)\n",
+    "self_ask_with_search.run(\"Who lived longer: Plato, Socrates, or Aristotle?\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "cc433d06-579b-45e5-a256-2bb30bbefb93",
+   "metadata": {},
+   "source": [
+    "## Custom parameters\n",
+    "\n",
+    "SearchApi wrapper can be customized to use different engines like [Google News](https://www.searchapi.io/docs/google-news), [Google Jobs](https://www.searchapi.io/docs/google-jobs), [Google Scholar](https://www.searchapi.io/docs/google-scholar), or others which can be found in [SearchApi](https://www.searchapi.io/docs/google) documentation. All parameters supported by SearchApi can be passed when executing the query. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "6d0b4411-780a-4dcf-91b6-f3544e31e532",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "search = SearchApiAPIWrapper(engine=\"google_jobs\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "34e79449-6b33-4b45-9306-7e3dab1b8599",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'Azure AI Engineer Be an XpanderCandidatar-meCandidatar-meCandidatar-me\\n\\nShare:\\n\\nAzure AI Engineer\\n\\nA área Digital Xperience da Xpand IT é uma equipa tecnológica de rápido crescimento que se concentra em tecnologias Microsoft e Mobile. A sua principal missão é fornecer soluções de software de alta qualidade que atendam às necessidades do utilizador final, num mundo tecnológico continuamente exigente e em ritmo acelerado, proporcionando a melhor experiência em termos de personalização, performance'"
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "search.run(\"AI Engineer\", location=\"Portugal\", gl=\"pt\")[0:500]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "d414513d-f374-4af0-a129-e878d4311a1e",
+   "metadata": {},
+   "source": [
+    "## Getting results with metadata"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "b16b7cd9-f0fe-4030-a36b-bbb52b19da18",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import pprint"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "e8adb325-2ad0-4a39-9bc2-d220ec3a29be",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "{'search_metadata': {'id': 'search_qVdXG2jzvrlqTzayeYoaOb8A',\n",
+      "                     'status': 'Success',\n",
+      "                     'created_at': '2023-09-25T15:22:30Z',\n",
+      "                     'request_time_taken': 3.21,\n",
+      "                     'parsing_time_taken': 0.03,\n",
+      "                     'total_time_taken': 3.24,\n",
+      "                     'request_url': 'https://scholar.google.com/scholar?q=Large+Language+Models&hl=en',\n",
+      "                     'html_url': 'https://www.searchapi.io/api/v1/searches/search_qVdXG2jzvrlqTzayeYoaOb8A.html',\n",
+      "                     'json_url': 'https://www.searchapi.io/api/v1/searches/search_qVdXG2jzvrlqTzayeYoaOb8A'},\n",
+      " 'search_parameters': {'engine': 'google_scholar',\n",
+      "                       'q': 'Large Language Models',\n",
+      "                       'hl': 'en'},\n",
+      " 'search_information': {'query_displayed': 'Large Language Models',\n",
+      "                        'total_results': 6420000,\n",
+      "                        'page': 1,\n",
+      "                        'time_taken_displayed': 0.06},\n",
+      " 'organic_results': [{'position': 1,\n",
+      "                      'title': 'ChatGPT for good? On opportunities and '\n",
+      "                               'challenges of large language models for '\n",
+      "                               'education',\n",
+      "                      'data_cid': 'uthwmf2nU3EJ',\n",
+      "                      'link': 'https://www.sciencedirect.com/science/article/pii/S1041608023000195',\n",
+      "                      'publication': 'E Kasneci, K Seßler, S Küchemann, M '\n",
+      "                                     'Bannert… - Learning and individual …, '\n",
+      "                                     '2023 - Elsevier',\n",
+      "                      'snippet': '… state of large language models and their '\n",
+      "                                 'applications. We then highlight how these '\n",
+      "                                 'models can be … With regard to challenges, '\n",
+      "                                 'we argue that large language models in '\n",
+      "                                 'education require …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '8166055256995715258',\n",
+      "                                                    'total': 410,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=8166055256995715258&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '8166055256995715258',\n",
+      "                                                    'total': 10,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=8166055256995715258&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:uthwmf2nU3EJ:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'edarxiv.org',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://edarxiv.org/5er8f/download?format=pdf'},\n",
+      "                      'authors': [{'name': 'E Kasneci',\n",
+      "                                   'id': 'bZVkVvoAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=bZVkVvoAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'K Seßler',\n",
+      "                                   'id': 'MbMBoN4AAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=MbMBoN4AAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'S Küchemann',\n",
+      "                                   'id': 'g1jX5QUAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=g1jX5QUAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'M Bannert',\n",
+      "                                   'id': 'TjfQ8QkAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=TjfQ8QkAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 2,\n",
+      "                      'title': 'Large language models in medicine',\n",
+      "                      'data_cid': 'Ph9AwHTmhzAJ',\n",
+      "                      'link': 'https://www.nature.com/articles/s41591-023-02448-8',\n",
+      "                      'publication': 'AJ Thirunavukarasu, DSJ Ting, K '\n",
+      "                                     'Elangovan… - Nature medicine, 2023 - '\n",
+      "                                     'nature.com',\n",
+      "                      'snippet': '… HuggingChat offers a free-to-access '\n",
+      "                                 'chatbot with a similar interface to ChatGPT '\n",
+      "                                 'but uses Large Language Model Meta AI '\n",
+      "                                 '(LLaMA) as its backend model 30 . Finally, '\n",
+      "                                 'cheap imitations of …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '3497017024792502078',\n",
+      "                                                    'total': 25,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=3497017024792502078&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '3497017024792502078',\n",
+      "                                                    'total': 3,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=3497017024792502078&hl=en&as_sdt=0,33'}},\n",
+      "                      'authors': [{'name': 'AJ Thirunavukarasu',\n",
+      "                                   'id': '3qb1AYwAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=3qb1AYwAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'DSJ Ting',\n",
+      "                                   'id': 'KbrpC8cAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=KbrpC8cAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'K Elangovan',\n",
+      "                                   'id': 'BE_lVTQAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=BE_lVTQAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 3,\n",
+      "                      'title': 'Extracting training data from large language '\n",
+      "                               'models',\n",
+      "                      'data_cid': 'mEYsWK6bWKoJ',\n",
+      "                      'link': 'https://www.usenix.org/conference/usenixsecurity21/presentation/carlini-extracting',\n",
+      "                      'publication': 'N Carlini, F Tramer, E Wallace, M '\n",
+      "                                     'Jagielski… - 30th USENIX Security …, '\n",
+      "                                     '2021 - usenix.org',\n",
+      "                      'snippet': '… language model trained on scrapes of the '\n",
+      "                                 'public Internet, and are able to extract '\n",
+      "                                 'hundreds of verbatim text sequences from the '\n",
+      "                                 'model’… models are more vulnerable than '\n",
+      "                                 'smaller models. …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '12274731957504198296',\n",
+      "                                                    'total': 742,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=12274731957504198296&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '12274731957504198296',\n",
+      "                                                    'total': 8,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=12274731957504198296&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:mEYsWK6bWKoJ:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:mEYsWK6bWKoJ:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'usenix.org',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://www.usenix.org/system/files/sec21-carlini-extracting.pdf'},\n",
+      "                      'authors': [{'name': 'N Carlini',\n",
+      "                                   'id': 'q4qDvAoAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=q4qDvAoAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'F Tramer',\n",
+      "                                   'id': 'ijH0-a8AAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=ijH0-a8AAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'E Wallace',\n",
+      "                                   'id': 'SgST3LkAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=SgST3LkAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'M Jagielski',\n",
+      "                                   'id': '_8rw_GMAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=_8rw_GMAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 4,\n",
+      "                      'title': 'Emergent abilities of large language models',\n",
+      "                      'data_cid': 'hG0iVOrOguoJ',\n",
+      "                      'link': 'https://arxiv.org/abs/2206.07682',\n",
+      "                      'publication': 'J Wei, Y Tay, R Bommasani, C Raffel, B '\n",
+      "                                     'Zoph… - arXiv preprint arXiv …, 2022 - '\n",
+      "                                     'arxiv.org',\n",
+      "                      'snippet': 'Scaling up language models has been shown to '\n",
+      "                                 'predictably improve performance and sample '\n",
+      "                                 'efficiency on a wide range of downstream '\n",
+      "                                 'tasks. This paper instead discusses an …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '16898296257676733828',\n",
+      "                                                    'total': 621,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=16898296257676733828&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '16898296257676733828',\n",
+      "                                                    'total': 12,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=16898296257676733828&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:hG0iVOrOguoJ:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:hG0iVOrOguoJ:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'arxiv.org',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://arxiv.org/pdf/2206.07682.pdf?trk=cndc-detail'},\n",
+      "                      'authors': [{'name': 'J Wei',\n",
+      "                                   'id': 'wA5TK_0AAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=wA5TK_0AAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'Y Tay',\n",
+      "                                   'id': 'VBclY_cAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=VBclY_cAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'R Bommasani',\n",
+      "                                   'id': 'WMBXw1EAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=WMBXw1EAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'C Raffel',\n",
+      "                                   'id': 'I66ZBYwAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=I66ZBYwAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'B Zoph',\n",
+      "                                   'id': 'NL_7iTwAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=NL_7iTwAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 5,\n",
+      "                      'title': 'A survey on evaluation of large language '\n",
+      "                               'models',\n",
+      "                      'data_cid': 'ZYohnzOz-XgJ',\n",
+      "                      'link': 'https://arxiv.org/abs/2307.03109',\n",
+      "                      'publication': 'Y Chang, X Wang, J Wang, Y Wu, K Zhu… - '\n",
+      "                                     'arXiv preprint arXiv …, 2023 - arxiv.org',\n",
+      "                      'snippet': '… 3.1 Natural Language Processing Tasks … '\n",
+      "                                 'the development of language models, '\n",
+      "                                 'particularly large language models, was to '\n",
+      "                                 'enhance performance on natural language '\n",
+      "                                 'processing tasks, …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '8717195588046785125',\n",
+      "                                                    'total': 31,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=8717195588046785125&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '8717195588046785125',\n",
+      "                                                    'total': 3,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=8717195588046785125&hl=en&as_sdt=0,33'},\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:ZYohnzOz-XgJ:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'arxiv.org',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://arxiv.org/pdf/2307.03109'},\n",
+      "                      'authors': [{'name': 'X Wang',\n",
+      "                                   'id': 'Q7Ieos8AAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=Q7Ieos8AAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'J Wang',\n",
+      "                                   'id': 'YomxTXQAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=YomxTXQAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'Y Wu',\n",
+      "                                   'id': 'KVeRu2QAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=KVeRu2QAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'K Zhu',\n",
+      "                                   'id': 'g75dFLYAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=g75dFLYAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 6,\n",
+      "                      'title': 'Evaluating large language models trained on '\n",
+      "                               'code',\n",
+      "                      'data_cid': '3tNvW3l5nU4J',\n",
+      "                      'link': 'https://arxiv.org/abs/2107.03374',\n",
+      "                      'publication': 'M Chen, J Tworek, H Jun, Q Yuan, HPO '\n",
+      "                                     'Pinto… - arXiv preprint arXiv …, 2021 - '\n",
+      "                                     'arxiv.org',\n",
+      "                      'snippet': '… We introduce Codex, a GPT language model '\n",
+      "                                 'finetuned on publicly available code from '\n",
+      "                                 'GitHub, and study its Python code-writing '\n",
+      "                                 'capabilities. A distinct production version '\n",
+      "                                 'of Codex …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '5664817468434011102',\n",
+      "                                                    'total': 941,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=5664817468434011102&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '5664817468434011102',\n",
+      "                                                    'total': 2,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=5664817468434011102&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:3tNvW3l5nU4J:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:3tNvW3l5nU4J:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'arxiv.org',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://arxiv.org/pdf/2107.03374.pdf?trk=public_post_comment-text'},\n",
+      "                      'authors': [{'name': 'M Chen',\n",
+      "                                   'id': '5fU-QMwAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=5fU-QMwAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'J Tworek',\n",
+      "                                   'id': 'ZPuESCQAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=ZPuESCQAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'Q Yuan',\n",
+      "                                   'id': 'B059m2EAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=B059m2EAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 7,\n",
+      "                      'title': 'Large language models in machine translation',\n",
+      "                      'data_cid': 'sY5m_Y3-0Y4J',\n",
+      "                      'link': 'http://research.google/pubs/pub33278.pdf',\n",
+      "                      'publication': 'T Brants, AC Popat, P Xu, FJ Och, J Dean '\n",
+      "                                     '- 2007 - research.google',\n",
+      "                      'snippet': '… the benefits of largescale statistical '\n",
+      "                                 'language modeling in ma… trillion tokens, '\n",
+      "                                 'resulting in language models having up to '\n",
+      "                                 '300 … is inexpensive to train on large data '\n",
+      "                                 'sets and approaches the …',\n",
+      "                      'type': 'PDF',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '10291286509313494705',\n",
+      "                                                    'total': 737,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=10291286509313494705&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '10291286509313494705',\n",
+      "                                                    'total': 31,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=10291286509313494705&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:sY5m_Y3-0Y4J:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:sY5m_Y3-0Y4J:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'research.google',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'http://research.google/pubs/pub33278.pdf'},\n",
+      "                      'authors': [{'name': 'FJ Och',\n",
+      "                                   'id': 'ITGdg6oAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=ITGdg6oAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'J Dean',\n",
+      "                                   'id': 'NMS69lQAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=NMS69lQAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 8,\n",
+      "                      'title': 'A watermark for large language models',\n",
+      "                      'data_cid': 'BlSyLHT4iiEJ',\n",
+      "                      'link': 'https://arxiv.org/abs/2301.10226',\n",
+      "                      'publication': 'J Kirchenbauer, J Geiping, Y Wen, J '\n",
+      "                                     'Katz… - arXiv preprint arXiv …, 2023 - '\n",
+      "                                     'arxiv.org',\n",
+      "                      'snippet': '… To derive this watermark, we examine what '\n",
+      "                                 'happens in the language model just before it '\n",
+      "                                 'produces a probability vector. The last '\n",
+      "                                 'layer of the language model outputs a vector '\n",
+      "                                 'of logits l(t). …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '2417017327887471622',\n",
+      "                                                    'total': 104,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=2417017327887471622&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '2417017327887471622',\n",
+      "                                                    'total': 4,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=2417017327887471622&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:BlSyLHT4iiEJ:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:BlSyLHT4iiEJ:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'arxiv.org',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://arxiv.org/pdf/2301.10226.pdf?curius=1419'},\n",
+      "                      'authors': [{'name': 'J Kirchenbauer',\n",
+      "                                   'id': '48GJrbsAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=48GJrbsAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'J Geiping',\n",
+      "                                   'id': '206vNCEAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=206vNCEAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'Y Wen',\n",
+      "                                   'id': 'oUYfjg0AAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=oUYfjg0AAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'J Katz',\n",
+      "                                   'id': 'yPw4WjoAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=yPw4WjoAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 9,\n",
+      "                      'title': 'ChatGPT and other large language models are '\n",
+      "                               'double-edged swords',\n",
+      "                      'data_cid': 'So0q8TRvxhYJ',\n",
+      "                      'link': 'https://pubs.rsna.org/doi/full/10.1148/radiol.230163',\n",
+      "                      'publication': 'Y Shen, L Heacock, J Elias, KD Hentel, B '\n",
+      "                                     'Reig, G Shih… - Radiology, 2023 - '\n",
+      "                                     'pubs.rsna.org',\n",
+      "                      'snippet': '… Large Language Models (LLMs) are deep '\n",
+      "                                 'learning models trained to understand and '\n",
+      "                                 'generate natural language. Recent studies '\n",
+      "                                 'demonstrated that LLMs achieve great success '\n",
+      "                                 'in a …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '1641121387398204746',\n",
+      "                                                    'total': 231,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=1641121387398204746&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '1641121387398204746',\n",
+      "                                                    'total': 3,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=1641121387398204746&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:So0q8TRvxhYJ:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'authors': [{'name': 'Y Shen',\n",
+      "                                   'id': 'XaeN2zgAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=XaeN2zgAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'L Heacock',\n",
+      "                                   'id': 'tYYM5IkAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=tYYM5IkAAAAJ&hl=en&oi=sra'}]},\n",
+      "                     {'position': 10,\n",
+      "                      'title': 'Pythia: A suite for analyzing large language '\n",
+      "                               'models across training and scaling',\n",
+      "                      'data_cid': 'aaIDvsMAD8QJ',\n",
+      "                      'link': 'https://proceedings.mlr.press/v202/biderman23a.html',\n",
+      "                      'publication': 'S Biderman, H Schoelkopf… - '\n",
+      "                                     'International …, 2023 - '\n",
+      "                                     'proceedings.mlr.press',\n",
+      "                      'snippet': '… large language models, we prioritize '\n",
+      "                                 'consistency in model … out the most '\n",
+      "                                 'performance from each model. For example, we '\n",
+      "                                 '… models, as it is becoming widely used for '\n",
+      "                                 'the largest models, …',\n",
+      "                      'inline_links': {'cited_by': {'cites_id': '14127511396791067241',\n",
+      "                                                    'total': 89,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cites=14127511396791067241&as_sdt=5,33&sciodt=0,33&hl=en'},\n",
+      "                                       'versions': {'cluster_id': '14127511396791067241',\n",
+      "                                                    'total': 3,\n",
+      "                                                    'link': 'https://scholar.google.com/scholar?cluster=14127511396791067241&hl=en&as_sdt=0,33'},\n",
+      "                                       'related_articles_link': 'https://scholar.google.com/scholar?q=related:aaIDvsMAD8QJ:scholar.google.com/&scioq=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                       'cached_page_link': 'https://scholar.googleusercontent.com/scholar?q=cache:aaIDvsMAD8QJ:scholar.google.com/+Large+Language+Models&hl=en&as_sdt=0,33'},\n",
+      "                      'resource': {'name': 'mlr.press',\n",
+      "                                   'format': 'PDF',\n",
+      "                                   'link': 'https://proceedings.mlr.press/v202/biderman23a/biderman23a.pdf'},\n",
+      "                      'authors': [{'name': 'S Biderman',\n",
+      "                                   'id': 'bO7H0DAAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=bO7H0DAAAAAJ&hl=en&oi=sra'},\n",
+      "                                  {'name': 'H Schoelkopf',\n",
+      "                                   'id': 'XLahYIYAAAAJ',\n",
+      "                                   'link': 'https://scholar.google.com/citations?user=XLahYIYAAAAJ&hl=en&oi=sra'}]}],\n",
+      " 'related_searches': [{'query': 'large language models machine',\n",
+      "                       'highlighted': ['machine'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=1&q=large+language+models+machine&qst=ib'},\n",
+      "                      {'query': 'large language models pruning',\n",
+      "                       'highlighted': ['pruning'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=2&q=large+language+models+pruning&qst=ib'},\n",
+      "                      {'query': 'large language models multitask learners',\n",
+      "                       'highlighted': ['multitask learners'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=3&q=large+language+models+multitask+learners&qst=ib'},\n",
+      "                      {'query': 'large language models speech recognition',\n",
+      "                       'highlighted': ['speech recognition'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=4&q=large+language+models+speech+recognition&qst=ib'},\n",
+      "                      {'query': 'large language models machine translation',\n",
+      "                       'highlighted': ['machine translation'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=5&q=large+language+models+machine+translation&qst=ib'},\n",
+      "                      {'query': 'emergent abilities of large language models',\n",
+      "                       'highlighted': ['emergent abilities of'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=6&q=emergent+abilities+of+large+language+models&qst=ir'},\n",
+      "                      {'query': 'language models privacy risks',\n",
+      "                       'highlighted': ['privacy risks'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=7&q=language+models+privacy+risks&qst=ir'},\n",
+      "                      {'query': 'language model fine tuning',\n",
+      "                       'highlighted': ['fine tuning'],\n",
+      "                       'link': 'https://scholar.google.com/scholar?hl=en&as_sdt=0,33&qsp=8&q=language+model+fine+tuning&qst=ir'}],\n",
+      " 'pagination': {'current': 1,\n",
+      "                'next': 'https://scholar.google.com/scholar?start=10&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                'other_pages': {'2': 'https://scholar.google.com/scholar?start=10&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '3': 'https://scholar.google.com/scholar?start=20&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '4': 'https://scholar.google.com/scholar?start=30&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '5': 'https://scholar.google.com/scholar?start=40&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '6': 'https://scholar.google.com/scholar?start=50&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '7': 'https://scholar.google.com/scholar?start=60&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '8': 'https://scholar.google.com/scholar?start=70&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '9': 'https://scholar.google.com/scholar?start=80&q=Large+Language+Models&hl=en&as_sdt=0,33',\n",
+      "                                '10': 'https://scholar.google.com/scholar?start=90&q=Large+Language+Models&hl=en&as_sdt=0,33'}}}\n"
+     ]
+    }
+   ],
+   "source": [
+    "search = SearchApiAPIWrapper(engine=\"google_scholar\")\n",
+    "results = search.results(\"Large Language Models\")\n",
+    "pprint.pp(results)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.12"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/integrations/vectorstores/elasticsearch.ipynb

@@ -44,7 +44,7 @@
    "source": [
     "There are two main ways to setup an Elasticsearch instance for use with:\n",
     "\n",
-    "1. Elastic Cloud: Elastic Cloud is a managed Elasticsearch service. Signup for a [free trial](https://cloud.elastic.co/registration?storm=langchain-notebook).\n",
+    "1. Elastic Cloud: Elastic Cloud is a managed Elasticsearch service. Signup for a [free trial](https://cloud.elastic.co/registration?utm_source=langchain&utm_content=documentation).\n",
     "\n",
     "To connect to an Elasticsearch instance that does not require\n",
     "login credentials (starting the docker instance with security enabled), pass the Elasticsearch URL and index name along with the\n",
@@ -662,7 +662,7 @@
    "id": "0960fa0a",
    "metadata": {},
    "source": [
-    "# Customise the Query\n",
+    "## Customise the Query\n",
     "With `custom_query` parameter at search, you are able to adjust the query that is used to retrieve documents from Elasticsearch. This is useful if you want to want to use a more complex query, to support linear boosting of fields."
    ]
   },
@@ -720,6 +720,35 @@
     "print(results[0])"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "3242fd42",
+   "metadata": {},
+   "source": [
+    "# FAQ\n",
+    "\n",
+    "## Question: Im getting timeout errors when indexing documents into Elasticsearch. How do I fix this?\n",
+    "One possible issue is your documents might take longer to index into Elasticsearch. ElasticsearchStore uses the Elasticsearch bulk API which has a few defaults that you can adjust to reduce the chance of timeout errors.\n",
+    "\n",
+    "This is also a good idea when you're using SparseVectorRetrievalStrategy.\n",
+    "\n",
+    "The defaults are:\n",
+    "- `chunk_size`: 500\n",
+    "- `max_chunk_bytes`: 100MB\n",
+    "\n",
+    "To adjust these, you can pass in the `chunk_size` and `max_chunk_bytes` parameters to the ElasticsearchStore `add_texts` method.\n",
+    "\n",
+    "```python\n",
+    "    vector_store.add_texts(\n",
+    "        texts,\n",
+    "        bulk_kwargs={\n",
+    "            \"chunk_size\": 50,\n",
+    "            \"max_chunk_bytes\": 200000000\n",
+    "        }\n",
+    "    )\n",
+    "```"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "604c66ea",

docs/extras/integrations/vectorstores/milvus.ipynb

@@ -140,12 +140,67 @@
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "id": "e40d558b",
    "metadata": {},
+   "source": [
+    "### Compartmentalize the data with Milvus Collections\n",
+    "\n",
+    "You can store different unrelated documents in different collections within same Milvus instance to maintain the context"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "82c00f6e",
+   "metadata": {},
+   "source": [
+    "Here's how you can create a new collection"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f7ff38ab",
+   "metadata": {},
    "outputs": [],
-   "source": []
+   "source": [
+    "vector_db = Milvus.from_documents(\n",
+    "    docs,\n",
+    "    embeddings,\n",
+    "    collection_name = 'collection_1',\n",
+    "    connection_args={\"host\": \"127.0.0.1\", \"port\": \"19530\"},\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "891cec1f",
+   "metadata": {},
+   "source": [
+    "And here is how you retrieve that stored collection"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e9e873e9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "vector_db = Milvus(\n",
+    "           embeddings,\n",
+    "           connection_args={\"host\": \"127.0.0.1\", \"port\": \"19530\"},\n",
+    "           collection_name = 'collection_1'\n",
+    "       )"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "9cc65535",
+   "metadata": {},
+   "source": [
+    "After retreival you can go on querying it as usual."
+   ]
   }
  ],
  "metadata": {

docs/extras/integrations/vectorstores/neo4jvector.ipynb

@@ -107,7 +107,7 @@
     "password = \"pleaseletmein\"\n",
     "\n",
     "# You can also use environment variables instead of directly passing named parameters\n",
-    "#os.environ[\"NEO4J_URL\"] = \"bolt://localhost:7687\"\n",
+    "#os.environ[\"NEO4J_URI\"] = \"bolt://localhost:7687\"\n",
     "#os.environ[\"NEO4J_USERNAME\"] = \"neo4j\"\n",
     "#os.environ[\"NEO4J_PASSWORD\"] = \"pleaseletmein\""
    ]
@@ -123,7 +123,16 @@
    "cell_type": "code",
    "execution_count": 6,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/tomaz/neo4j/langchain/libs/langchain/langchain/vectorstores/neo4j_vector.py:165: ExperimentalWarning: The configuration may change in the future.\n",
+      "  self._driver.verify_connectivity()\n"
+     ]
+    }
+   ],
    "source": [
     "# The Neo4jVector Module will connect to Neo4j and create a vector index if needed.\n",
     "\n",
@@ -139,7 +148,7 @@
    "outputs": [],
    "source": [
     "query = \"What did the president say about Ketanji Brown Jackson\"\n",
-    "docs_with_score = db.similarity_search_with_score(query)"
+    "docs_with_score = db.similarity_search_with_score(query, k=2)"
    ]
   },
   {
@@ -152,7 +161,7 @@
      "output_type": "stream",
      "text": [
       "--------------------------------------------------------------------------------\n",
-      "Score:  0.9077161550521851\n",
+      "Score:  0.9099836349487305\n",
       "Tonight. I call on the Senate to: Pass the Freedom to Vote Act. Pass the John Lewis Voting Rights Act. And while you’re at it, pass the Disclose Act so Americans can know who is funding our elections. \n",
       "\n",
       "Tonight, I’d like to honor someone who has dedicated his life to serve this country: Justice Stephen Breyer—an Army veteran, Constitutional scholar, and retiring Justice of the United States Supreme Court. Justice Breyer, thank you for your service. \n",
@@ -162,50 +171,14 @@
       "And I did that 4 days ago, when I nominated Circuit Court of Appeals Judge Ketanji Brown Jackson. One of our nation’s top legal minds, who will continue Justice Breyer’s legacy of excellence.\n",
       "--------------------------------------------------------------------------------\n",
       "--------------------------------------------------------------------------------\n",
-      "Score:  0.891287088394165\n",
-      "A former top litigator in private practice. A former federal public defender. And from a family of public school educators and police officers. A consensus builder. Since she’s been nominated, she’s received a broad range of support—from the Fraternal Order of Police to former judges appointed by Democrats and Republicans. \n",
+      "Score:  0.9099686145782471\n",
+      "Tonight. I call on the Senate to: Pass the Freedom to Vote Act. Pass the John Lewis Voting Rights Act. And while you’re at it, pass the Disclose Act so Americans can know who is funding our elections. \n",
       "\n",
-      "And if we are to advance liberty and justice, we need to secure the Border and fix the immigration system. \n",
+      "Tonight, I’d like to honor someone who has dedicated his life to serve this country: Justice Stephen Breyer—an Army veteran, Constitutional scholar, and retiring Justice of the United States Supreme Court. Justice Breyer, thank you for your service. \n",
       "\n",
-      "We can do both. At our border, we’ve installed new technology like cutting-edge scanners to better detect drug smuggling.  \n",
+      "One of the most serious constitutional responsibilities a President has is nominating someone to serve on the United States Supreme Court. \n",
       "\n",
-      "We’ve set up joint patrols with Mexico and Guatemala to catch more human traffickers.  \n",
-      "\n",
-      "We’re putting in place dedicated immigration judges so families fleeing persecution and violence can have their cases heard faster. \n",
-      "\n",
-      "We’re securing commitments and supporting partners in South and Central America to host more refugees and secure their own borders.\n",
-      "--------------------------------------------------------------------------------\n",
-      "--------------------------------------------------------------------------------\n",
-      "Score:  0.8867912292480469\n",
-      "And for our LGBTQ+ Americans, let’s finally get the bipartisan Equality Act to my desk. The onslaught of state laws targeting transgender Americans and their families is wrong. \n",
-      "\n",
-      "As I said last year, especially to our younger transgender Americans, I will always have your back as your President, so you can be yourself and reach your God-given potential. \n",
-      "\n",
-      "While it often appears that we never agree, that isn’t true. I signed 80 bipartisan bills into law last year. From preventing government shutdowns to protecting Asian-Americans from still-too-common hate crimes to reforming military justice. \n",
-      "\n",
-      "And soon, we’ll strengthen the Violence Against Women Act that I first wrote three decades ago. It is important for us to show the nation that we can come together and do big things. \n",
-      "\n",
-      "So tonight I’m offering a Unity Agenda for the Nation. Four big things we can do together.  \n",
-      "\n",
-      "First, beat the opioid epidemic.\n",
-      "--------------------------------------------------------------------------------\n",
-      "--------------------------------------------------------------------------------\n",
-      "Score:  0.8866499662399292\n",
-      "Tonight, I’m announcing a crackdown on these companies overcharging American businesses and consumers. \n",
-      "\n",
-      "And as Wall Street firms take over more nursing homes, quality in those homes has gone down and costs have gone up.  \n",
-      "\n",
-      "That ends on my watch. \n",
-      "\n",
-      "Medicare is going to set higher standards for nursing homes and make sure your loved ones get the care they deserve and expect. \n",
-      "\n",
-      "We’ll also cut costs and keep the economy going strong by giving workers a fair shot, provide more training and apprenticeships, hire them based on their skills not degrees. \n",
-      "\n",
-      "Let’s pass the Paycheck Fairness Act and paid leave.  \n",
-      "\n",
-      "Raise the minimum wage to $15 an hour and extend the Child Tax Credit, so no one has to raise a family in poverty. \n",
-      "\n",
-      "Let’s increase Pell Grants and increase our historic support of HBCUs, and invest in what Jill—our First Lady who teaches full-time—calls America’s best-kept secret: community colleges.\n",
+      "And I did that 4 days ago, when I nominated Circuit Court of Appeals Judge Ketanji Brown Jackson. One of our nation’s top legal minds, who will continue Justice Breyer’s legacy of excellence.\n",
       "--------------------------------------------------------------------------------\n"
      ]
     }
@@ -232,7 +205,16 @@
    "cell_type": "code",
    "execution_count": 9,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/tomaz/neo4j/langchain/libs/langchain/langchain/vectorstores/neo4j_vector.py:165: ExperimentalWarning: The configuration may change in the future.\n",
+      "  self._driver.verify_connectivity()\n"
+     ]
+    }
+   ],
    "source": [
     "index_name = \"vector\"  # default index name\n",
     "\n",
@@ -249,8 +231,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Add documents\n",
-    "We can add documents to the existing vectorstore."
+    "We can also initialize a vectorstore from existing graph using the `from_existing_graph` method. This method pulls relevant text information from the database, and calculates and stores the text embeddings back to the database."
    ]
   },
   {
@@ -261,7 +242,7 @@
     {
      "data": {
       "text/plain": [
-       "['064c7032-5093-11ee-8041-3b350f274873']"
+       "[]"
       ]
      },
      "execution_count": 10,
@@ -269,13 +250,93 @@
      "output_type": "execute_result"
     }
    ],
+   "source": [
+    "# First we create sample data in graph\n",
+    "store.query(\n",
+    "    \"CREATE (p:Person {name: 'Tomaz', location:'Slovenia', hobby:'Bicycle'})\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/tomaz/neo4j/langchain/libs/langchain/langchain/vectorstores/neo4j_vector.py:165: ExperimentalWarning: The configuration may change in the future.\n",
+      "  self._driver.verify_connectivity()\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Now we initialize from existing graph\n",
+    "existing_graph = Neo4jVector.from_existing_graph(\n",
+    "        embedding=OpenAIEmbeddings(),\n",
+    "        url=url,\n",
+    "        username=username,\n",
+    "        password=password,\n",
+    "        index_name=\"person_index\",\n",
+    "        node_label=\"Person\",\n",
+    "        text_node_properties=[\"name\", \"location\"],\n",
+    "        embedding_node_property=\"embedding\",\n",
+    "    )\n",
+    "result = existing_graph.similarity_search(\"Slovenia\", k = 1)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Document(page_content='\\nname: Tomaz\\nlocation: Slovenia', metadata={'hobby': 'Bicycle'})"
+      ]
+     },
+     "execution_count": 12,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "result[0]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Add documents\n",
+    "We can add documents to the existing vectorstore."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "['187fc53a-5dde-11ee-ad78-1f6b05bf8513']"
+      ]
+     },
+     "execution_count": 13,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "store.add_documents([Document(page_content=\"foo\")])"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -284,7 +345,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 15,
    "metadata": {
     "scrolled": true
    },
@@ -295,7 +356,7 @@
        "(Document(page_content='foo', metadata={}), 1.0)"
       ]
      },
-     "execution_count": 12,
+     "execution_count": 15,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -315,9 +376,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 16,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/tomaz/neo4j/langchain/libs/langchain/langchain/vectorstores/neo4j_vector.py:165: ExperimentalWarning: The configuration may change in the future.\n",
+      "  self._driver.verify_connectivity()\n"
+     ]
+    }
+   ],
    "source": [
     "# The Neo4jVector Module will connect to Neo4j and create a vector and keyword indices if needed.\n",
     "hybrid_db = Neo4jVector.from_documents(\n",
@@ -339,9 +409,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 17,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/tomaz/neo4j/langchain/libs/langchain/langchain/vectorstores/neo4j_vector.py:165: ExperimentalWarning: The configuration may change in the future.\n",
+      "  self._driver.verify_connectivity()\n"
+     ]
+    }
+   ],
    "source": [
     "index_name = \"vector\"  # default index name\n",
     "keyword_index_name = \"keyword\" #default keyword index name\n",
@@ -368,7 +447,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 18,
    "metadata": {},
    "outputs": [
     {
@@ -377,7 +456,7 @@
        "Document(page_content='Tonight. I call on the Senate to: Pass the Freedom to Vote Act. Pass the John Lewis Voting Rights Act. And while you’re at it, pass the Disclose Act so Americans can know who is funding our elections. \\n\\nTonight, I’d like to honor someone who has dedicated his life to serve this country: Justice Stephen Breyer—an Army veteran, Constitutional scholar, and retiring Justice of the United States Supreme Court. Justice Breyer, thank you for your service. \\n\\nOne of the most serious constitutional responsibilities a President has is nominating someone to serve on the United States Supreme Court. \\n\\nAnd I did that 4 days ago, when I nominated Circuit Court of Appeals Judge Ketanji Brown Jackson. One of our nation’s top legal minds, who will continue Justice Breyer’s legacy of excellence.', metadata={'source': '../../modules/state_of_the_union.txt'})"
       ]
      },
-     "execution_count": 15,
+     "execution_count": 18,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -398,7 +477,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 19,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -408,7 +487,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 20,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -419,17 +498,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": 21,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "{'answer': \"The president honored Justice Stephen Breyer, who is retiring from the United States Supreme Court, and thanked him for his service. The president also mentioned that he nominated Circuit Court of Appeals Judge Ketanji Brown Jackson to continue Justice Breyer's legacy of excellence. \\n\",\n",
+       "{'answer': \"The president honored Justice Stephen Breyer, who is retiring from the United States Supreme Court. He thanked him for his service and mentioned that he nominated Circuit Court of Appeals Judge Ketanji Brown Jackson to continue Justice Breyer's legacy of excellence. \\n\",\n",
        " 'sources': '../../modules/state_of_the_union.txt'}"
       ]
      },
-     "execution_count": 18,
+     "execution_count": 21,
      "metadata": {},
      "output_type": "execute_result"
     }

docs/extras/integrations/vectorstores/supabase.ipynb

@@ -92,7 +92,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 15,
    "id": "19846a7b-99bc-47a7-8e1c-f13c2497f1ae",
    "metadata": {},
    "outputs": [],
@@ -105,7 +105,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 16,
    "id": "c71c3901-d44b-4d09-92c5-3018628c28fa",
    "metadata": {},
    "outputs": [],
@@ -115,7 +115,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 17,
    "id": "8b91ecfa-f61b-489a-a337-dff1f12f6ab2",
    "metadata": {},
    "outputs": [],
@@ -138,51 +138,66 @@
     "load_dotenv()"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "924d4df5",
+   "metadata": {},
+   "source": [
+    "First we'll create a Supabase client and instantiate a OpenAI embeddings class."
+   ]
+  },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 19,
    "id": "5ce44f7c",
    "metadata": {},
    "outputs": [],
    "source": [
     "import os\n",
     "from supabase.client import Client, create_client\n",
+    "from langchain.embeddings.openai import OpenAIEmbeddings\n",
+    "from langchain.vectorstores import SupabaseVectorStore\n",
     "\n",
     "supabase_url = os.environ.get(\"SUPABASE_URL\")\n",
     "supabase_key = os.environ.get(\"SUPABASE_SERVICE_KEY\")\n",
-    "supabase: Client = create_client(supabase_url, supabase_key)"
+    "supabase: Client = create_client(supabase_url, supabase_key)\n",
+    "\n",
+    "embeddings = OpenAIEmbeddings()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "0c707d4c",
+   "metadata": {},
+   "source": [
+    "Next we'll load and parse some data for our vector store (skip if you already have documents with embeddings stored in your DB)."
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 20,
    "id": "aac9563e",
    "metadata": {
     "tags": []
    },
    "outputs": [],
    "source": [
-    "from langchain.embeddings.openai import OpenAIEmbeddings\n",
+    "\n",
     "from langchain.text_splitter import CharacterTextSplitter\n",
-    "from langchain.vectorstores import SupabaseVectorStore\n",
-    "from langchain.document_loaders import TextLoader"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "a3c3999a",
-   "metadata": {},
-   "outputs": [],
-   "source": [
     "from langchain.document_loaders import TextLoader\n",
     "\n",
     "loader = TextLoader(\"../../../state_of_the_union.txt\")\n",
     "documents = loader.load()\n",
     "text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)\n",
-    "docs = text_splitter.split_documents(documents)\n",
-    "\n",
-    "embeddings = OpenAIEmbeddings()"
+    "docs = text_splitter.split_documents(documents)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "5abb9b93",
+   "metadata": {},
+   "source": [
+    "Insert the above documents into the database. Embeddings will automatically be generated for each document."
    ]
   },
   {
@@ -192,13 +207,39 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# We're using the default `documents` table here. You can modify this by passing in a `table_name` argument to the `from_documents` method.\n",
-    "vector_store = SupabaseVectorStore.from_documents(docs, embeddings, client=supabase)"
+    "\n",
+    "vector_store = SupabaseVectorStore.from_documents(docs, embeddings, client=supabase, table_name=\"documents\", query_name=\"match_documents\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "e169345d",
+   "metadata": {},
+   "source": [
+    "Alternatively if you already have documents with embeddings in your database, simply instantiate a new `SupabaseVectorStore` directly:"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 10,
+   "id": "397e3e7d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "vector_store = SupabaseVectorStore(embedding=embeddings, client=supabase, table_name=\"documents\", query_name=\"match_documents\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "e28ce092",
+   "metadata": {},
+   "source": [
+    "Finally, test it out by performing a similarity search:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
    "id": "5eabdb75",
    "metadata": {},
    "outputs": [],
@@ -209,7 +250,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "4b172de8",
    "metadata": {},
    "outputs": [
@@ -431,7 +472,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,

docs/extras/modules/agents/agent_types/chat_conversation_agent.ipynb

@@ -0,0 +1,561 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "69014601",
+   "metadata": {},
+   "source": [
+    "# Conversational\n",
+    "\n",
+    "This walkthrough demonstrates how to use an agent optimized for conversation. Other agents are often optimized for using tools to figure out the best response, which is not ideal in a conversational setting where you may want the agent to be able to chat with the user as well.\n",
+    "\n",
+    "If we compare it to the standard ReAct agent, the main difference is the prompt.\n",
+    "We want it to be much more conversational."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "cc3fad9e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import Tool\n",
+    "from langchain.agents import AgentType\n",
+    "from langchain.memory import ConversationBufferMemory\n",
+    "from langchain.llms import OpenAI\n",
+    "from langchain.utilities import SerpAPIWrapper\n",
+    "from langchain.agents import initialize_agent"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "2d84b9bc",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "search = SerpAPIWrapper()\n",
+    "tools = [\n",
+    "    Tool(\n",
+    "        name = \"Current Search\",\n",
+    "        func=search.run,\n",
+    "        description=\"useful for when you need to answer questions about current events or the current state of the world\"\n",
+    "    ),\n",
+    "]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "799a31bf",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm=OpenAI(temperature=0)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "f9d11cb6",
+   "metadata": {},
+   "source": [
+    "## Using LCEL\n",
+    "\n",
+    "We will first show how to create this agent using LCEL"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "03c09ef9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.tools.render import render_text_description\n",
+    "from langchain.agents.output_parsers import ReActSingleInputOutputParser\n",
+    "from langchain.agents.format_scratchpad import format_log_to_str\n",
+    "from langchain import hub"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 28,
+   "id": "6bd84102",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = hub.pull(\"hwchase17/react-chat\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 29,
+   "id": "7ccc785d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = prompt.partial(\n",
+    "    tools=render_text_description(tools),\n",
+    "    tool_names=\", \".join([t.name for t in tools]),\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "d7aac2b0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm_with_stop = llm.bind(stop=[\"\\nObservation\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 15,
+   "id": "a028bca6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent = {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"agent_scratchpad\": lambda x: format_log_to_str(x['intermediate_steps']),\n",
+    "    \"chat_history\": lambda x: x[\"chat_history\"]\n",
+    "} | prompt | llm_with_stop | ReActSingleInputOutputParser()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0b354cfe",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import AgentExecutor"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 23,
+   "id": "9b044ae9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "memory = ConversationBufferMemory(memory_key=\"chat_history\")\n",
+    "agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, memory=memory)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 24,
+   "id": "adcdd0c7",
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m\n",
+      "Thought: Do I need to use a tool? No\n",
+      "Final Answer: Hi Bob, nice to meet you! How can I help you today?\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'Hi Bob, nice to meet you! How can I help you today?'"
+      ]
+     },
+     "execution_count": 24,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"hi, i am bob\"})['output']"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 25,
+   "id": "c5846cd1",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m\n",
+      "Thought: Do I need to use a tool? No\n",
+      "Final Answer: Your name is Bob.\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'Your name is Bob.'"
+      ]
+     },
+     "execution_count": 25,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"whats my name?\"})['output']"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 26,
+   "id": "95a1192a",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m\n",
+      "Thought: Do I need to use a tool? Yes\n",
+      "Action: Current Search\n",
+      "Action Input: Movies showing 9/21/2023\u001b[0m\u001b[36;1m\u001b[1;3m['September 2023 Movies: The Creator • Dumb Money • Expend4bles • The Kill Room • The Inventor • The Equalizer 3 • PAW Patrol: The Mighty Movie, ...']\u001b[0m\u001b[32;1m\u001b[1;3m Do I need to use a tool? No\n",
+      "Final Answer: According to current search, some movies showing on 9/21/2023 are The Creator, Dumb Money, Expend4bles, The Kill Room, The Inventor, The Equalizer 3, and PAW Patrol: The Mighty Movie.\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'According to current search, some movies showing on 9/21/2023 are The Creator, Dumb Money, Expend4bles, The Kill Room, The Inventor, The Equalizer 3, and PAW Patrol: The Mighty Movie.'"
+      ]
+     },
+     "execution_count": 26,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"what are some movies showing 9/21/2023?\"})['output']"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c0b2d86d",
+   "metadata": {},
+   "source": [
+    "## Use the off-the-shelf agent\n",
+    "\n",
+    "We can also create this agent using the off-the-shelf agent class"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
+   "id": "53e43064",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = initialize_agent(tools, llm, agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION, verbose=True, memory=memory)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "68e45a24",
+   "metadata": {},
+   "source": [
+    "## Use a chat model\n",
+    "\n",
+    "We can also use a chat model here. The main difference here is in the prompts used."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a5a705b2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from langchain import hub"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "16b17ca8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = hub.pull(\"hwchase17/react-chat-json\")\n",
+    "chat_model = ChatOpenAI(temperature=0, model='gpt-4')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 24,
+   "id": "c8a94b0b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = prompt.partial(\n",
+    "    tools=render_text_description(tools),\n",
+    "    tool_names=\", \".join([t.name for t in tools]),\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 25,
+   "id": "c5d710f2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "chat_model_with_stop = chat_model.bind(stop=[\"\\nObservation\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f50a5ea8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents.output_parsers import JSONAgentOutputParser\n",
+    "from langchain.agents.format_scratchpad import format_log_to_messages"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 26,
+   "id": "2c845796",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# We need some extra steering, or the chat model forgets how to respond sometimes\n",
+    "TEMPLATE_TOOL_RESPONSE = \"\"\"TOOL RESPONSE: \n",
+    "---------------------\n",
+    "{observation}\n",
+    "\n",
+    "USER'S INPUT\n",
+    "--------------------\n",
+    "\n",
+    "Okay, so what is the response to my last comment? If using information obtained from the tools you must mention it explicitly without mentioning the tool names - I have forgotten all TOOL RESPONSES! Remember to respond with a markdown code snippet of a json blob with a single action, and NOTHING else - even if you just want to respond to the user. Do NOT respond with anything except a JSON snippet no matter what!\"\"\"\n",
+    "\n",
+    "agent = {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"agent_scratchpad\": lambda x: format_log_to_messages(x['intermediate_steps'], template_tool_response=TEMPLATE_TOOL_RESPONSE),\n",
+    "    \"chat_history\": lambda x: x[\"chat_history\"],\n",
+    "} | prompt | chat_model_with_stop | JSONAgentOutputParser()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6cc033fc",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import AgentExecutor"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
+   "id": "332ba2ff",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "memory = ConversationBufferMemory(memory_key=\"chat_history\", return_messages=True)\n",
+    "agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, memory=memory)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 28,
+   "id": "139717b4",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m```json\n",
+      "{\n",
+      "    \"action\": \"Final Answer\",\n",
+      "    \"action_input\": \"Hello Bob, how can I assist you today?\"\n",
+      "}\n",
+      "```\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'Hello Bob, how can I assist you today?'"
+      ]
+     },
+     "execution_count": 28,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"hi, i am bob\"})['output']"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 29,
+   "id": "7e7cf6d3",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m```json\n",
+      "{\n",
+      "    \"action\": \"Final Answer\",\n",
+      "    \"action_input\": \"Your name is Bob.\"\n",
+      "}\n",
+      "```\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'Your name is Bob.'"
+      ]
+     },
+     "execution_count": 29,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"whats my name?\"})['output']"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 30,
+   "id": "3fc00073",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m```json\n",
+      "{\n",
+      "    \"action\": \"Current Search\",\n",
+      "    \"action_input\": \"movies showing on 9/21/2023\"\n",
+      "}\n",
+      "```\u001b[0m\u001b[36;1m\u001b[1;3m['September 2023 Movies: The Creator • Dumb Money • Expend4bles • The Kill Room • The Inventor • The Equalizer 3 • PAW Patrol: The Mighty Movie, ...']\u001b[0m\u001b[32;1m\u001b[1;3m```json\n",
+      "{\n",
+      "    \"action\": \"Final Answer\",\n",
+      "    \"action_input\": \"Some movies that are showing on 9/21/2023 include 'The Creator', 'Dumb Money', 'Expend4bles', 'The Kill Room', 'The Inventor', 'The Equalizer 3', and 'PAW Patrol: The Mighty Movie'.\"\n",
+      "}\n",
+      "```\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "\"Some movies that are showing on 9/21/2023 include 'The Creator', 'Dumb Money', 'Expend4bles', 'The Kill Room', 'The Inventor', 'The Equalizer 3', and 'PAW Patrol: The Mighty Movie'.\""
+      ]
+     },
+     "execution_count": 30,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"what are some movies showing 9/21/2023?\"})['output']"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8d464ead",
+   "metadata": {},
+   "source": [
+    "We can also initialize the agent executor with a predefined agent type"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "141f2469",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.memory import ConversationBufferMemory\n",
+    "from langchain.chat_models import ChatOpenAI"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "734d1b21",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "memory = ConversationBufferMemory(memory_key=\"chat_history\", return_messages=True)\n",
+    "llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY, temperature=0)\n",
+    "agent_chain = initialize_agent(tools, llm, agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION, verbose=True, memory=memory)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.1"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/modules/agents/agent_types/openai_functions_agent.ipynb

@@ -0,0 +1,295 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "e10aa932",
+   "metadata": {},
+   "source": [
+    "# OpenAI functions\n",
+    "\n",
+    "Certain OpenAI models (like gpt-3.5-turbo-0613 and gpt-4-0613) have been fine-tuned to detect when a function should be called and respond with the inputs that should be passed to the function. In an API call, you can describe functions and have the model intelligently choose to output a JSON object containing arguments to call those functions. The goal of the OpenAI Function APIs is to more reliably return valid and useful function calls than a generic text completion or chat API.\n",
+    "\n",
+    "The OpenAI Functions Agent is designed to work with these models.\n",
+    "\n",
+    "Install `openai`, `google-search-results` packages which are required as the LangChain packages call them internally."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ec89be68",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "! pip install openai google-search-results"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "82787d8d",
+   "metadata": {},
+   "source": [
+    "## Initialize tools\n",
+    "\n",
+    "We will first create some tools we can use"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "b812b982",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import initialize_agent, AgentType, Tool\n",
+    "from langchain.chains import LLMMathChain\n",
+    "from langchain.chat_models import ChatOpenAI\n",
+    "from langchain.llms import OpenAI\n",
+    "from langchain.utilities import SerpAPIWrapper, SQLDatabase\n",
+    "from langchain_experimental.sql import SQLDatabaseChain"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "23fc0aa6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm = ChatOpenAI(temperature=0, model=\"gpt-3.5-turbo-0613\")\n",
+    "search = SerpAPIWrapper()\n",
+    "llm_math_chain = LLMMathChain.from_llm(llm=llm, verbose=True)\n",
+    "db = SQLDatabase.from_uri(\"sqlite:///../../../../../notebooks/Chinook.db\")\n",
+    "db_chain = SQLDatabaseChain.from_llm(llm, db, verbose=True)\n",
+    "tools = [\n",
+    "    Tool(\n",
+    "        name = \"Search\",\n",
+    "        func=search.run,\n",
+    "        description=\"useful for when you need to answer questions about current events. You should ask targeted questions\"\n",
+    "    ),\n",
+    "    Tool(\n",
+    "        name=\"Calculator\",\n",
+    "        func=llm_math_chain.run,\n",
+    "        description=\"useful for when you need to answer questions about math\"\n",
+    "    ),\n",
+    "    Tool(\n",
+    "        name=\"FooBar-DB\",\n",
+    "        func=db_chain.run,\n",
+    "        description=\"useful for when you need to answer questions about FooBar. Input should be in the form of a question containing full context\"\n",
+    "    )\n",
+    "]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "39c3ba21",
+   "metadata": {},
+   "source": [
+    "## Using LCEL\n",
+    "\n",
+    "We will first use LangChain Expression Language to create this agent"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "eac103f1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "55292bed",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = ChatPromptTemplate.from_messages([\n",
+    "    (\"system\", \"You are a helpful assistant\"),\n",
+    "    (\"user\", \"{input}\"),\n",
+    "    MessagesPlaceholder(variable_name=\"agent_scratchpad\"),\n",
+    "])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "50f40df4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.tools.render import format_tool_to_openai_function"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "552421b3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm_with_tools = llm.bind(\n",
+    "    functions=[format_tool_to_openai_function(t) for t in tools]\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3cafa0a3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents.format_scratchpad import format_to_openai_functions\n",
+    "from langchain.agents.output_parsers import OpenAIFunctionsAgentOutputParser"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "bf514eb4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent = {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"agent_scratchpad\": lambda x: format_to_openai_functions(x['intermediate_steps'])\n",
+    "} | prompt | llm_with_tools | OpenAIFunctionsAgentOutputParser()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5125573e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import AgentExecutor"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "bdc7e506",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "2cd65218",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m\n",
+      "Invoking: `Search` with `Leo DiCaprio's girlfriend`\n",
+      "\n",
+      "\n",
+      "\u001b[0m\u001b[36;1m\u001b[1;3m['Blake Lively and DiCaprio are believed to have enjoyed a whirlwind five-month romance in 2011. The pair were seen on a yacht together in Cannes, ...']\u001b[0m\u001b[32;1m\u001b[1;3m\n",
+      "Invoking: `Calculator` with `0.43`\n",
+      "\n",
+      "\n",
+      "\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Entering new LLMMathChain chain...\u001b[0m\n",
+      "0.43\u001b[32;1m\u001b[1;3m```text\n",
+      "0.43\n",
+      "```\n",
+      "...numexpr.evaluate(\"0.43\")...\n",
+      "\u001b[0m\n",
+      "Answer: \u001b[33;1m\u001b[1;3m0.43\u001b[0m\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n",
+      "\u001b[33;1m\u001b[1;3mAnswer: 0.43\u001b[0m\u001b[32;1m\u001b[1;3mI'm sorry, but I couldn't find any information about Leo DiCaprio's current girlfriend. As for raising her age to the power of 0.43, I'm not sure what her current age is, so I can't provide an answer for that.\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'input': \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\",\n",
+       " 'output': \"I'm sorry, but I couldn't find any information about Leo DiCaprio's current girlfriend. As for raising her age to the power of 0.43, I'm not sure what her current age is, so I can't provide an answer for that.\"}"
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\"})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8e91393f",
+   "metadata": {},
+   "source": [
+    "## Using OpenAIFunctionsAgent\n",
+    "\n",
+    "We can now use `OpenAIFunctionsAgent`, which creates this agent under the hood"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "9ed07c8f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = initialize_agent(tools, llm, agent=AgentType.OPENAI_FUNCTIONS, verbose=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8d9fb674",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor.invoke({\"input\": \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\"})"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2bc581dc",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.1"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/modules/agents/agent_types/openai_multi_functions_agent.ipynb

@@ -444,9 +444,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "venv",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "venv"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -458,7 +458,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.3"
+   "version": "3.10.1"
   }
  },
  "nbformat": 4,

docs/extras/modules/agents/agent_types/react.ipynb

@@ -0,0 +1,391 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "d82e62ec",
+   "metadata": {},
+   "source": [
+    "# ReAct\n",
+    "\n",
+    "This walkthrough showcases using an agent to implement the [ReAct](https://react-lm.github.io/) logic."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "102b0e52",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import load_tools\n",
+    "from langchain.agents import initialize_agent\n",
+    "from langchain.agents import AgentType\n",
+    "from langchain.llms import OpenAI"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "e0c9c056",
+   "metadata": {},
+   "source": [
+    "First, let's load the language model we're going to use to control the agent."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "184f0682",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm = OpenAI(temperature=0)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "2e67a000",
+   "metadata": {},
+   "source": [
+    "Next, let's load some tools to use. Note that the `llm-math` tool uses an LLM, so we need to pass that in."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "256408d5",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "tools = load_tools([\"serpapi\", \"llm-math\"], llm=llm)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "b7d04f53",
+   "metadata": {},
+   "source": [
+    "## Using LCEL\n",
+    "\n",
+    "We will first show how to create the agent using LCEL"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "bb0813a3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.tools.render import render_text_description\n",
+    "from langchain.agents.output_parsers import ReActSingleInputOutputParser\n",
+    "from langchain.agents.format_scratchpad import format_log_to_str\n",
+    "from langchain import hub"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "d3ae5fcd",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = hub.pull(\"hwchase17/react\")\n",
+    "prompt = prompt.partial(\n",
+    "    tools=render_text_description(tools),\n",
+    "    tool_names=\", \".join([t.name for t in tools]),\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "bf47a3c7",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm_with_stop = llm.bind(stop=[\"\\nObservation\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "b3d3958b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent = {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"agent_scratchpad\": lambda x: format_log_to_str(x['intermediate_steps'])\n",
+    "} | prompt | llm_with_stop | ReActSingleInputOutputParser()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a0a57769",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import AgentExecutor"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "026de6cd",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "57780ce1",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m I need to find out who Leo DiCaprio's girlfriend is and then calculate her age raised to the 0.43 power.\n",
+      "Action: Search\n",
+      "Action Input: \"Leo DiCaprio girlfriend\"\u001b[0m\u001b[36;1m\u001b[1;3mmodel Vittoria Ceretti\u001b[0m\u001b[32;1m\u001b[1;3m I need to find out Vittoria Ceretti's age\n",
+      "Action: Search\n",
+      "Action Input: \"Vittoria Ceretti age\"\u001b[0m\u001b[36;1m\u001b[1;3m25 years\u001b[0m\u001b[32;1m\u001b[1;3m I need to calculate 25 raised to the 0.43 power\n",
+      "Action: Calculator\n",
+      "Action Input: 25^0.43\u001b[0m\u001b[33;1m\u001b[1;3mAnswer: 3.991298452658078\u001b[0m\u001b[32;1m\u001b[1;3m I now know the final answer\n",
+      "Final Answer: Leo DiCaprio's girlfriend is Vittoria Ceretti and her current age raised to the 0.43 power is 3.991298452658078.\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'input': \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\",\n",
+       " 'output': \"Leo DiCaprio's girlfriend is Vittoria Ceretti and her current age raised to the 0.43 power is 3.991298452658078.\"}"
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\"})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "b4a33ea8",
+   "metadata": {},
+   "source": [
+    "## Using ZeroShotReactAgent\n",
+    "\n",
+    "We will now show how to use the agent with an off-the-shelf agent implementation"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "9752e90e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "04c5bcf6",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m I need to find out who Leo DiCaprio's girlfriend is and then calculate her age raised to the 0.43 power.\n",
+      "Action: Search\n",
+      "Action Input: \"Leo DiCaprio girlfriend\"\u001b[0m\n",
+      "Observation: \u001b[36;1m\u001b[1;3mmodel Vittoria Ceretti\u001b[0m\n",
+      "Thought:\u001b[32;1m\u001b[1;3m I need to find out Vittoria Ceretti's age\n",
+      "Action: Search\n",
+      "Action Input: \"Vittoria Ceretti age\"\u001b[0m\n",
+      "Observation: \u001b[36;1m\u001b[1;3m25 years\u001b[0m\n",
+      "Thought:\u001b[32;1m\u001b[1;3m I need to calculate 25 raised to the 0.43 power\n",
+      "Action: Calculator\n",
+      "Action Input: 25^0.43\u001b[0m\n",
+      "Observation: \u001b[33;1m\u001b[1;3mAnswer: 3.991298452658078\u001b[0m\n",
+      "Thought:\u001b[32;1m\u001b[1;3m I now know the final answer\n",
+      "Final Answer: Leo DiCaprio's girlfriend is Vittoria Ceretti and her current age raised to the 0.43 power is 3.991298452658078.\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'input': \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\",\n",
+       " 'output': \"Leo DiCaprio's girlfriend is Vittoria Ceretti and her current age raised to the 0.43 power is 3.991298452658078.\"}"
+      ]
+     },
+     "execution_count": 11,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\"})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "7f3e8fc8",
+   "metadata": {},
+   "source": [
+    "## Using chat models\n",
+    "\n",
+    "You can also create ReAct agents that use chat models instead of LLMs as the agent driver.\n",
+    "\n",
+    "The main difference here is a different prompt. We will use JSON to encode the agent's actions (chat models are a bit tougher to steet, so using JSON helps to enforce the output format)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6eeb1693",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.chat_models import ChatOpenAI"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 29,
+   "id": "fe846c48",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "chat_model = ChatOpenAI(temperature=0)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
+   "id": "0843590d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = hub.pull(\"hwchase17/react-json\")\n",
+    "prompt = prompt.partial(\n",
+    "    tools=render_text_description(tools),\n",
+    "    tool_names=\", \".join([t.name for t in tools]),\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 30,
+   "id": "a863b763",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "chat_model_with_stop = chat_model.bind(stop=[\"\\nObservation\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "deaeb1f6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents.output_parsers import ReActJsonSingleInputOutputParser"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 31,
+   "id": "6336a378",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent = {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    \"agent_scratchpad\": lambda x: format_log_to_str(x['intermediate_steps'])\n",
+    "} | prompt | chat_model_with_stop | ReActJsonSingleInputOutputParser()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 32,
+   "id": "13ad514e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3a3394a4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor.invoke({\"input\": \"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\"})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "ffc28e29",
+   "metadata": {},
+   "source": [
+    "We can also use an off-the-shelf agent class"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6c41464c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "\n",
+    "agent = initialize_agent(tools, chat_model, agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION, verbose=True)\n",
+    "agent.run(\"Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.1"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

docs/extras/modules/agents/agent_types/self_ask_with_search.ipynb

@@ -13,6 +13,154 @@
   {
    "cell_type": "code",
    "execution_count": 1,
+   "id": "2018da2d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.llms import OpenAI\n",
+    "from langchain.utilities import SerpAPIWrapper\n",
+    "from langchain.agents import initialize_agent, Tool\n",
+    "from langchain.agents import AgentType\n",
+    "\n",
+    "llm = OpenAI(temperature=0)\n",
+    "search = SerpAPIWrapper()\n",
+    "tools = [\n",
+    "    Tool(\n",
+    "        name=\"Intermediate Answer\",\n",
+    "        func=search.run,\n",
+    "        description=\"useful for when you need to ask with search\",\n",
+    "    )\n",
+    "]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "769c5940",
+   "metadata": {},
+   "source": [
+    "## Using LangChain Expression Language\n",
+    "\n",
+    "First we will show how to construct this agent from components using LangChain Expression Language"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "6be0e94d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents.output_parsers import SelfAskOutputParser\n",
+    "from langchain.agents.format_scratchpad import format_log_to_str\n",
+    "from langchain import hub"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "id": "933ca47b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = hub.pull(\"hwchase17/self-ask-with-search\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "d1437a27",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "llm_with_stop = llm.bind(stop=[\"\\nIntermediate answer:\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "d793401e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent = {\n",
+    "    \"input\": lambda x: x[\"input\"],\n",
+    "    # Use some custom observation_prefix/llm_prefix for formatting\n",
+    "    \"agent_scratchpad\": lambda x: format_log_to_str(\n",
+    "        x['intermediate_steps'], \n",
+    "        observation_prefix=\"\\nIntermediate answer: \",\n",
+    "        llm_prefix=\"\",\n",
+    "    ),\n",
+    "} | prompt | llm_with_stop | SelfAskOutputParser()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "643c3bfa",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.agents import AgentExecutor"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "a1bb513c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 15,
+   "id": "5181f35f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "\n",
+      "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m Yes.\n",
+      "Follow up: Who is the reigning men's U.S. Open champion?\u001b[0m\u001b[36;1m\u001b[1;3mMen's US Open Tennis Champions Novak Djokovic earned his 24th major singles title against 2021 US Open champion Daniil Medvedev, 6-3, 7-6 (7-5), 6-3. The victory ties the Serbian player with the legendary Margaret Court for the most Grand Slam wins across both men's and women's singles.\u001b[0m\u001b[32;1m\u001b[1;3m\n",
+      "Follow up: Where is Novak Djokovic from?\u001b[0m\u001b[36;1m\u001b[1;3mBelgrade, Serbia\u001b[0m\u001b[32;1m\u001b[1;3m\n",
+      "So the final answer is: Belgrade, Serbia\u001b[0m\n",
+      "\n",
+      "\u001b[1m> Finished chain.\u001b[0m\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'input': \"What is the hometown of the reigning men's U.S. Open champion?\",\n",
+       " 'output': 'Belgrade, Serbia'}"
+      ]
+     },
+     "execution_count": 15,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent_executor.invoke({\"input\": \"What is the hometown of the reigning men's U.S. Open champion?\"})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "6556f348",
+   "metadata": {},
+   "source": [
+    "## Use off-the-shelf agent"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
    "id": "7e3b513e",
    "metadata": {},
    "outputs": [
@@ -25,10 +173,11 @@
       "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
       "\u001b[32;1m\u001b[1;3m Yes.\n",
       "Follow up: Who is the reigning men's U.S. Open champion?\u001b[0m\n",
-      "Intermediate answer: \u001b[36;1m\u001b[1;3mCarlos Alcaraz Garfia\u001b[0m\n",
-      "\u001b[32;1m\u001b[1;3mFollow up: Where is Carlos Alcaraz Garfia from?\u001b[0m\n",
-      "Intermediate answer: \u001b[36;1m\u001b[1;3mEl Palmar, Spain\u001b[0m\n",
-      "\u001b[32;1m\u001b[1;3mSo the final answer is: El Palmar, Spain\u001b[0m\n",
+      "Intermediate answer: \u001b[36;1m\u001b[1;3mMen's US Open Tennis Champions Novak Djokovic earned his 24th major singles title against 2021 US Open champion Daniil Medvedev, 6-3, 7-6 (7-5), 6-3. The victory ties the Serbian player with the legendary Margaret Court for the most Grand Slam wins across both men's and women's singles.\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3m\n",
+      "Follow up: Where is Novak Djokovic from?\u001b[0m\n",
+      "Intermediate answer: \u001b[36;1m\u001b[1;3mBelgrade, Serbia\u001b[0m\n",
+      "\u001b[32;1m\u001b[1;3mSo the final answer is: Belgrade, Serbia\u001b[0m\n",
       "\n",
       "\u001b[1m> Finished chain.\u001b[0m\n"
      ]
@@ -36,29 +185,15 @@
     {
      "data": {
       "text/plain": [
-       "'El Palmar, Spain'"
+       "'Belgrade, Serbia'"
       ]
      },
-     "execution_count": 1,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "from langchain.llms import OpenAI\nfrom langchain.utilities import SerpAPIWrapper\n",
-    "from langchain.agents import initialize_agent, Tool\n",
-    "from langchain.agents import AgentType\n",
-    "\n",
-    "llm = OpenAI(temperature=0)\n",
-    "search = SerpAPIWrapper()\n",
-    "tools = [\n",
-    "    Tool(\n",
-    "        name=\"Intermediate Answer\",\n",
-    "        func=search.run,\n",
-    "        description=\"useful for when you need to ask with search\",\n",
-    "    )\n",
-    "]\n",
-    "\n",
     "self_ask_with_search = initialize_agent(\n",
     "    tools, llm, agent=AgentType.SELF_ASK_WITH_SEARCH, verbose=True\n",
     ")\n",
@@ -92,7 +227,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.3"
+   "version": "3.10.1"
   },
   "vscode": {
    "interpreter": {