refactor(graph): move the cfg(test) OUTPUT_KEY import into llm's test module
PR #24 follow-up 7.
This commit is contained in:
1 file changed
+1
-2
+1
-2
@@ -1,7 +1,5 @@
|
|||||||
use super::state::StateManager;
|
use super::state::StateManager;
|
||||||
use super::state_updates;
|
use super::state_updates;
|
||||||
#[cfg(test)]
|
|
||||||
use super::state_updates::OUTPUT_KEY;
|
|
||||||
use super::structured;
|
use super::structured;
|
||||||
use super::types::LlmNode;
|
use super::types::LlmNode;
|
||||||
use crate::client::{Model, ModelType, call_chat_completions};
|
use crate::client::{Model, ModelType, call_chat_completions};
|
||||||
@@ -475,6 +473,7 @@ fn format_schema_hint(schema: &Value) -> String {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
use super::super::state_updates::OUTPUT_KEY;
|
||||||
use super::super::types::*;
|
use super::super::types::*;
|
||||||
use super::*;
|
use super::*;
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
|
|||||||
Reference in new issue
Block a user