Skip to content

Commit

Permalink
Merge pull request #660 from PrefectHQ/fix-slackbot-containerization
Browse files Browse the repository at this point in the history
scoop aux settings from regular env if no ~/.marvin/.env
  • Loading branch information
zzstoatzz authored Dec 1, 2023
2 parents 2edaaa1 + fe40e95 commit 1ed69f2
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion src/marvin/tools/retrieval.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,21 @@
import asyncio
import json
import os
from typing import Optional

import httpx
from typing_extensions import Literal

import marvin

HOST, PORT = marvin.settings.chroma_server_host, marvin.settings.chroma_server_http_port
try:
HOST, PORT = (
marvin.settings.chroma_server_host,
marvin.settings.chroma_server_http_port,
)
except AttributeError:
HOST = os.environ.get("MARVIN_CHROMA_SERVER_HOST", "localhost")
PORT = os.environ.get("MARVIN_CHROMA_SERVER_HTTP_PORT", 8000)

QueryResultType = Literal["documents", "distances", "metadatas"]

Expand Down

0 comments on commit 1ed69f2

Please sign in to comment.