= ({
return (
<>
- {isHTMLDocumentString(content) ? (
-
- ) : (
-
- {renderTokens(tokens)}
-
- )}
+ (
+
+ {renderTokens(tokens)}
+
+ )
>
);
};
diff --git a/src/locales/en-US/playground.ts b/src/locales/en-US/playground.ts
index 2cfed088..2207ff7e 100644
--- a/src/locales/en-US/playground.ts
+++ b/src/locales/en-US/playground.ts
@@ -74,5 +74,6 @@ export default {
'playground.rerank.query': 'Query',
'playground.rerank.rank': 'Rank',
'playground.rerank.score': 'Score',
- 'playground.rerank.query.holder': 'Input your query'
+ 'playground.rerank.query.holder': 'Input your query',
+ 'playground.image.prompt': 'Input Prompt'
};
diff --git a/src/locales/zh-CN/playground.ts b/src/locales/zh-CN/playground.ts
index ffff46ef..789aff9b 100644
--- a/src/locales/zh-CN/playground.ts
+++ b/src/locales/zh-CN/playground.ts
@@ -74,5 +74,6 @@ export default {
'playground.rerank.query': '查询',
'playground.rerank.rank': '排序',
'playground.rerank.score': '分数',
- 'playground.rerank.query.holder': '输入查询'
+ 'playground.rerank.query.holder': '输入查询',
+ 'playground.image.prompt': '输入提示'
};
diff --git a/src/pages/llmodels/components/update-modal.tsx b/src/pages/llmodels/components/update-modal.tsx
index f23e77e0..51550115 100644
--- a/src/pages/llmodels/components/update-modal.tsx
+++ b/src/pages/llmodels/components/update-modal.tsx
@@ -34,7 +34,6 @@ const SEARCH_SOURCE = [
];
const UpdateModal: React.FC = (props) => {
- console.log('addmodel====');
const { title, action, open, onOk, onCancel } = props || {};
const [form] = Form.useForm();
const intl = useIntl();
@@ -261,6 +260,13 @@ const UpdateModal: React.FC = (props) => {
};
const handleOk = (formdata: FormData) => {
+ let obj = {};
+ if (formdata.backend === backendOptionsMap.vllm) {
+ obj = {
+ distributed_inference_across_workers: false,
+ cpu_offloading: false
+ };
+ }
if (formdata.scheduleType === 'manual') {
const gpu = _.find(gpuOptions, (item: any) => {
return item.value === formdata.gpu_selector;
@@ -275,12 +281,14 @@ const UpdateModal: React.FC = (props) => {
gpu_index: gpu.index,
worker_name: gpu.worker_name
}
- : null
+ : null,
+ ...obj
});
} else {
onOk({
..._.omit(formdata, ['scheduleType']),
- gpu_selector: null
+ gpu_selector: null,
+ ...obj
});
}
};
diff --git a/src/pages/playground/components/ground-images.tsx b/src/pages/playground/components/ground-images.tsx
index b2017f1a..8c4dad6d 100644
--- a/src/pages/playground/components/ground-images.tsx
+++ b/src/pages/playground/components/ground-images.tsx
@@ -480,6 +480,11 @@ const GroundImages: React.FC = forwardRef((props, ref) => {
handleAbortFetch={handleStopConversation}
shouldResetMessage={false}
clearAll={handleClear}
+ tools={
+
+ {intl.formatMessage({ id: 'playground.image.prompt' })}
+
+ }
/>
diff --git a/src/utils/index.ts b/src/utils/index.ts
index 8171b543..43e03824 100644
--- a/src/utils/index.ts
+++ b/src/utils/index.ts
@@ -128,12 +128,8 @@ export const cosineSimilarity = (vec1: number[], vec2: number[]) => {
return dotProduct / (magnitudeA * magnitudeB);
};
+const htmlSpecialTags = /^(.|\n|\r)*<\/html>$/i;
+
export const isHTMLDocumentString = (str: string) => {
- try {
- const parser = new DOMParser();
- const doc = parser.parseFromString(str, 'text/html');
- return doc.documentElement.tagName.toLowerCase() === 'html';
- } catch (e) {
- return false;
- }
+ return htmlSpecialTags.test(str?.trim());
};