|
|
|
@ -4,7 +4,7 @@ from typing import List
|
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
|
|
|
|
from langchain.agents.agent_toolkits.base import BaseToolkit
|
|
|
|
|
from langchain.llms.base import BaseLLM
|
|
|
|
|
from langchain.base_language import BaseLanguageModel
|
|
|
|
|
from langchain.llms.openai import OpenAI
|
|
|
|
|
from langchain.tools import BaseTool
|
|
|
|
|
from langchain.tools.vectorstore.tool import (
|
|
|
|
@ -31,7 +31,7 @@ class VectorStoreToolkit(BaseToolkit):
|
|
|
|
|
"""Toolkit for interacting with a vector store."""
|
|
|
|
|
|
|
|
|
|
vectorstore_info: VectorStoreInfo = Field(exclude=True)
|
|
|
|
|
llm: BaseLLM = Field(default_factory=lambda: OpenAI(temperature=0))
|
|
|
|
|
llm: BaseLanguageModel = Field(default_factory=lambda: OpenAI(temperature=0))
|
|
|
|
|
|
|
|
|
|
class Config:
|
|
|
|
|
"""Configuration for this pydantic object."""
|
|
|
|
@ -65,7 +65,7 @@ class VectorStoreRouterToolkit(BaseToolkit):
|
|
|
|
|
"""Toolkit for routing between vectorstores."""
|
|
|
|
|
|
|
|
|
|
vectorstores: List[VectorStoreInfo] = Field(exclude=True)
|
|
|
|
|
llm: BaseLLM = Field(default_factory=lambda: OpenAI(temperature=0))
|
|
|
|
|
llm: BaseLanguageModel = Field(default_factory=lambda: OpenAI(temperature=0))
|
|
|
|
|
|
|
|
|
|
class Config:
|
|
|
|
|
"""Configuration for this pydantic object."""
|
|
|
|
|