fix linting

This commit is contained in:
Hamzah Abdulfattah
2023-08-20 01:43:35 +01:00
parent 3b0c5b0c0d
commit 939daff0a1
2 changed files with 52 additions and 77 deletions
@@ -1,35 +1,36 @@
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface';
import { TextSplitter } from 'langchain/text_splitter';
import { SerpAPILoader } from 'langchain/document_loaders/web/serpapi';import { getCredentialData, getCredentialParam } from '../../../src';
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
import { TextSplitter } from 'langchain/text_splitter'
import { SerpAPILoader } from 'langchain/document_loaders/web/serpapi'
import { getCredentialData, getCredentialParam } from '../../../src'
class SerpAPI_DocumentLoaders implements INode {
label: string;
name: string;
version: number;
description: string;
type: string;
icon: string;
category: string;
baseClasses: string[];
credential: INodeParams;
inputs: INodeParams[];
label: string
name: string
version: number
description: string
type: string
icon: string
category: string
baseClasses: string[]
credential: INodeParams
inputs: INodeParams[]
constructor() {
this.label = 'SerpApi For Web Search';
this.name = 'serpApi';
this.version = 1.0;
this.type = 'Document';
this.icon = 'serp.png';
this.category = 'Document Loaders';
this.description = 'Load and process data from web search results';
this.baseClasses = [this.type];
this.label = 'SerpApi For Web Search'
this.name = 'serpApi'
this.version = 1.0
this.type = 'Document'
this.icon = 'serp.png'
this.category = 'Document Loaders'
this.description = 'Load and process data from web search results'
this.baseClasses = [this.type]
this.credential = {
label: 'Connect Credential',
name: 'credential',
type: 'credential',
optional: false,
credentialNames: ['serpApi']
};
label: 'Connect Credential',
name: 'credential',
type: 'credential',
optional: false,
credentialNames: ['serpApi']
}
this.inputs = [
{
label: 'Query',
@@ -49,21 +50,21 @@ class SerpAPI_DocumentLoaders implements INode {
optional: true,
additionalParams: true
}
];
]
}
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
const textSplitter = nodeData.inputs?.textSplitter as TextSplitter;
const query = nodeData.inputs?.query as string;
const metadata = nodeData.inputs?.metadata;
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
const textSplitter = nodeData.inputs?.textSplitter as TextSplitter
const query = nodeData.inputs?.query as string
const metadata = nodeData.inputs?.metadata
const credentialData = await getCredentialData(nodeData.credential ?? '', options);
const serpApiKey = getCredentialParam('serpApiKey', credentialData, nodeData);
const loader = new SerpAPILoader({ q: query, apiKey: serpApiKey });
const docs = textSplitter ? await loader.loadAndSplit() : await loader.load();
const credentialData = await getCredentialData(nodeData.credential ?? '', options)
const serpApiKey = getCredentialParam('serpApiKey', credentialData, nodeData)
const loader = new SerpAPILoader({ q: query, apiKey: serpApiKey })
const docs = textSplitter ? await loader.loadAndSplit() : await loader.load()
if (metadata) {
const parsedMetadata = typeof metadata === 'object' ? metadata : JSON.parse(metadata);
const parsedMetadata = typeof metadata === 'object' ? metadata : JSON.parse(metadata)
return docs.map((doc) => {
return {
...doc,
@@ -71,12 +72,12 @@ class SerpAPI_DocumentLoaders implements INode {
...doc.metadata,
...parsedMetadata
}
};
});
}
})
}
return docs;
return docs
}
}
module.exports = { nodeClass: SerpAPI_DocumentLoaders };
module.exports = { nodeClass: SerpAPI_DocumentLoaders }