Add Moderation trait to OpenRouter

block · Jan 9, 2025 · 97aa29b · 97aa29b
1 parent ba24be7
commit 97aa29b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/crates/goose/src/providers/openrouter.rs b/crates/goose/src/providers/openrouter.rs
@@ -6,6 +6,7 @@ use std::time::Duration;
 
 use super::base::ProviderUsage;
 use super::base::{Provider, Usage};
+use super::base::{Moderation, ModerationResult};
 use super::configs::OpenAiProviderConfig;
 use super::configs::{ModelConfig, ProviderModelConfig};
 use super::model_pricing::cost;
@@ -73,7 +74,7 @@ impl Provider for OpenRouterProvider {
             cost
         )
     )]
-    async fn complete(
+    async fn complete_internal(
         &self,
         system: &str,
         messages: &[Message],
@@ -112,6 +113,13 @@ impl Provider for OpenRouterProvider {
     }
 }
 
+#[async_trait]
+impl Moderation for OpenRouterProvider {
+    async fn moderate_content(&self, content: &str) -> Result<ModerationResult> {
+        Ok(ModerationResult::new(false, None, None))
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;