mirror of
https://github.com/Canner/WrenAI.git
synced 2026-10-24 02:25:59 +08:00
Chore/ai service/minor update (#497)
* add azure_deployment * update init_langfuse
This commit is contained in:
@@ -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(),
|
||||
)
|
||||
|
||||
@@ -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}")
|
||||
|
||||
Reference in New Issue
Block a user