kamp/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2024-07-07 09:30:07 +02:00
Code Issues Packages Projects Releases Wiki Activity

Prevent external service localhost question (#497)

Browse Source 
* Prevent external service localhost question

* add 0.0.0.0 to docker-invalid URL

* clarify hint
This commit is contained in:
Timothy Carambat 2023-12-28 10:47:02 -08:00 committed by GitHub
parent e0a0a8976d
commit d7481671ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.vscode/settings.json vendored
View File

@ -1,5 +1,6 @@
{ {
"cSpell.words": [ "cSpell.words": [
"Dockerized",
"Ollama", "Ollama",
"openai", "openai",
"Qdrant", "Qdrant",

1
docker/Dockerfile
View File

@ -83,6 +83,7 @@ RUN cd ./server && npx prisma migrate deploy --schema=./prisma/schema.prisma
# Setup the environment # Setup the environment
ENV NODE_ENV=production ENV NODE_ENV=production
ENV ANYTHING_LLM_RUNTIME=docker
# Expose the server port # Expose the server port
EXPOSE 3001 EXPOSE 3001

25
server/utils/helpers/updateENV.js
View File

@ -56,7 +56,7 @@ const KEY_MAPPING = {
// LMStudio Settings // LMStudio Settings
LMStudioBasePath: { LMStudioBasePath: {
envKey: "LMSTUDIO_BASE_PATH", envKey: "LMSTUDIO_BASE_PATH",
checks: [isNotEmpty, validLLMExternalBasePath], checks: [isNotEmpty, validLLMExternalBasePath, validDockerizedUrl],
}, },
LMStudioTokenLimit: { LMStudioTokenLimit: {
envKey: "LMSTUDIO_MODEL_TOKEN_LIMIT", envKey: "LMSTUDIO_MODEL_TOKEN_LIMIT",
@ -66,7 +66,7 @@ const KEY_MAPPING = {
// LocalAI Settings // LocalAI Settings
LocalAiBasePath: { LocalAiBasePath: {
envKey: "LOCAL_AI_BASE_PATH", envKey: "LOCAL_AI_BASE_PATH",
checks: [isNotEmpty, validLLMExternalBasePath], checks: [isNotEmpty, validLLMExternalBasePath, validDockerizedUrl],
}, },
LocalAiModelPref: { LocalAiModelPref: {
envKey: "LOCAL_AI_MODEL_PREF", envKey: "LOCAL_AI_MODEL_PREF",
@ -83,7 +83,7 @@ const KEY_MAPPING = {
OllamaLLMBasePath: { OllamaLLMBasePath: {
envKey: "OLLAMA_BASE_PATH", envKey: "OLLAMA_BASE_PATH",
checks: [isNotEmpty, validOllamaLLMBasePath], checks: [isNotEmpty, validOllamaLLMBasePath, validDockerizedUrl],
}, },
OllamaLLMModelPref: { OllamaLLMModelPref: {
envKey: "OLLAMA_MODEL_PREF", envKey: "OLLAMA_MODEL_PREF",
@ -106,7 +106,7 @@ const KEY_MAPPING = {
}, },
EmbeddingBasePath: { EmbeddingBasePath: {
envKey: "EMBEDDING_BASE_PATH", envKey: "EMBEDDING_BASE_PATH",
checks: [isNotEmpty, validLLMExternalBasePath], checks: [isNotEmpty, validLLMExternalBasePath, validDockerizedUrl],
}, },
EmbeddingModelPref: { EmbeddingModelPref: {
envKey: "EMBEDDING_MODEL_PREF", envKey: "EMBEDDING_MODEL_PREF",
@ -126,7 +126,7 @@ const KEY_MAPPING = {
// Chroma Options // Chroma Options
ChromaEndpoint: { ChromaEndpoint: {
envKey: "CHROMA_ENDPOINT", envKey: "CHROMA_ENDPOINT",
checks: [isValidURL, validChromaURL], checks: [isValidURL, validChromaURL, validDockerizedUrl],
}, },
ChromaApiHeader: { ChromaApiHeader: {
envKey: "CHROMA_API_HEADER", envKey: "CHROMA_API_HEADER",
@ -140,7 +140,7 @@ const KEY_MAPPING = {
// Weaviate Options // Weaviate Options
WeaviateEndpoint: { WeaviateEndpoint: {
envKey: "WEAVIATE_ENDPOINT", envKey: "WEAVIATE_ENDPOINT",
checks: [isValidURL], checks: [isValidURL, validDockerizedUrl],
}, },
WeaviateApiKey: { WeaviateApiKey: {
envKey: "WEAVIATE_API_KEY", envKey: "WEAVIATE_API_KEY",
@ -150,7 +150,7 @@ const KEY_MAPPING = {
// QDrant Options // QDrant Options
QdrantEndpoint: { QdrantEndpoint: {
envKey: "QDRANT_ENDPOINT", envKey: "QDRANT_ENDPOINT",
checks: [isValidURL], checks: [isValidURL, validDockerizedUrl],
}, },
QdrantApiKey: { QdrantApiKey: {
envKey: "QDRANT_API_KEY", envKey: "QDRANT_API_KEY",
@ -318,6 +318,17 @@ function isDownloadedModel(input = "") {
return files.includes(input); return files.includes(input);
} }
function validDockerizedUrl(input = "") {
if (process.env.ANYTHING_LLM_RUNTIME !== "docker") return null;
try {
const { hostname } = new URL(input);
if (["localhost", "127.0.0.1", "0.0.0.0"].includes(hostname.toLowerCase()))
return "Localhost, 127.0.0.1, or 0.0.0.0 origins cannot be reached from inside the AnythingLLM container. Please use host.docker.internal, a real machine ip, or domain to connect to your service.";
return null;
} catch {}
return null;
}
// This will force update .env variables which for any which reason were not able to be parsed or // This will force update .env variables which for any which reason were not able to be parsed or
// read from an ENV file as this seems to be a complicating step for many so allowing people to write // read from an ENV file as this seems to be a complicating step for many so allowing people to write
// to the process will at least alleviate that issue. It does not perform comprehensive validity checks or sanity checks // to the process will at least alleviate that issue. It does not perform comprehensive validity checks or sanity checks