Commit 7ec29bbe authored by Joel's avatar Joel

feat: node add default value

parent f1d44a4c
...@@ -48,21 +48,10 @@ export const NODES_INITIAL_DATA = { ...@@ -48,21 +48,10 @@ export const NODES_INITIAL_DATA = {
...KnowledgeRetrievalDefault.defaultValue, ...KnowledgeRetrievalDefault.defaultValue,
}, },
[BlockEnum.IfElse]: { [BlockEnum.IfElse]: {
_targetBranches: [
{
id: 'if-true',
name: 'IS TRUE',
},
{
id: 'if-false',
name: 'IS FALSE',
},
],
type: BlockEnum.IfElse, type: BlockEnum.IfElse,
title: '', title: '',
desc: '', desc: '',
logical_operator: 'and',
conditions: [],
...IfElseDefault.defaultValue, ...IfElseDefault.defaultValue,
}, },
[BlockEnum.Code]: { [BlockEnum.Code]: {
......
import type { NodeDefault } from '../../types' import type { NodeDefault } from '../../types'
import type { CodeNodeType } from './types' import { CodeLanguage, type CodeNodeType } from './types'
const nodeDefault: NodeDefault<CodeNodeType> = { const nodeDefault: NodeDefault<CodeNodeType> = {
defaultValue: {}, defaultValue: {
code: '',
code_language: CodeLanguage.python3,
variables: [],
outputs: [],
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
import type { NodeDefault } from '../../types' import type { NodeDefault } from '../../types'
import type { EndNodeType } from './types' import { type EndNodeType, EndVarType } from './types'
const nodeDefault: NodeDefault<EndNodeType> = { const nodeDefault: NodeDefault<EndNodeType> = {
defaultValue: {}, defaultValue: {
outputs: {
type: EndVarType.none,
plain_text_selector: [],
structured_variables: [],
},
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
import type { NodeDefault } from '../../types' import type { NodeDefault } from '../../types'
import type { HttpNodeType } from './types' import { AuthorizationType, BodyType, type HttpNodeType, Method } from './types'
const nodeDefault: NodeDefault<HttpNodeType> = { const nodeDefault: NodeDefault<HttpNodeType> = {
defaultValue: {}, defaultValue: {
variables: [],
method: Method.get,
url: '',
authorization: {
type: AuthorizationType.none,
config: null,
},
headers: '',
params: '',
body: {
type: BodyType.none,
data: '',
},
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
...@@ -44,7 +44,7 @@ export type Authorization = { ...@@ -44,7 +44,7 @@ export type Authorization = {
type: APIType type: APIType
api_key: string api_key: string
header?: string header?: string
} } | null
} }
export type HttpNodeType = CommonNodeType & { export type HttpNodeType = CommonNodeType & {
......
import type { NodeDefault } from '../../types' import type { NodeDefault } from '../../types'
import type { IfElseNodeType } from './types' import { type IfElseNodeType, LogicalOperator } from './types'
const nodeDefault: NodeDefault<IfElseNodeType> = { const nodeDefault: NodeDefault<IfElseNodeType> = {
defaultValue: {}, defaultValue: {
_targetBranches: [
{
id: 'if-true',
name: 'IS TRUE',
},
{
id: 'if-false',
name: 'IS FALSE',
},
],
logical_operator: LogicalOperator.and,
conditions: [],
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
import type { NodeDefault } from '../../types' import type { NodeDefault } from '../../types'
import type { KnowledgeRetrievalNodeType } from './types' import type { KnowledgeRetrievalNodeType } from './types'
import { RETRIEVE_TYPE } from '@/types/app'
const nodeDefault: NodeDefault<KnowledgeRetrievalNodeType> = { const nodeDefault: NodeDefault<KnowledgeRetrievalNodeType> = {
defaultValue: {}, defaultValue: {
query_variable_selector: [],
dataset_ids: [],
retrieval_mode: RETRIEVE_TYPE.oneWay,
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
...@@ -3,7 +3,23 @@ import type { LLMNodeType } from './types' ...@@ -3,7 +3,23 @@ import type { LLMNodeType } from './types'
const nodeDefault: NodeDefault<LLMNodeType> = { const nodeDefault: NodeDefault<LLMNodeType> = {
defaultValue: { defaultValue: {
model: {
provider: '',
name: '',
mode: 'chat',
completion_params: {
temperature: 0.7,
},
},
variables: [],
prompt: [],
context: {
enabled: false,
variable_selector: [],
},
vision: {
enabled: false,
},
}, },
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
......
...@@ -12,7 +12,7 @@ export type LLMNodeType = CommonNodeType & { ...@@ -12,7 +12,7 @@ export type LLMNodeType = CommonNodeType & {
} }
vision: { vision: {
enabled: boolean enabled: boolean
configs: { configs?: {
detail: Resolution detail: Resolution
} }
} }
......
...@@ -2,7 +2,18 @@ import type { NodeDefault } from '../../types' ...@@ -2,7 +2,18 @@ import type { NodeDefault } from '../../types'
import type { QuestionClassifierNodeType } from './types' import type { QuestionClassifierNodeType } from './types'
const nodeDefault: NodeDefault<QuestionClassifierNodeType> = { const nodeDefault: NodeDefault<QuestionClassifierNodeType> = {
defaultValue: {}, defaultValue: {
query_variable_selector: [],
model: {
provider: '',
name: '',
mode: 'chat',
completion_params: {
temperature: 0.7,
},
},
classes: [],
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
...@@ -14,7 +14,7 @@ export const mockData: QuestionClassifierNodeType = { ...@@ -14,7 +14,7 @@ export const mockData: QuestionClassifierNodeType = {
temperature: 0.7, temperature: 0.7,
}, },
}, },
topics: [ classes: [
{ {
id: '1', id: '1',
name: 'topic 1', name: 'topic 1',
......
...@@ -13,7 +13,7 @@ const Node: FC<NodeProps<QuestionClassifierNodeType>> = (props) => { ...@@ -13,7 +13,7 @@ const Node: FC<NodeProps<QuestionClassifierNodeType>> = (props) => {
const { data } = props const { data } = props
const { provider, name: modelId } = data.model const { provider, name: modelId } = data.model
// const tempTopics = data.topics // const tempTopics = data.topics
const topics = data.topics const topics = data.classes
const { const {
textGenerationModelList, textGenerationModelList,
} = useTextGenerationCurrentProviderAndModelAndModelList() } = useTextGenerationCurrentProviderAndModelAndModelList()
......
...@@ -62,7 +62,7 @@ const Panel: FC<NodeProps<QuestionClassifierNodeType>> = ({ ...@@ -62,7 +62,7 @@ const Panel: FC<NodeProps<QuestionClassifierNodeType>> = ({
title={t(`${i18nPrefix}.class`)} title={t(`${i18nPrefix}.class`)}
> >
<ClassList <ClassList
list={inputs.topics} list={inputs.classes}
onChange={handleTopicsChange} /> onChange={handleTopicsChange} />
</Field> </Field>
<Field <Field
......
...@@ -9,7 +9,7 @@ export type Topic = { ...@@ -9,7 +9,7 @@ export type Topic = {
export type QuestionClassifierNodeType = CommonNodeType & { export type QuestionClassifierNodeType = CommonNodeType & {
query_variable_selector: ValueSelector query_variable_selector: ValueSelector
model: ModelConfig model: ModelConfig
topics: Topic[] classes: Topic[]
instruction: string instruction: string
memory: Memory memory: Memory
} }
...@@ -2,7 +2,9 @@ import type { NodeDefault } from '../../types' ...@@ -2,7 +2,9 @@ import type { NodeDefault } from '../../types'
import type { TemplateTransformNodeType } from './types' import type { TemplateTransformNodeType } from './types'
const nodeDefault: NodeDefault<TemplateTransformNodeType> = { const nodeDefault: NodeDefault<TemplateTransformNodeType> = {
defaultValue: {}, defaultValue: {
variables: [],
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
...@@ -2,7 +2,10 @@ import type { NodeDefault } from '../../types' ...@@ -2,7 +2,10 @@ import type { NodeDefault } from '../../types'
import type { ToolNodeType } from './types' import type { ToolNodeType } from './types'
const nodeDefault: NodeDefault<ToolNodeType> = { const nodeDefault: NodeDefault<ToolNodeType> = {
defaultValue: {}, defaultValue: {
tool_inputs: [],
tool_parameters: {},
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
...@@ -2,7 +2,10 @@ import type { NodeDefault } from '../../types' ...@@ -2,7 +2,10 @@ import type { NodeDefault } from '../../types'
import type { VariableAssignerNodeType } from './types' import type { VariableAssignerNodeType } from './types'
const nodeDefault: NodeDefault<VariableAssignerNodeType> = { const nodeDefault: NodeDefault<VariableAssignerNodeType> = {
defaultValue: {}, defaultValue: {
output_type: 'string',
variables: [],
},
getAvailablePrevNodes() { getAvailablePrevNodes() {
return [] return []
}, },
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment