review 2

dust-tt · Sep 1, 2023 · ca0e3f3 · ca0e3f3
1 parent 876225a
commit ca0e3f3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 11 deletions.
diff --git a/core/bin/dust_api.rs b/core/bin/dust_api.rs
@@ -728,15 +728,20 @@ async fn runs_create_stream(
             });
         }
         Err((_, api_error)) => {
-            if let Some(error) = &api_error.0.error {
-                let _ = tx.send(json!({
-                    "type": "error",
-                    "content": {
-                        "code": error.code,
-                        "message": error.message,
-                    },
-                }));
-            }
+            let error = match api_error.0.error {
+                Some(error) => error,
+                None => APIError {
+                    code: "internal_server_error".to_string(),
+                    message: "The app execution failed unexpectedly".to_string(),
+                },
+            };
+            let _ = tx.send(json!({
+                "type": "error",
+                "content": {
+                    "code": error.code,
+                    "message": error.message,
+                },
+            }));
         }
     }
 
@@ -1219,7 +1224,7 @@ fn error_response(
             error: Some(APIError {
                 code: code.to_string(),
                 message: match error {
-                    Some(err) => format!("{}\nError: {:?}", message, err),
+                    Some(err) => format!("{} (error: {:?})", message, err),
                     None => message.to_string(),
                 },
             }),

diff --git a/core/src/data_sources/data_source.rs b/core/src/data_sources/data_source.rs
@@ -449,7 +449,7 @@ impl DataSource {
         payload.insert(field_name, field_value.into());
 
         let field_condition = qdrant::FieldCondition {
-            key: "document_id".to_string(),
+            key: "document_id_hash".to_string(),
             r#match: Some(qdrant::Match {
                 match_value: Some(qdrant::r#match::MatchValue::Text(document_id)),
             }),