Create q5.py

q5.py

@@ -0,0 +1,54 @@
+import openai
+
+def extract_entities(api_key, text, model="text-davinci-003"):
+    """
+    Perform basic entity extraction using LangChain (GPT-3 model).
+
+    Parameters:
+    - api_key (str): Your OpenAI API key.
+    - text (str): The text from which entities are to be extracted.
+    - model (str): The GPT-3 model to use. Defaults to 'text-davinci-003'.
+
+    Returns:
+    - entities (list): List of entities found in the text.
+    """
+    # Set the OpenAI API key
+    openai.api_key = api_key
+    
+    # Construct prompt for entity extraction
+    prompt = f"Extract entities from the following text:\n{text}\n\nEntities:"
+    
+    # Generate text completion using LangChain
+    response = openai.Completion.create(
+        engine=model,
+        prompt=prompt,
+        max_tokens=50,  # Adjust max_tokens based on expected entity length
+        stop=None  # Allow the model to generate until the max_tokens is reached
+    )
+
+    # Extract and parse entities from the response
+    extracted_text = response['choices'][0]['text'].strip()
+    entities = extracted_text.split(',')
+    
+    return entities
+
+# Example usage:
+def main():
+    # Replace 'your_openai_api_key_here' with your actual API key
+    api_key = 'your_openai_api_key_here'
+    
+    # Example text
+    text = """
+    Elon Musk is the CEO of SpaceX and Tesla. He was born in Pretoria, South Africa and went to Stanford University.
+    """
+
+    # Extract entities
+    entities = extract_entities(api_key, text)
+    
+    # Print extracted entities
+    print("Entities Found:")
+    for entity in entities:
+        print(entity.strip())
+
+if __name__ == "__main__":
+    main()