only_ever_generator 0.1.5 → 0.1.7

package/src/services/open_ai_service.ts

@@ -1,54 +1,54 @@
-import axios from "axios";
-import { parseOpenAiFailureResponse, parseOpenAiSuccessResponse } from "../utils/parse_openai_response";
-import { openAiEndPoint } from "../constants/api_constants";
-
-
-export class OpenAiService{
-    public api_key: string;
-    public model: string;
-    
-    constructor(apiKey: string, model:string){
-        this.api_key = apiKey;
-        this.model =model
-    }
-
-    async sendRequest(prompt: string,content: string){
-        try{
-            let message = [
-                {
-                    "role": "system", "content": prompt
-                },{
-                    "role": "user", "content": content
-                }
-            ];
-            const url = openAiEndPoint();
-            let response = await axios.post(
-                url,
-                { 
-                    "model": this.model,
-                    "messages": message,
-                    "response_format": {"type": "json_object"},
-                },
-                {
-                    headers : {
-                        Authorization : "Bearer "+ this.api_key,
-                        "Content-Type" : ['application/json']
-                    },
-                    
-                }
-        
-            );
-        
-            if(response.status == 200){
-                console.log('success');
-                return parseOpenAiSuccessResponse(response.data) as any
-        
-            }else{
-                console.log('failed');
-                return response.statusText as any;
-            }}catch (err:any) {
-                return parseOpenAiFailureResponse(err.response);
-            }
-    } 
-}
-
+import axios from "axios";
+import { parseOpenAiFailureResponse, parseOpenAiSuccessResponse } from "../utils/parse_openai_response";
+import { openAiEndPoint } from "../constants/api_constants";
+
+
+export class OpenAiService{
+    public api_key: string;
+    public model: string;
+    
+    constructor(apiKey: string, model:string){
+        this.api_key = apiKey;
+        this.model =model
+    }
+
+    async sendRequest(prompt: string,content: string){
+        try{
+            let message = [
+                {
+                    "role": "system", "content": prompt
+                },{
+                    "role": "user", "content": content
+                }
+            ];
+            const url = openAiEndPoint();
+            let response = await axios.post(
+                url,
+                { 
+                    "model": this.model,
+                    "messages": message,
+                    "response_format": {"type": "json_object"},
+                },
+                {
+                    headers : {
+                        Authorization : "Bearer "+ this.api_key,
+                        "Content-Type" : ['application/json']
+                    },
+                    
+                }
+        
+            );
+        
+            if(response.status == 200){
+                console.log('success');
+                return parseOpenAiSuccessResponse(response.data) as any
+        
+            }else{
+                console.log('failed');
+                return response.statusText as any;
+            }}catch (err:any) {
+                return parseOpenAiFailureResponse(err.response);
+            }
+    } 
+}
+

package/src/typology_gen/generate_typology.ts

@@ -1,65 +1,65 @@
-import { OpenAiService } from "../services/open_ai_service";
-import { returnTypologyData } from "../parse/response_format_typology";
-
-export class GenerateTypology{
-    public openAiService: OpenAiService;
-    public prompt:string = '';
-    public content:string = '';
-    expectedFields: Array<string>;
-    constructor(openAiService: OpenAiService, prompt:string,content:string,expected_fields: Array<string>){
-        this.openAiService = openAiService;
-        this.prompt = prompt;
-        this.content = content;
-        this.expectedFields = expected_fields.map((elem:string)=>elem.toLowerCase());
-    }
-    async generate(){
-      const response =  await this.openAiService?.sendRequest(this.prompt,this.content);
-      response['type'] = 'typology';
-      response.metadata = {
-        "req_time": response.generated_at,
-        "req_type": response.type,
-        "req_tokens": response.usage_data?.prompt_tokens,
-        "res_tokens": response.usage_data?.completion_tokens,
-    };
-      if(response.status_code == 200){
-        return this.parseTypologyOnSuccess(response);
-      } else {
-        return response;
-      }
-    }
-
-
-    parseTypologyOnSuccess(responseData : any){
-        responseData.metadata.status = "completed";
-        const generatedContent = responseData.generated_content;
-        return {
-            status_code: 200,
-            metadata: responseData.metadata,
-            field: this.parseFields(generatedContent.field),
-            concepts: generatedContent.concepts,
-            facts: generatedContent.facts,
-            generate_cards : generatedContent.generate_cards,
-            summary_cards: generatedContent.summary_cards,
-            type: responseData.type
-        }
-    }
-
-    parseFields(fields: Array<string>){
-      const fieldKeys = ['primary_field', 'secondary_field', 'tertiary_field'];
-      return fields.slice(0,3).map((item,index)=> (
-        {
-          [fieldKeys[index]]: item,
-          "reconcile" : !(this.expectedFields.includes(item.toLowerCase()))
-        }
-      ));
-    }
-
-    async parseTypologyOnFailure(responseData:any){
-        responseData.metadata.status = 'failed';
-        return {
-            status_code: responseData.status_code,
-            metadata: responseData.metadata,
-        }
-    }
-
+import { OpenAiService } from "../services/open_ai_service";
+import { returnTypologyData } from "../parse/response_format_typology";
+
+export class GenerateTypology{
+    public openAiService: OpenAiService;
+    public prompt:string = '';
+    public content:string = '';
+    expectedFields: Array<string>;
+    constructor(openAiService: OpenAiService, prompt:string,content:string,expected_fields: Array<string>){
+        this.openAiService = openAiService;
+        this.prompt = prompt;
+        this.content = content;
+        this.expectedFields = expected_fields.map((elem:string)=>elem.toLowerCase());
+    }
+    async generate(){
+      const response =  await this.openAiService?.sendRequest(this.prompt,this.content);
+      response['type'] = 'typology';
+      response.metadata = {
+        "req_time": response.generated_at,
+        "req_type": response.type,
+        "req_tokens": response.usage_data?.prompt_tokens,
+        "res_tokens": response.usage_data?.completion_tokens,
+    };
+      if(response.status_code == 200){
+        return this.parseTypologyOnSuccess(response);
+      } else {
+        return response;
+      }
+    }
+
+
+    parseTypologyOnSuccess(responseData : any){
+        responseData.metadata.status = "completed";
+        const generatedContent = responseData.generated_content;
+        return {
+            status_code: 200,
+            metadata: responseData.metadata,
+            field: this.parseFields(generatedContent.field),
+            concepts: generatedContent.concepts,
+            facts: generatedContent.facts,
+            generate_cards : generatedContent.generate_cards,
+            summary_cards: generatedContent.summary_cards,
+            type: responseData.type
+        }
+    }
+
+    parseFields(fields: Array<string>){
+      const fieldKeys = ['primary_field', 'secondary_field', 'tertiary_field'];
+      return fields.slice(0,3).map((item,index)=> (
+        {
+          [fieldKeys[index]]: item,
+          "reconcile" : !(this.expectedFields.includes(item.toLowerCase()))
+        }
+      ));
+    }
+
+    async parseTypologyOnFailure(responseData:any){
+        responseData.metadata.status = 'failed';
+        return {
+            status_code: responseData.status_code,
+            metadata: responseData.metadata,
+        }
+    }
+
 }

package/src/utils/generate_args.ts

@@ -1,38 +1,38 @@
-export class GenerateArgs{
-    public generate_card : boolean = false;
-    public generate_typology: boolean = false;
-    public generate_summary: boolean = false;
-    public prompts = {
-        typology_prompt: '',
-        card_gen_prompt: '',
-        summary_prompt: ''
-    }
-
-    constructor(generate_card: boolean,generate_typology: boolean, generate_summary: boolean, prompts = {
-        typology_prompt: '',
-        card_gen_prompt: '',
-        summary_prompt: ''
-    }){
-        this.generate_card = generate_card;
-        this.generate_typology = generate_typology;
-        this.generate_summary = generate_summary;
-        this.prompts = prompts
-    }
-
-    getWhatNeedsToBeGenerated(){
-        let returnData = [];
-        if(this.generate_typology == true){
-            returnData.push('generate_tyopology')
-        }
-        if(this.generate_summary == true){
-            returnData.push('generate_summary')
-        }
-        if(this.generate_card == true){
-            returnData.push('generate_card');
-        }
-        return returnData;
-        
-        
-    }
-
+export class GenerateArgs{
+    public generate_card : boolean = false;
+    public generate_typology: boolean = false;
+    public generate_summary: boolean = false;
+    public prompts = {
+        typology_prompt: '',
+        card_gen_prompt: '',
+        summary_prompt: ''
+    }
+
+    constructor(generate_card: boolean,generate_typology: boolean, generate_summary: boolean, prompts = {
+        typology_prompt: '',
+        card_gen_prompt: '',
+        summary_prompt: ''
+    }){
+        this.generate_card = generate_card;
+        this.generate_typology = generate_typology;
+        this.generate_summary = generate_summary;
+        this.prompts = prompts
+    }
+
+    getWhatNeedsToBeGenerated(){
+        let returnData = [];
+        if(this.generate_typology == true){
+            returnData.push('generate_tyopology')
+        }
+        if(this.generate_summary == true){
+            returnData.push('generate_summary')
+        }
+        if(this.generate_card == true){
+            returnData.push('generate_card');
+        }
+        return returnData;
+        
+        
+    }
+
 }

package/src/utils/parse_openai_response.ts

@@ -1,21 +1,21 @@
-export function parseOpenAiSuccessResponse(responseData: any){
-    let choices =  JSON.parse(responseData.choices[0].message.content);
-    let usuage = responseData.usage;
-    let createdTime = responseData.created;
-    return {
-        'status_code': 200,
-        'usage_data': usuage,
-        'generated_content':choices,
-        'generated_at': createdTime
-    }
-}
-
-export function parseOpenAiFailureResponse(errorResponse: any){
-    let usuage = errorResponse.usuage;
-    // let statusCode = 
-    return {
-        'status_code': errorResponse.status,
-
-
-    }
+export function parseOpenAiSuccessResponse(responseData: any){
+    let choices =  JSON.parse(responseData.choices[0].message.content);
+    let usuage = responseData.usage;
+    let createdTime = responseData.created;
+    return {
+        'status_code': 200,
+        'usage_data': usuage,
+        'generated_content':choices,
+        'generated_at': createdTime
+    }
+}
+
+export function parseOpenAiFailureResponse(errorResponse: any){
+    let usuage = errorResponse.usuage;
+    // let statusCode = 
+    return {
+        'status_code': errorResponse.status,
+
+
+    }
 }

package/tsconfig.json

@@ -1,12 +1,12 @@
-{
-  "compilerOptions": {
-    "target": "es2016" /* Set the JavaScript language version for emitted JavaScript and include compatible library declarations. */,
-    "module": "commonjs" /* Specify what module code is generated. */,
-    "esModuleInterop": true /* Emit additional JavaScript to ease support for importing CommonJS modules. This enables 'allowSyntheticDefaultImports' for type compatibility. */,
-    "forceConsistentCasingInFileNames": true /* Ensure that casing is correct in imports. */,
-    "strict": true,
-    "rootDir": "./src",
-    "outDir": "./dist" /* Enable all strict type-checking options. */
-  },
-  "exclude": ["./src/*/*.d.ts"]
-}
+{
+  "compilerOptions": {
+    "target": "es2016" /* Set the JavaScript language version for emitted JavaScript and include compatible library declarations. */,
+    "module": "commonjs" /* Specify what module code is generated. */,
+    "esModuleInterop": true /* Emit additional JavaScript to ease support for importing CommonJS modules. This enables 'allowSyntheticDefaultImports' for type compatibility. */,
+    "forceConsistentCasingInFileNames": true /* Ensure that casing is correct in imports. */,
+    "strict": true,
+    "rootDir": "./src",
+    "outDir": "./dist" /* Enable all strict type-checking options. */
+  },
+  "exclude": ["./src/*/*.d.ts"]
+}