github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-18 08:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format Templates (#12396)

Browse Source
This commit is contained in:
Erick Friis
2023-10-26 19:44:30 -07:00
committed by GitHub
parent 25c98dbba9
commit 4b16601d33
59 changed files with 800 additions and 441 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
templates/cassandra-synonym-caching/cassandra_synonym_caching/__init__.py
View File

@@ -1,13 +1,12 @@
import os
import cassio
import langchain
from langchain.schema import BaseMessage
from langchain.prompts import ChatPromptTemplate
from langchain.chat_models import ChatOpenAI
from langchain.schema.runnable import RunnableLambda
from langchain.cache import CassandraCache
from langchain.chat_models import ChatOpenAI
from langchain.prompts import ChatPromptTemplate
from langchain.schema import BaseMessage
from langchain.schema.runnable import RunnableLambda
use_cassandra = int(os.environ.get("USE_CASSANDRA_CLUSTER", "0"))
if use_cassandra:

8
templates/cassandra-synonym-caching/cassandra_synonym_caching/cassandra_cluster_init.py
View File

@@ -1,13 +1,13 @@
import os
from cassandra.cluster import Cluster
from cassandra.auth import PlainTextAuthProvider
from cassandra.cluster import Cluster
def get_cassandra_connection():
contact_points = [
cp.strip()
for cp in os.environ.get("CASSANDRA_CONTACT_POINTS", "").split(',')
for cp in os.environ.get("CASSANDRA_CONTACT_POINTS", "").split(",")
if cp.strip()
]
CASSANDRA_KEYSPACE = os.environ["CASSANDRA_KEYSPACE"]
@@ -22,6 +22,8 @@ def get_cassandra_connection():
else:
auth_provider = None
c_cluster = Cluster(contact_points if contact_points else None, auth_provider=auth_provider)
c_cluster = Cluster(
contact_points if contact_points else None, auth_provider=auth_provider
)
session = c_cluster.connect()
return (session, CASSANDRA_KEYSPACE)