notebook fmt (#12498)

This commit is contained in:
Bagatur
2023-10-29 15:50:09 -07:00
committed by GitHub
parent 56cc5b847c
commit 2424fff3f1
342 changed files with 8261 additions and 6796 deletions

View File

@@ -1,3 +1,4 @@
import argparse
import importlib
import inspect
import json
@@ -5,7 +6,6 @@ import logging
import os
import re
from pathlib import Path
import argparse
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

View File

@@ -3,13 +3,9 @@ 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
from langchain.llms.base import LLM, BaseLLM
INTEGRATIONS_DIR = (
Path(os.path.abspath(__file__)).parents[1]
/ "docs"
/ "integrations"
)
INTEGRATIONS_DIR = Path(os.path.abspath(__file__)).parents[1] / "docs" / "integrations"
LLM_IGNORE = ("FakeListLLM", "OpenAIChat", "PromptLayerOpenAIChat")
LLM_FEAT_TABLE_CORRECTION = {
"TextGen": {"_astream": False, "_agenerate": False},
@@ -43,7 +39,7 @@ Each LLM integration can optionally provide native implementations for async, st
{table}
"""
""" # noqa: E501
CHAT_MODEL_TEMPLATE = """\
---
@@ -64,7 +60,7 @@ The table shows, for each integration, which features have been implemented with
{table}
"""
""" # noqa: E501
def get_llm_table():