Commit 7ec29bbe authored by Joel's avatar Joel

feat: node add default value

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