feat: add proxy config
このコミットが含まれているのは:
コミット
311197873b
3
index.ts
3
index.ts
|
@ -2,12 +2,11 @@ import {You} from "./model/you";
|
|||
import Koa from 'koa';
|
||||
import Router from 'koa-router'
|
||||
import bodyParser from 'koa-bodyparser';
|
||||
import {Readable} from "stream";
|
||||
|
||||
const app = new Koa();
|
||||
const router = new Router();
|
||||
app.use(bodyParser());
|
||||
const you = new You();
|
||||
const you = new You({proxy: process.env.https_proxy || process.env.http_proxy});
|
||||
|
||||
interface AskReq {
|
||||
prompt: string;
|
||||
|
|
|
@ -1,5 +1,9 @@
|
|||
import {Stream} from "stream";
|
||||
|
||||
export interface ChatOptions {
|
||||
proxy?: string;
|
||||
}
|
||||
|
||||
export interface Response {
|
||||
text: string | null;
|
||||
other: any;
|
||||
|
@ -23,6 +27,13 @@ export interface HistoryItem {
|
|||
|
||||
|
||||
export abstract class Chat {
|
||||
protected proxy: string | undefined;
|
||||
|
||||
constructor(options?: ChatOptions) {
|
||||
this.proxy = options?.proxy;
|
||||
}
|
||||
|
||||
public abstract ask(req: Request): Promise<Response>
|
||||
|
||||
public abstract askStream(req: Request): Promise<ResponseStream>
|
||||
}
|
||||
|
|
|
@ -5,7 +5,7 @@ import tlsClient from 'tls-client';
|
|||
import {Session} from "tls-client/dist/esm/sessions";
|
||||
import {Params} from "tls-client/dist/esm/types";
|
||||
import {toEventCB, toEventStream} from "../../utils";
|
||||
import {Chat, Request, Response, ResponseStream} from "../index";
|
||||
import {Chat, ChatOptions, Request, Response, ResponseStream} from "../index";
|
||||
|
||||
const userAgent = new UserAgent();
|
||||
|
||||
|
@ -65,11 +65,13 @@ interface SearchResult {
|
|||
export class You extends Chat {
|
||||
private session: Session;
|
||||
|
||||
constructor() {
|
||||
super();
|
||||
constructor(props: ChatOptions) {
|
||||
super(props);
|
||||
this.session = new tlsClient.Session({clientIdentifier: 'chrome_108'});
|
||||
this.session.headers = this.getHeaders();
|
||||
this.session.proxy = "http://192.168.0.155:10811";
|
||||
if (this.proxy) {
|
||||
this.session.proxy = this.proxy;
|
||||
}
|
||||
}
|
||||
|
||||
private async request(req: Request) {
|
||||
|
|
読み込み中…
新しいイシューから参照