mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
26 lines
767 B
Python
26 lines
767 B
Python
|
import os
|
||
|
|
from langchain.chat_models import ChatOpenAI
|
||
|
|
from langchain.output_parsers.json import SimpleJsonOutputParser
|
||
|
|
from elasticsearch import Elasticsearch
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from .prompts import DSL_PROMPT
|
||
|
|
from .elastic_index_info import get_indices_infos
|
||
|
|
|
||
|
|
es_host = os.environ["ELASTIC_SEARCH_SERVER"]
|
||
|
|
es_password = os.environ["ELASTIC_PASSWORD"]
|
||
|
|
|
||
|
|
db = Elasticsearch(
|
||
|
|
es_host,
|
||
|
|
http_auth=('elastic', es_password),
|
||
|
|
ca_certs=Path(__file__).parents[1] / 'http_ca.crt' # Replace with your actual path
|
||
|
|
)
|
||
|
|
|
||
|
|
_model = ChatOpenAI(temperature=0, model="gpt-4")
|
||
|
|
|
||
|
|
chain = {
|
||
|
|
"input": lambda x: x["input"],
|
||
|
|
"indices_info": lambda _: get_indices_infos(db),
|
||
|
|
"top_k": lambda x: x.get("top_k", 5),
|
||
|
|
} | DSL_PROMPT | _model | SimpleJsonOutputParser()
|