Chore/ai service/minor update (#497)

* add azure_deployment

* update init_langfuse
This commit is contained in:
Chih-Yu Yeh
2024-07-10 10:30:00 +08:00
committed by GitHub
parent ae657bb0ac
commit 2d157c7b29
2 changed files with 9 additions and 0 deletions
@@ -49,6 +49,7 @@ class AsyncTextEmbedder(AzureOpenAITextEmbedder):
self.client = AsyncAzureOpenAI(
azure_endpoint=api_base_url,
azure_deployment=model,
api_version=api_version,
api_key=api_key.resolve_value(),
)
+8
View File
@@ -126,6 +126,14 @@ def remove_trailing_slash(endpoint: str) -> str:
def init_langfuse():
enabled = os.getenv("LANGFUSE_ENABLE", "false")
host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com")
langfuse_context.configure(
enabled=False if enabled.lower() == "false" else True,
public_key=os.getenv("LANGFUSE_PUBLIC_KEY", ""),
secret_key=os.getenv("LANGFUSE_SECRET_KEY", ""),
host=host,
)
logger.info(f"LANGFUSE_ENABLE: {enabled}")
logger.info(f"LANGFUSE_HOST: {host}")