enable .env with defaults

.env-example

@@ -0,0 +1,22 @@
+APP_PORT=7860
+
+LANGCHAIN_ENDPOINT="https://api.smith.langchain.com"
+LANGCHAIN_API_KEY="ls__..."
+LANGCHAIN_TRACING_V2="true"
+LANGCHAIN_PROJECT="streamlit_test"
+
+ANYSCALE_API_KEY="secret_..."
+OPENAI_API_KEY="sk-..."
+ANTHROPIC_API_KEY="sk-ant-..."
+
+DEFAULT_SYSTEM_PROMPT="You are a helpful chatbot."
+
+DEFAULT_MODEL="gpt-3.5-turbo"
+
+DEFAULT_TEMPERATURE=0.7
+MIN_TEMPERATURE=0.0
+MAX_TEMPERATURE=1.0
+
+DEFAULT_MAX_TOKENS=1000
+MIN_MAX_TOKENS=1
+MAX_MAX_TOKENS=100000

docker-compose.yml

@@ -4,6 +4,8 @@ services:
   langchain-streamlit-demo:
     image: langchain-streamlit-demo:latest
     build: .
+    env_file:
+      - .env
     ports:
       - "${APP_PORT:-7860}:${APP_PORT:-7860}"
     command: [

langchain-streamlit-demo/app.py

@@ -1,4 +1,5 @@
 import os
+from typing import Union
 
 import anthropic
 import openai
@@ -53,8 +54,25 @@ model = st.sidebar.selectbox(
 )
 provider = _MODEL_DICT[model]
 
-provider_api_key = st.sidebar.text_input(f"{provider} API key", type="password")
-langsmith_api_key = st.sidebar.text_input(
+
+def api_key_from_env(_provider: str) -> Union[str, None]:
+    if _provider == "OpenAI":
+        return os.environ.get("OPENAI_API_KEY")
+    elif _provider == "Anthropic":
+        return os.environ.get("ANTHROPIC_API_KEY")
+    elif _provider == "Anyscale Endpoints":
+        return os.environ.get("ANYSCALE_API_KEY")
+    elif _provider == "LANGSMITH":
+        return os.environ.get("LANGCHAIN_API_KEY")
+    else:
+        return None
+
+
+provider_api_key = api_key_from_env(provider) or st.sidebar.text_input(
+    f"{provider} API key",
+    type="password",
+)
+langsmith_api_key = api_key_from_env("LANGSMITH") or st.sidebar.text_input(
     "LangSmith API Key (optional)",
     type="password",
 )

langchain-streamlit-demo/llm_stuff.py

@@ -1,3 +1,4 @@
+import os
 from datetime import datetime
 
 import streamlit as st
@@ -16,7 +17,10 @@ _MEMORY = ConversationBufferMemory(
     memory_key="chat_history",
 )
 
-_DEFAULT_SYSTEM_PROMPT = "You are a helpful chatbot."
+_DEFAULT_SYSTEM_PROMPT = os.environ.get(
+    "DEFAULT_SYSTEM_PROMPT",
+    "You are a helpful chatbot.",
+)
 
 _MODEL_DICT = {
     "gpt-3.5-turbo": "OpenAI",
@@ -28,15 +32,15 @@ _MODEL_DICT = {
     "meta-llama/Llama-2-70b-chat-hf": "Anyscale Endpoints",
 }
 _SUPPORTED_MODELS = list(_MODEL_DICT.keys())
-_DEFAULT_MODEL = "gpt-3.5-turbo"
+_DEFAULT_MODEL = os.environ.get("DEFAULT_MODEL", "gpt-3.5-turbo")
 
-_DEFAULT_TEMPERATURE = 0.7
-_MIN_TEMPERATURE = 0.0
-_MAX_TEMPERATURE = 1.0
+_DEFAULT_TEMPERATURE = float(os.environ.get("DEFAULT_TEMPERATURE", 0.7))
+_MIN_TEMPERATURE = float(os.environ.get("MIN_TEMPERATURE", 0.0))
+_MAX_TEMPERATURE = float(os.environ.get("MAX_TEMPERATURE", 1.0))
 
-_DEFAULT_MAX_TOKENS = 1000
-_MIN_TOKENS = 1
-_MAX_TOKENS = 100000
+_DEFAULT_MAX_TOKENS = int(os.environ.get("DEFAULT_MAX_TOKENS", 1000))
+_MIN_TOKENS = int(os.environ.get("MIN_MAX_TOKENS", 1))
+_MAX_TOKENS = int(os.environ.get("MAX_MAX_TOKENS", 100000))
 
 
 def get_llm(