Add optional public URL for langfuse

Author: scezen

gen-ai/orchestrator-core/src/main/kotlin/ai/tock/genai/orchestratorcore/mappers/ObservabilitySettingMapper.kt

@@ -34,7 +34,7 @@ object ObservabilitySettingMapper {
             when(this){
                 is LangfuseObservabilitySetting -> {
                     val secretKey = SecurityUtils.fetchSecretKeyValue(secretKey)
-                    return LangfuseObservabilitySetting(secretKey, publicKey, url)
+                    return LangfuseObservabilitySetting(secretKey, publicKey, url, publicUrl)
                 }
                 else ->
                     throw IllegalArgumentException("Unsupported Observability Setting")
@@ -54,7 +54,7 @@ object ObservabilitySettingMapper {
             when (this) {
                 is LangfuseObservabilitySetting -> {
                     val secretKey = SecurityUtils.createSecretKey(namespace, botId, feature, secretKey)
-                    return LangfuseObservabilitySetting(secretKey, publicKey, url)
+                    return LangfuseObservabilitySetting(secretKey, publicKey, url, publicUrl)
                 }
                 else ->
                     throw IllegalArgumentException("Unsupported Observability Setting")

gen-ai/orchestrator-core/src/main/kotlin/ai/tock/genai/orchestratorcore/models/observability/LangfuseObservabilitySetting.kt

@@ -21,5 +21,6 @@ import ai.tock.shared.security.key.HasSecretKey
 data class LangfuseObservabilitySetting<T>(
     override val secretKey: T,
     val publicKey: String,
-    val url: String
+    val url: String,
+    val publicUrl : String? = null,
 ) : ObservabilitySettingBase<T>(ObservabilityProvider.Langfuse), HasSecretKey<T>

gen-ai/orchestrator-server/src/main/python/server/src/gen_ai_orchestrator/models/observability/langfuse/langfuse_setting.py

@@ -14,7 +14,7 @@
 #
 """Model for creating LangfuseObservabilitySetting."""
 
-from typing import Literal
+from typing import Literal, Optional
 
 from pydantic import AnyUrl, Field
 
@@ -42,6 +42,11 @@ class LangfuseObservabilitySetting(BaseObservabilitySetting):
     url: AnyUrl = Field(
         description='The Langfuse server url', examples=['https://cloud.langfuse.com'], default='http://localhost:3000'
     )
+    public_url: Optional[AnyUrl] = Field(
+        default=None,
+        description="Optional public URL for Langfuse server",
+        examples=["https://public.langfuse.com"]
+    )
     secret_key: SecretKey = Field(
         description='Stores the secret key used to authenticate requests to the Observability Provider API.',
         examples=[RawSecretKey(secret='sk-********************be8f')],

gen-ai/orchestrator-server/src/main/python/server/src/gen_ai_orchestrator/routers/observability_providers_router.py

@@ -119,7 +119,7 @@ async def get_observability_provider_setting_by_id(
             provider=ObservabilityProvider.LANGFUSE,
             secret_key=RawSecretKey(secret='sk-********************be8f'),
             public_key='pk-lf-5e374dc6-e194-4b37-9c07-b77e68ef7d2c',
-            url='https://cloud.langfuse.com'
+            url='https://cloud.langfuse.com',
         )
 
 

gen-ai/orchestrator-server/src/main/python/server/src/gen_ai_orchestrator/services/langchain/factories/callback_handlers/langfuse_callback_handler_factory.py

@@ -63,9 +63,10 @@ def check_observability_setting(self) -> bool:
         return True
 
     def _fetch_settings(self):
+        host = str(self.setting.public_url or self.setting.url)
+        logger.info(f"Using Langfuse host: {host}")
         return {
-            'host': str(self.setting.url),
-            'public_key': self.setting.public_key,
+            'host': host,
             'secret_key': fetch_secret_key_value(self.setting.secret_key),
             'timeout': application_settings.observability_provider_timeout,
             'max_retries': application_settings.observability_provider_max_retries