Merge delimiter with last known buffer
This commit is contained in:
		
							parent
							
								
									4d27497bc7
								
							
						
					
					
						commit
						ddd17798e7
					
				| 
						 | 
				
			
			@ -163,7 +163,14 @@ export const chatRequest = async (
 | 
			
		|||
 | 
			
		||||
      let maxLen = Math.min(opts.maxTokens || chatSettings.max_tokens || maxTokens, maxTokens)
 | 
			
		||||
 | 
			
		||||
      let inputPrompt = startSequence
 | 
			
		||||
      let midDel = ''
 | 
			
		||||
      for (let i = 0, l = delimiter.length; i < l; i++) {
 | 
			
		||||
        const chk = delimiter.slice(0, i)
 | 
			
		||||
        if ((providerData.knownBuffer || '').slice(0 - (i + 1)) === chk) midDel = chk
 | 
			
		||||
      }
 | 
			
		||||
      midDel = midDel.length ? delimiter.slice(0, 0 - midDel.length) : delimiter
 | 
			
		||||
 | 
			
		||||
      let inputPrompt = midDel
 | 
			
		||||
 | 
			
		||||
      const getNewWs = ():Promise<WebSocket> => new Promise<WebSocket>((resolve, reject) => {
 | 
			
		||||
        // console.warn('requesting new ws')
 | 
			
		||||
| 
						 | 
				
			
			@ -185,7 +192,7 @@ export const chatRequest = async (
 | 
			
		|||
            throw err
 | 
			
		||||
          }
 | 
			
		||||
          // console.warn('got new ws')
 | 
			
		||||
          inputPrompt = lastPrompt
 | 
			
		||||
          inputPrompt = lastPrompt + delimiter
 | 
			
		||||
          providerData.knownBuffer = ''
 | 
			
		||||
          providerData.ws = nws
 | 
			
		||||
          resolve(nws)
 | 
			
		||||
| 
						 | 
				
			
			@ -234,7 +241,7 @@ export const chatRequest = async (
 | 
			
		|||
        ws = await getNewWs()
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      inputPrompt += delimiter + nextPrompt
 | 
			
		||||
      inputPrompt += nextPrompt
 | 
			
		||||
      providerData.knownBuffer += inputPrompt
 | 
			
		||||
    
 | 
			
		||||
      // console.log(
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue